給定一個包含 n 個整數的數組 nums 和一個目標值 target,判斷 nums 中是否存在四個元素 a,b,c 和 d ,使得 a + b + c + d 的值與 target 相等?找出所有滿足條件且不重復的四元組。
注意:
答案中不可以包含重復的四元組。
示例:
給定數組 nums = [1, 0, -1, 0, -2, 2],和 target = 0。
滿足要求的四元組集合為:
[
[-1, 0, 0, 1],
[-2, -1, 1, 2],
[-2, 0, 0, 2]
]
解答:
排序,兩層循環(huán)遍歷,雙指針,轉化為兩數之和。
注意去重的處理方法:第一個數和第二個數是比較現數和前一個數。第三和第四個數是比較現數和下一個數。
class Solution {
public:
vector<vector<int>> fourSum(vector<int>& nums, int target) {
vector<vector<int> > ret;
sort(nums.begin(),nums.end());
if(nums.size()<4) // 若數組長度小于4,返回空ret
return ret;
for(int i=0;i<nums.size();i++){
if(i&&nums[i]==nums[i-1]) // 第一個數的去重
continue;
for(int j=i+1;j<nums.size();j++){
if(j>i+1&&nums[j]==nums[j-1]) // 第二個數的去重
continue;
int t1=j+1,t2=nums.size()-1; // 雙動態(tài)指針
while(t1<t2){
vector<int> v; // 注意v的位置,易錯?。?!
int sum=nums[i]+nums[j]+nums[t1]+nums[t2]; // 四數之和
if(sum==target){
v.push_back(nums[i]);
v.push_back(nums[j]);
v.push_back(nums[t1]);
v.push_back(nums[t2]);
while(t1<t2&&nums[t1]==nums[t1+1]) t1++; // 第三個數的去重,不需使用unique函數
while(t1<t2&&nums[t2]==nums[t2-1]) t2--; // 第四個數的去重
t1++;
t2--;
ret.push_back(v);
}
else if(sum<target)
t1++;
else
t2--;
}
}
}
return ret;
}
};