首页 > 其他 > 详细

LeetCode "3Sum"

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

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"

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!