LintCode 57 三数之和

来源:互联网 发布:java客户管理系统源码 编辑:程序博客网 时间:2024/06/13 12:47

题目: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;    }
0 0