LintCode 57 三数之和

题目:solveNQueens

---

要求:

给出一个有n个整数的数组S，在S中找到三个整数a, b, c，找到所有使得a + b + c = 0的三元组。
注意事项
在三元组(a, b, c)，要求a <= b <= c。
结果不能包含重复的三元组。

样例:

如S = {-1 0 1 2 -1 -4}, 你需要返回的三元组集合的是：
(-1, 0, 1)
(-1, -1, 2)

算法要求:

无

解题思路:

直接三重循环求解，用set去除重复的组合。

算法如下:

    vector<vector<int> > threeSum(vector<int> &nums) {
        // write your code here
        vector<vector<int> > vec;
        set<vector<int> > s;
        sort(nums.begin(), nums.end());
        int size = nums.size();
        for (int i = 0; i < size && nums[i] <= 0; i++) {
            for (int j = i + 1; j < size; j++) {
                for (int k = j + 1; k < size; k++) {
                    if (nums[i] + nums[j] + nums[k] == 0) {
                        vector<int> tempVec;
                        tempVec.push_back(nums[i]);
                        tempVec.push_back(nums[j]);
                        tempVec.push_back(nums[k]);
                        s.insert(tempVec);
                    }
                }
            }
        }
        set<vector<int> >::iterator it;
        for (it = s.begin(); it != s.end(); it++) {
            vec.push_back(*it);
        }
        return vec;
    }