前端算法(中等难度)——三数之和

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

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

给定数组 nums = [-1, 0, 1, 2, -1, -4],

满足要求的三元组集合为:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum

思路:先选定第一个数,然后再把问题转换为两数之和问题求解!

var threeSum = function(nums) {
    let len=nums.length;
    if(len<3) return []
    // 如果顺序不一样,数值一样也被认为是重复,参考例题两个[0,1,-1]只有一个可以
    nums.sort((a,b)=> a-b);
    // 固定左边的值
    let arr=[];// 遍历的自增在最后面while实现
    for(let i=0;i<len-2;){
        let ll=i+1;// 第二个数
        let r=len-1;
        if(nums[i]>0) break;  // 如果最左边大于0则肯定不行
        while(ll<r){
            let sum=nums[i]+nums[ll]+nums[r];
            if(sum==0){
                arr.push([nums[i],nums[ll],nums[r]]);
                ll++;
                r--;
                // 去重,和上一个比较
                while(nums[r]==nums[r+1]) r--;
                while(nums[ll]==nums[ll-1]) ll++;
            }else if(sum>0){
                r--;
            }else{
                ll++;
            }
        }
        // 比较最左边是否相等,相等则不要
        while(nums[i]==nums[++i]){}
    }
    return arr;
};

你可能感兴趣的:(数据结构与算法,前端)