Leetcode数组Medium | 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;
    }
};
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 专业考题类型管理运行工作负责人一般作业考题内容选项A选项B选项C选项D选项E选项F正确答案 变电单选GYSZ本规程...
    小白兔去钓鱼阅读 12,943评论 0 13
  • <center>#1 Two Sum</center> link Description:Given an arr...
    铛铛铛clark阅读 6,600评论 0 3
  • 【1】7,9,-1,5,( ) A、4;B、2;C、-1;D、-3 分析:选D,7+9=16;9+(-1)=8;(...
    Alex_bingo阅读 19,398评论 1 19
  • 算法思想贪心思想双指针排序快速选择堆排序桶排序荷兰国旗问题二分查找搜索BFSDFSBacktracking分治动态...
    第六象限阅读 8,777评论 0 0
  • 自然选择还在进行。 这依赖于极强大、稳固的时间机制。宇宙的膨胀速度会不会被生命体激增的强占侵略性越过头?这个毁灭性...
    藏巳阅读 1,794评论 0 0