[LeetCode] 15、三数之和

题目描述

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

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

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[
	[-1, 0, 1],
	[-1, -1, 2]
]

解题思路

要求不重复，就要去重。（去重稍微复杂一丢丢）

• 先对数组进行排序，排序后固定一个数$nums[i]$，再使用左右指针指向$nums[i]$后面部分的两端（转换为easy题目），数字分别为 $nums[L]$和 $nums[R]$，计算三个数的和 $sum$ 判断是否满足为 $0$，满足则添加进结果集。
• 如果 $nums[i]$大于 $0$，则三数之和必然无法等于$0$，结束循环（continue）。
• 如果$nums[i]==nums[i-1]$，则说明该数字重复，会导致结果重复，所以应该跳过。
• 当 $sum==0$ 时，若$nums[L]==nums[L+1]$ 则会导致结果重复，应该跳过，$L++$。
• 当 $sum==0$ 时，若$nums[R]==nums[R-1]$则会导致结果重复，应该跳过，$R--$。
• 时间复杂度$O(n^2)$，不要过于纠结，就是这个方法。（《剑指offer》中有这个方法）

参考代码

常规题目，要想办法将三数之和转换为两数之和问题，时间复杂度$O(n^2)$。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;
        int length = nums.size();
        if(length < 3)
            return res;
        
        sort(nums.begin(), nums.end());  // 排序（为了去重）
        
        for(int i = 0; i <= length-3; i++){
            int low = i + 1;
            int high = length - 1;
            // 如果当前数字大于0，则三数之和一定大于0，所以结束循环
            if(nums[i] > 0)
                break;
            //去重
            if(i > 0 && nums[i] == nums[i-1])
                continue;
            
            while(low < high){
                int sum = nums[i] + nums[low] + nums[high];
                if(sum == 0){
                    res.push_back({nums[i], nums[low], nums[high]});
                    
                    // 去重
                    while(low < high && nums[low] == nums[low+1])
                        low++;
                    while(low < high && nums[high] == nums[high-1])
                        high--;
                    low++;
                    high--;
                    
                }else if(sum < 0){
                    low++;
                }else{
                    high--;
                }
            }
 
        }
        
        return res;
    }
    
};