leetcode 15. 3Sum

相关问题

1. Two Sum
15. 3Sum
16. 3Sum Closest
18. 4Sum
454. 4Sum II

Discription

Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

Note: The solution set must not contain duplicate triplets.

For example, given array S = [-1, 0, 1, 2, -1, -4],

A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]

思路

双指针。
固定 $a$，在剩余元素中寻找 $b,c$ 使得 $b+c = -a$ , 这样原问题就转化为了 Two Sum。
考虑到可能出现重复元素，首先对所有元素排序，选定 $a$ 之后，使用双指针夹逼的方式确定 $b,c$。通过排序以及合理的选择策略避免重复结果。

注意：排序+选择策略是常用的避免重复结果的方法。

时间复杂度：$O(n^2)$
空间复杂度：$O(1)$

代码

class Solution {
public:
    // 排序+双指针
    vector<vector<int>> threeSum(vector<int>& nums) {
        // 排序+策略处理重复元素
        sort(nums.begin(), nums.end());

        vector<vector<int>> res;

        if (nums.size() < 3)
            return res;

        int i = 0;
        while(i < nums.size() - 2)
        {
            if (i > 0 && nums[i] == nums[i - 1])    // 避免重复
            {
                i++;
                continue;
            }

            int left = i + 1, right = nums.size() - 1;
            while(left < right)
            {
                if (left > i + 1 && nums[left] == nums[left - 1])   // 避免重复
                {
                    left++; 
                    continue;
                }

                if (right < nums.size() - 1 && nums[right] == nums[right + 1])  // 避免重复
                {
                    right--;
                    continue;
                }

                if (nums[i] + nums[left] + nums[right] == 0)
                {
                    res.push_back({nums[i], nums[left], nums[right]});
                    left++;
                    right--;
                    continue;
                }

                if (nums[i] + nums[left] + nums[right] < 0)
                {
                    left++;
                    continue;
                }

                if (nums[i] + nums[left] + nums[right] > 0)
                {
                    right--;
                    continue;
                }
            }

            i++;      
        }

        return res;

    }
};