码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode#18. 4Sum

时间:2018-09-28 22:13:28      阅读:227      评论:0      收藏:0      [点我收藏+]

标签:包含   vector   targe   int   end   注意   遍历   ==   相等   

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


//n sum的题看来上瘾了,看来要写个n可变的 sum的dfs版本。
class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        if(nums.size()<4) return {};
        //wow!!!
        //先排序,再遍历,对于每个元素,查找剩余元素里是不是存在3个数之和满足target1
        //内循环,看剩余元素里是不是存在2个数之和满足target2
        //内循环,看剩余元素里是不是存在1个数之和满足target3
        //这样是不是可行,先看一看target=10,[12345], 1,9
        sort(nums.begin(),nums.end());
        vector<vector<int>> result;
        for(int i=0;i!=nums.size();++i)//nums.size()-3
        {
            if(i-1>0&&nums[i]==nums[i-1]) continue;//j > i + 1
            for(int j=i+1;j!=nums.size();++j)
            {
                int m=j+1;
                int n=nums.size()-1;
                while(m<n)
                {
                    auto sum=nums[i]+nums[j]+nums[m]+nums[n];
                    if(sum==target)
                        result.push_back({nums[i],nums[j],nums[m],nums[n]});
                    if(sum<target) m++;
                    else n--;
                }
            }
        }
        sort(result.begin(),result.end());
        result.erase(unique(result.begin(),result.end()),result.end());
        return result;
    }
};

 

leetcode#18. 4Sum

标签:包含   vector   targe   int   end   注意   遍历   ==   相等   

原文地址:https://www.cnblogs.com/lsaejn/p/9720910.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!