LeetCode: 3Sum

最新推荐文章于 2019-05-07 10:47:06 发布

AIvin24

最新推荐文章于 2019-05-07 10:47:06 发布

阅读量743

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/AIvin24/article/details/38901187

版权

LeetCode 专栏收录该内容

144 篇文章 0 订阅

订阅专栏

思路：先排序，然后遍历数组得到A[i]，设置两个指针指向i后的第一个元素和最后一个元素，分别为j,k，如果A[i] + A[j] + A[k] < 0，j后移，如果大于0，k 左移，如果相等则加入结果向量中。注意排除重复解。

code:

class Solution {
public:
    vector<vector<int> > threeSum(vector<int> &num) {
        sort(num.begin(),num.end());
        vector<vector<int> > ret;
        for(int i = 0;i < num.size();i++){
            if(i > 0 && num[i] == num[i-1])
                continue;
            int j = i + 1, k = num.size() - 1;
            while(j < k){
                if(j > i+1 && num[j] == num[j-1]){
                    j++;
                    continue;
                }
                if(k < num.size() - 2 && num[k] == num[k+1]){
                    k--;
                    continue;
                }
                if(num[i] + num[j] + num[k] > 0)
                    k--;
                else if(num[i] + num[j] + num[k] < 0)
                    j++;
                else{
                    vector<int> t = {num[i],num[j],num[k]};
                    ret.push_back(t);
                    j++;
                }
            }
        }
        return ret;
    }
};