LeetCode 15 3Sum

LeetCode 15 3Sum

思路

题目要求a+b+c = 0，可以将等式变换为b+c = -a。将此题转换为LeetCode 1 Two Sum。为了去除重复三元组，需要其他操作，但是，这么做会超时。

为了去除重复三元组，可以考虑先将数组排序。在遍历的过程中，跳过已经使用过的数字。外层循环设置-a: target = -nums[i]，内层循环从两边往中间遍历找到b: nums[left]和c: nums[right]。

排序后，a < b < c, b+c = -a。如果-a < 0，则 b + c < 0。但是0 < a < b < c，有b + c > 0。因此，当-a < 0时，直接break。

代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> rs;
        sort(nums.begin(), nums.end());
        
        for (int i = 0; i < nums.size(); ++i) {
            int target = -nums[i];
            if (target < 0) break;
            int left = i + 1, right = nums.size() - 1;
            while (left < right) {
                if (nums[left] + nums[right] < target) ++left;
                else if (nums[left] + nums[right] > target) --right;
                else {
                    rs.emplace_back(vector<int>{nums[i], nums[left], nums[right]});
                    while (left < right && nums[left+1] == nums[left]) ++left;
                    while (left < right && nums[right-1] == nums[right]) --right;
                    ++left;
                    --right;
                }
            }
            
            while (i+1 < nums.size() && nums[i+1] == nums[i]) ++i;
        }
        
        return rs;
    }
};