LeetCode——3sum

题目描述:

给出一个有n个元素的数组S，S中是否有元素a,b,c满足a+b+c=0？找出数组S中所有满足条件的三元组。
注意：
三元组（a、b、c、d）中的元素必须按非降序排列。（即a≤b≤c）
解集中不能包含重复的三元组。
例如，给定的数组 S = {-1 0 1 2 -1 -4},↵↵ 解集为：↵ (-1, 0, 1)↵ (-1, -1, 2)

解题思路：
由于最终输出要求元素必须按照非降序排列，所以一开始就可以先给数组排序。对于解集中不能包含重复的三元组，可以使用数组的find函数寻找，若找不到相同的，则保存。

class Solution {
public:
    vector<vector<int> > threeSum(vector<int> &num) {
        sort(num.begin(),num.end());
        vector<vector<int>>output;
        if(num.size()<=2)return output;
        if(num.size()==3&&(num[0]+num[1]+num[2]==0)){
            output.push_back(num);
            return output;
        }
        for(int i=0;i<num.size();i++){
            //vector<int>tmp;
            int sum=0;
            int sum1=sum-num[i];
            for(int j=i+1;j<num.size();j++){
                vector<int>tmp;
                int sum2=sum1-num[j];
                int t=j+1;
                while(t<num.size()){
                    if(sum2==num[t]){
                        tmp.push_back(num[i]);
                        tmp.push_back(num[j]);
                        tmp.push_back(num[t]);
                        if(find(output.begin(),output.end(),tmp)==output.end()){
                            output.push_back(tmp);
                        }
                        break;
                    }
                    t++;
                }
                
            }
        }
        return output;
    }
};

“代码中每一个变量都要想清楚其含义及作用，考虑好每一句代码应该放的位置！”

【run】