1. 題目

給定一個整數數組 A，以及一個整數 target 作為目標值，返回滿足 i < j < k 且 A[i] + A[j] + A[k] == target 的元組 i, j, k 的數量。

由于結果會非常大，請返回 結果除以 10^9 + 7 的余數。

示例 1： 輸入：A = [1,1,2,2,3,3,4,4,5,5], target = 8 輸出：20 解釋： 按值枚舉（A[i]，A[j]，A[k]）： (1, 2, 5) 出現 8 次； (1, 3, 4) 出現 8 次； (2, 2, 4) 出現 2 次； (2, 3, 3) 出現 2 次。示例 2： 輸入：A = [1,1,2,2,2,2], target = 5 輸出：12 解釋： A[i] = 1，A[j] = A[k] = 2 出現 12 次： 我們從 [1,1] 中選擇一個 1，有 2 種情況， 從 [2,2,2,2] 中選出兩個 2，有 6 種情況。提示： 3 <= A.length <= 3000 0 <= A[i] <= 100 0 <= target <= 300

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/3sum-with-multiplicity
著作權歸領扣網絡所有。商業轉載請聯系官方授權，非商業轉載請注明出處。

2. 解題

• 先排序
• 固定第一個數 i，然后 j=i+1, k = n-1
• 三數和不為target，則相應的左右端點，移動
• 三數和為target，左右端點旁邊可能有相等，都找出來，左右相乘（左右數值不相等）
• 三數和為target，左右相等的話，左右的總個數取2個的組合數

class Solution { public:int threeSumMulti(vector<int>& A, int target) {int i, j, k, t, n = A.size(), mod = 1e9+7, count = 0, sum, l, r, n1, n2;sort(A.begin(), A.end());for(i = 0; i < n-2; ++i){j = i+1, k = n-1;while(j < k){sum = A[i]+A[j]+A[k];if(sum==target){l = A[j], r = A[k];n1 = n2 = 0;while(j < k && A[j]==l)n1++,j++;while(j <= k && A[k]==r)//等號不能少，上面j++后可能到k了n2++,k--;if(l != r)count += n1*n2;elsecount += (n1+n2)*(n1+n2-1)/2;count %= mod;}else if(sum < target)j++;elsek--;}}return count;} };

244 ms 10.4 MB

總結

以上是生活随笔為你收集整理的LeetCode 923. 三数之和的多种可能（双指针）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。