Leetcode第15题. 三数之和

题目描述

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

注意：答案中不可以包含重复的三元组。

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]
解释：
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。
注意，输出的顺序和三元组的顺序并不重要。

示例 2：

输入：nums = [0,1,1]
输出：[]
解释：唯一可能的三元组和不为 0 。

示例 3：

输入：nums = [0,0,0]
输出：[[0,0,0]]
解释：唯一可能的三元组和为 0 。

提示：

• 3 <= nums.length <= 3000
• -100000 <= nums[i] <= 100000

 双指针法

思路：

如图所示：先将数组进行升序排列，在一个for循环内，定义两个指针一个左指针为i+1；一个右指针为nums.size()-1;在循环遍历数组时，找到和为0的三元组。

由于是升序排列 ，所以会有以下现象：

1. 当nums[i] > 0时，就不会出现nums[i] + nums[left] + nums[right] = 0的情况；
2. 当nums[i] + nums[left] + nums[right] > 0 时就说明我们需要减小和的值 ----- 只需要right--即可；
3. 当nums[i] + nums[left] + nums[right] < 0 时就说明我们需要增加和的值 ----- 只需要left++即可；

NOTE:因为题目要求答案中不能出现重复的三元组，所以需要对 i left right 分别去重（防止出现重复）

对于i的去重

if(i > 0 && nums[i] == nums[i-1]) {
                continue;
}

if( nums[i] == nums[i+1]) {
            continue;
}

这两个是不一样的 

对于这个[-1，-1，2] 数组 ，如果使用 nums[i] == nums[i+1] 去重 那么就会直接忽略掉 [-1,-1,2]这个正确的三元组；当i移动到第二个-1时，就会判断 nums[i] == nums[i+1] 直接跳过

而对于i > 0 && nums[i] == nums[i-1] 这个条件来说 就相当先对于i=0去查找符合的答案 再进行去重，避免出现遗漏答案的情况，

查看上面这个数组[-1,-1,3,3] 

如果有答案出现 在第一个-1的时已经获取 ，没有答案出现时 第一个-1也已经循环遍历过了，到了第二个-1也是一样的结果，直接跳过

这与下面的left ， right去重的原理类似，在得到一个正确答案的前提下去重，要不然去除的是答案中的相同元素；

排序

在C++中，sort() 函数用于对数组或容器中的元素进行排序。sort() 函数是标准模板库（STL）中的一部分，定义在 <algorithm> 头文件中

基本用法

sort() 函数可以对给定范围内的所有元素进行排序，默认情况下，它按照升序排列元素。其基本形式如下：

sort(RandomAccessIterator first, RandomAccessIterator last);

• 参数:
  • first: 范围的起始迭代器。
  • last: 范围的结束迭代器（指向最后一个元素之后的位置，符合C++的半开区间规则）。

自定义比较函数

你可以提供自己的比较函数来定义不同的排序规则，比如降序排列或按特定标准排序。比较函数需要定义为第三个参数：

sort(RandomAccessIterator first, RandomAccessIterator last, Compare comp);

• 参数:
  • comp: 一个可调用的比较函数或者函数对象，它接受两个元素作为参数，返回一个 bool 值。当第一个参数应该排在第二个参数之前时，应返回 true。

#include <iostream>
#include <vector>
#include <algorithm>

bool compare(int a, int b) {
    return a > b; // 降序排列
}

int main() {
    std::vector<int> nums = {4, 1, 3, 9, 7};
    sort(nums.begin(), nums.end(), compare); // 使用自定义的比较函数

    for (int num : nums) {
        std::cout << num << " ";
    }
    std::cout << std::endl;

    return 0;
}

完整代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;
        //sort()默认升序排列
        sort(nums.begin() , nums.end());
        for (int i = 0; i < nums.size() ;i++) {
        //如果出现nums[i] > 0 直接返回，不会出现答案了
            if(nums[i] > 0) {
                return result;
            }
         // 错误去重a方法，将会漏掉-1,-1,2 这种情况
            /*
            if (nums[i] == nums[i + 1]) {
                continue;
            }
            */
        //对i去重 先对i=0 进行查找答案
            if(i > 0 && nums[i] == nums[i-1]) {
                continue;
            }
            int left = i+1;
            int right = nums.size() -1;
            while(left < right) {
 // 去重复逻辑如果放在这里，0，0，0 的情况，可能直接导致 right<=left 了，从而漏掉了 0,0,0 这种三元组
                /*
                while (right > left && nums[right] == nums[right - 1]) right--;
                while (right > left && nums[left] == nums[left + 1]) left++;
                */
                if(nums[i] + nums[left] + nums[right] > 0) {
                    right--;
                }
                else if(nums[i] + nums[left] + nums[right] < 0) {
                    left++;
                }
                else {
                //获得答案 存入result二维数组中
                    result.push_back(vector<int>{nums[i],nums[left],nums[right]});
                //对left和right去重
                    while(right > left && nums[left] == nums[left+1]) {
                        left++;
                    }
                     while(right > left && nums[right] == nums[right-1]) {
                        right--;
                    }
                //得到答案要更新right和left
                    right--;
                    left++;
                }
            }
        }
            
        return result;
    }
};

参考文章：代码随想录 (programmercarl.com)