lintcode 57. 三数之和

对着背影说爱祢 提交于 2020-01-29 01:29:21

给出一个有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;
    }
};
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!