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

LeetCode "3Sum"

时间:2014-08-18 10:45:23      阅读:202      评论:0      收藏:0      [点我收藏+]

标签:style   blog   color   io   for   ar   div   amp   

Now that you‘ve solved 2sum, 3sum can be easily converted as 2sum.

Also please note the while loops for i1 and i2: they are to make sure no duplicates.

class Solution {
public:
    vector<vector<int> > threeSum(vector<int> &num) {
        vector<vector<int> > ret;
        if (num.size() < 3) return ret;

        std::sort(num.begin(), num.end());

        for (int i0 = 0; i0 < num.size() - 2; i0++)
        {
            if (i0 > 0 && num[i0] == num[i0 - 1]) continue;

            int tgt = -num[i0];
            int i1 = i0 + 1, i2 = num.size() - 1;

            while (i1 < i2)
            {
                int curr = num[i1] + num[i2];
                if (curr == tgt)
                {
                    vector<int> rt;
                    rt.push_back(num[i0]);
                    rt.push_back(num[i1]);
                    rt.push_back(num[i2]);
                    ret.push_back(rt);
                    while (++i1, num[i1] == num[i1-1]);
                }
                else if (curr < tgt)
                {
                    while (++i1, num[i1] == num[i1-1]);
                }
                else if (curr > tgt)
                {
                    while (--i2, num[i2] == num[i2 + 1]);
                }
            }
        }

        return ret;
    }
};

LeetCode "3Sum",布布扣,bubuko.com

LeetCode "3Sum"

标签:style   blog   color   io   for   ar   div   amp   

原文地址:http://www.cnblogs.com/tonix/p/3918887.html

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