美文网首页
Leetcode数组Medium | 18. 四数之和

Leetcode数组Medium | 18. 四数之和

作者: Ivan_Lan | 来源:发表于2018-11-30 20:49 被阅读18次

给定一个包含 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]
]

解答:
排序,两层循环遍历,双指针,转化为两数之和。
注意去重的处理方法:第一个数和第二个数是比较现数和前一个数。第三和第四个数是比较现数和下一个数。

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;  //  双动态指针
                
                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;
    }
};

相关文章

网友评论

      本文标题:Leetcode数组Medium | 18. 四数之和

      本文链接:https://www.haomeiwen.com/subject/gzrkcqtx.html