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(n2)
空间复杂度：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;    }};