[Leetcode] 15 - 3Sum

来源:互联网 发布:淘宝花呗分期额度不够 编辑:程序博客网 时间:2024/05/22 01:13

原题链接:https://oj.leetcode.com/problems/3sum/


经典3sum题,不过要注意这里的去重,去重是给原来这道题的基本形式加的难点。

1. 数组如果大小小于3,那就没有任何结果。

2. 将数组排序

3. 然后循环数组,每次都选取i所在的当前值为第一个结果,然后选取left = i + 1和right = size - 1。则问题变为num[left] + num[right]是否等于-num[i]。

4. 每次选取第一个元素的时候,要将其与之前的第一个元素相比,如果相同,则说明如果继续计算,此轮的计算结果已经被包含在上轮的计算结果之内,所以跳至下一轮。

5. 当left和right满足条件时候,我们需要将left和right移至下一个位置,这里需要将left之后和right之前的重复元素全部跳过,否则将会出现重复答案,则不符合题意。


class Solution {public:    vector<vector<int> > threeSum(vector<int> &num) {        vector<vector<int> > res;                if (num.size() < 3) {            return res;        }        sort(num.begin(), num.end());        int prev = INT_MIN;        for (int i = 0; i < num.size() - 2; ++i) {            if (num[i] == prev) {                prev = num[i];                continue;            } else {                prev = num[i];            }                        int target = -num[i];            int left = i + 1;            int right = num.size() - 1;            while (left < right) {                int curSum = num[left] + num[right];                if (curSum == target) {                    vector<int> curRes;                    curRes.push_back(num[i]);                    curRes.push_back(num[left]);                    curRes.push_back(num[right]);                    res.push_back(curRes);                    while (num[left] == num[left + 1] && left + 1 < num.size()) ++left;                    while (num[right] == num[right - 1] && right - 1 >= 0) --right;                    ++left;                    --right;                } else if (curSum < target) {                    ++left;                } else {                    --right;                }            }        }                return res;    }};


0 0