[Leetcode]_15 3Sum

/**
 *  Index: 15
 *  Title: 3Sum
 *  Author: ltree98
 **/


暴力肯定TLE,不用多想。

先对所有数据进行排序,
然后从第一个数开始,取一个数,计算这个数后面是否有两个数加这个数等于0。
最关键就是:

  • 排序(此方法的基点)
  • 确定一组答案后过滤前后同值的数(过滤同样的答案, 且不影响其他不同的答案)


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

        std::sort(nums.begin(), nums.end());
        vector<vector<int>> ans;

        for(int i = 0; i < nums.size(); )   {
            int l = i+1, h = nums.size()-1, s = -nums[i];
            while(l < h)    {
                if(nums[l] + nums[h] == s)  {
                    ans.push_back({nums[i], nums[l], nums[h]});
                    l++, h--;           
                    while(l < h && nums[l] == nums[l-1])
                        l++;
                    while(l < h && nums[h] == nums[h+1])
                        h--;
                }
                else if(nums[l] + nums[h] < s)
                    l++;
                else
                    h--;
            }

            i++;
            while(i < nums.size() && nums[i] == nums[i-1])
                i++;
        }

        return ans;
    }
};
发布了472 篇原创文章 · 获赞 387 · 访问量 78万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览