leetcode第一刷_3Sum

最新推荐文章于 2024-09-30 07:31:33 发布

Mr_Giraffe

最新推荐文章于 2024-09-30 07:31:33 发布

阅读量656

点赞数

分类专栏：算法 leetcode 文章标签： c++ leetcode 算法

本文链接：https://blog.csdn.net/u012792219/article/details/25833195

版权

算法同时被 2 个专栏收录

158 篇文章 0 订阅

订阅专栏

leetcode

143 篇文章 0 订阅

订阅专栏

估计大家都会做twoSum，一头一尾两个指针，跟据和的大小移动就行了。

3sum能不能用相同的方法呢，我尝试用暴力做，居然过了。思路是先把数组排个序，让相同数字的都靠在一起，然后固定一个数，其他两个数就按照twosum的那一套来，只不过计算sum的时候多算了一个数而已。需要注意一个问题，靠在一起一样的数，只能在第一次遇到它的时候用，更准确一点说，每个相同的数，只有一次作为i或j或k的机会，而且不用担心是不是会漏掉情况，如果数组是111，要求3，这不都用了吗，是的，但是是i,j,k各用了他们一次。表达的不是很好，看代码应该就明白了。这个解法太暴力，不优雅。

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