三数之和--力扣15

这里写目录标题

• 题目
• 思路
• 代码

题目

思路

题目要求三元组不能重复，如果使用哈希表来做，去重很复杂，而且需要额外的空间，我们这里使用双指针法直接针对数组操作。注意题目要求返回的是二维数组！

最重要的是，我们要对数组进行排序，排序后若数组中，第一个元素大于零，则后续不可能存在和等于零的三元组。

其次是对abc去重：

对a去重：当i大于0并且nums[i] == nums[i - 1]时，说明a重复了，直接结束本次循环。i等于0时，是第一次，所以不用去重。为什么i要大于0因为要保证i - 1大于等于0。为什么不是nums[i] == nums[i -+1]？因为nums[i -+1]是left。

对b和c去重：在保证循环体条件right > left情况下，类似于a，判断b的后一个和c的前一个是否有重复的，如果有，b++或者c--。

代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        //返回二维数组 要求a+b+c=0
        vector<vector<int>> res;
        //排序
        sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size(); i++) {
            //如果排序后数组的第一个值大于零，直接返回
            if (nums[i] > 0) {
                return res;
            }
            //对a去重
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }
            //双指针法，寻找b和c
            int left = i + 1;
            int right = nums.size() - 1;
            while (right > left) {
                if (nums[i] + nums[left] + nums[right] > 0) {
                    right--;
                } else if (nums[i] + nums[left] + nums[right] < 0) {
                    left++;
                } else {
                    res.push_back(vector<int>{nums[i], nums[left], nums[right]});
                    //对b和c去重
                    while (right > left && nums[right] == nums[right - 1]) right--;
                    while (right > left && nums[left] == nums[left + 1]) left++;

                    //双指针收缩，继续循环 
                    right--;
                    left++;
                }
            }
        }
        return res;
    }
};