lintcode 57. 三数之和

给出一个有n个整数的数组S，在S中找到三个整数a, b, c，找到所有使得a + b + c = 0的三元组。

样例
例1:

输入:[2,7,11,15]
输出:[]
例2:

输入:[-1,0,1,2,-1,-4]
输出:[[-1, 0, 1],[-1, -1, 2]]
注意事项
在三元组(a, b, c)，要求a <= b <= c。

结果不能包含重复的三元组。

思路：先排序，挑选三个数相加判断是否为0，若为0，利用set容器进行去重，加入结果即可

class Solution {
public:
    /**
     * @param numbers: Give an array numbers of n integer
     * @return: Find all unique triplets in the array which gives the sum of zero.
     */
    vector<vector<int>> threeSum(vector<int> &numbers) {
        vector<vector<int>> result;
        sort(numbers.begin(),numbers.end());
        vector<int>tmp(3);
        std::set<vector<int>> judge;
        for (int i = 0; i < numbers.size(); i++) {
            /* code */
            tmp[0]=numbers[i];
            for (int j = i+1; j < numbers.size(); j++) {
                /* code */
                tmp[1]=numbers[j];
                for (int k = j+1; k < numbers.size(); k++) {
                /* code */
                    tmp[2]=numbers[k];
                    if(tmp[0]+tmp[1]+tmp[2]==0&&!judge.count(tmp)) 
                    {
                        judge.insert(tmp);
                        result.push_back(tmp);
                        break;
                    }
                }
            }
        }
        return result;
    }
};