leetcode15. 三数之和

题目

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

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

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]

题解

能用哈希表吗？

两数之和

出于之前两数之和，就会往哈希表方面想，两数之和是判断数组中是否有两个数之和等于target，则可以利用哈希表，遍历一次数组，对于每个nums[i]，判断哈希表中有没有target-nums[i]，有的话就找到答案了，没有的话就将nums[i]设置为哈希值，这样依次迭代，就能找到答案

利用哈希表的话，两层for循环就能确定a, b了，剩下的就是找c了，也就是-a-b了，理论上利用哈希表找的话是可以的，但是题目中说到不能包含重复的三元组，但是利用哈希表就会存在元素重复出现的问题，比如，现在a = -1, b = 2, 那就得找c = -1，但是目前哈希表已经有把a = -1这个设置为哈西值了，所以会找到-1，所以就出现了a这个元素出现两次的情况，所以利用哈希表就得去重，但似乎很麻烦（我太菜了，学不会！）

综上，用哈希表不太适合

排序 + 双指针

• 先数组排序
• 两层for循环，确定a, b，再在剩余元素中查找c，
• 找c的过程是从数组最后一个元素遍历，循环至c的值小于或等于-(a + b)，此时因为数组是排序过的，所以大小关系是确定的
• 如果没有找到的话，即b之后的元素都会大于-(a + b) ，证明此时a必不可能会出现三数之后等于0的，因为a存在时，b和c取a后面的元素，都会大于0，所以直接跳出第二层for循环

为什么要排序？

排序的话是为了确定第三个元素，如果不排序的话，就得遍历a，b后面的全部元素，时间复杂度就会上升

复杂度分析

• 时间复杂度

  • 第一层循环确定a的值，需要全部遍历元素，所以复杂度为O(n)
  • 第二层使用了双指针确定两个数的值，每次遍历移动左边的元素一次，移动最右边的元素若干次，均摊下来的话复杂度也为O(n)
  • 两层合起来的时间复杂度是O(N ^ 2)，由于排序的复杂度是O(nlogn)，小于n ^ 2，所以，总的时间复杂度是O(N ^ 2)

• 空间复杂度

  O(logN)，为排序时所消耗的

具体代码

/**
 * @param {number[]} nums
 * @return {number[][]}
 */
var threeSum = function(nums) {
    const len = nums.length
    if(len < 3) return []
    const res = []
    nums.sort((a, b) => a - b)

    for(let i = 0; i < len - 2; i++) {
        if(nums[i] > 0) break
        if(i > 0 && nums[i] == nums[i - 1]) continue
        const target = -nums[i]
        for(let j = i + 1; j < len; j++) {
            if(j != i + 1 && nums[j] == nums[j - 1]) continue
            let k = len - 1
            while(j < k && nums[j] + nums[k] > target) {
                k--
            }
            if(k == j) break  // 表示当前j加上后面任何数都不小于target, 可以直接break,因为j之后的数都不小于j，所以不可能会出现等于target的情况！
            
            // 找到的情况
            if(nums[j] + nums[k] == target) {
                res.push([nums[i], nums[j], nums[k]])
            }
        }
    }

    return res
};