Leetcode: 4Sum

此间少年_leozp

于 2016-08-31 16:41:16 发布

阅读量304

点赞数

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/u013712847/article/details/52386540

版权

leetcode 专栏收录该内容

36 篇文章 0 订阅

订阅专栏

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int> >answer;
        set<vector<int> >ans;
        int n=nums.size();
        if(n<4){
            return answer;
        }
        sort(nums.begin(),nums.end());
        for(int i=0;i<n;i++){
            for(int j=i+1;j<n;j++){
                int sum=nums[i]+nums[j];
                int flag=target-sum;
                int begin=j+1,end=n-1;
                
                while(begin<end){
                if(nums[begin]+nums[end]==flag){
                    vector<int>Q;
                    Q.push_back(nums[i]);
                    Q.push_back(nums[j]);
                    Q.push_back(nums[begin]);
                    Q.push_back(nums[end]);
                    ans.insert(Q);
                    begin++;
                    end--;
                }
                else if(nums[begin]+nums[end]>flag){
                    end--;
                }
                else{
                    begin++;
                 }
                }
            }
        }
                set<vector<int> >::iterator iter=ans.begin();
                for(;iter!=ans.end();iter++){
                    answer.push_back(*iter);
                }
            
        
        return answer;
    }
};