1. 題目

給定一個(gè)整數(shù)數(shù)組 A，以及一個(gè)整數(shù) target 作為目標(biāo)值，返回滿足 i < j < k 且 A[i] + A[j] + A[k] == target 的元組 i, j, k 的數(shù)量。

由于結(jié)果會(huì)非常大，請(qǐng)返回 結(jié)果除以 10^9 + 7 的余數(shù)。

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

來(lái)源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/3sum-with-multiplicity
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有。商業(yè)轉(zhuǎn)載請(qǐng)聯(lián)系官方授權(quán)，非商業(yè)轉(zhuǎn)載請(qǐng)注明出處。

2. 解題

• 先排序
• 固定第一個(gè)數(shù) i，然后 j=i+1, k = n-1
• 三數(shù)和不為target，則相應(yīng)的左右端點(diǎn)，移動(dòng)
• 三數(shù)和為target，左右端點(diǎn)旁邊可能有相等，都找出來(lái)，左右相乘（左右數(shù)值不相等）
• 三數(shù)和為target，左右相等的話，左右的總個(gè)數(shù)取2個(gè)的組合數(shù)

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)//等號(hào)不能少，上面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

總結(jié)

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

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。