一起养成写作习惯！这是我参与「掘金日新计划 · 4 月更文挑战」的第22天，点击查看活动详情。

Hi, 大家好。我是新人，程序员库里。

今后会按类分享算法题。

今天给大家分享第17道leetcode上数组中使用双指针相关的算法题。

15.三数之和

leetcode地址：leetcode-cn.com/problems/3s…

难度

中等

描述

给你一个包含 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]]

示例2

输入： nums = []
输出： []

示例3

输入： nums = [0]
输出： []

解法

1. 使用滑动窗口的原理
2. 定义变量res，将符合的结果添加到res中，最后返回res
3. 定义变量len，表示数组的长度
4. 先判断当前数组如果是不存在或者数组的长度小于3，则直接返回空数组res
5. 然后将数组以升序的方式进行排序
6. 开始遍历数组
7. 这里有个优化点，遍历的当前元素和上一个元素相等的时候，需要去重，跳过即可
8. 定义左指针left，初始值是当前遍历元素的下一个元素的索引
9. 定义右指针right，初始值指向数组的最后一位
10. 当左指针小于右指针的时候，定义变量sum，表示计算当前遍历的元素、左指针指向的元素、右指针指向的元素之和
11. 判断当sum大于0，因为数组是升序的，为了使得三个数之和正好等于0，需要将右指针向左移动一位
12. 判断当sum小于0，因为数组是升序的，为了使得三个数之和正好等于0，需要将左指针向右移动一位
13. 判断当sum正好等于0 的时候，表示找到了三个元素，将这三个元素添加到变量res中。同时将左指针向右移动一位，将右指针向左移动一位，继续找三数之和等于0 的元素。在这个时候，需要去重，也就是当左指针指向的元素等于左指针下一个元素的时候，将左指针向右移动一位。同时当右指针指向的元素等于右指针下一个元素的时候，将右指针向左移动一位。来达到去重的效果
14. 最后返回res即可

代码

/**
 * @param {number[]} nums
 * @return {number[][]}
 */
var threeSum = function(nums) {
    let res = [];
    const len = nums.length;
    if(nums === null || len < 3) return res;

    nums.sort((a,b) => a-b)
    for(let i =0;i<len;i++){
        if(nums[i] === nums[i-1]) continue;

        let l = i + 1;
        let r = len - 1;

        while ( l < r) {
            const sum = nums[i] + nums[l] + nums[r];
            if(sum === 0){
                res.push([nums[i], nums[l], nums[r]])
                while(l<r && nums[l] === nums[l+1]){
                    l++
                }
                while(l<r && nums[r] === nums[r-1]){
                    r--
                }
                l++
                r--
            }else if(sum < 0 ){
                l++
            }else if(sum > 0){
                r--
            }
        }
    }
    return res;
};