码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode第一刷_3Sum

时间:2014-05-15 01:29:41      阅读:248      评论:0      收藏:0      [点我收藏+]

标签:c++   leetcode   算法   

估计大家都会做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;
    }
};


leetcode第一刷_3Sum,布布扣,bubuko.com

leetcode第一刷_3Sum

标签:c++   leetcode   算法   

原文地址:http://blog.csdn.net/u012792219/article/details/25833195

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!