三数之和,我如何从超时优化到AC的

这个题很多小伙伴都有遇到过,它也是比较高频的面试题,题目如下:

三数之和,我如何从超时优化到AC的_第1张图片

从题目我们可以知道,答案必须符合以下3个条件:

  • 返回的最终结果一定是一个二维数组。
  • 每一个子数组的长度必须是3,并且每个子数组里的所有项加起来,必须等于0。
  • 子数组之间不能重复,必须唯一(其实这个可以不用管)。

超时版本

3数之和嘛,那就3个循环,如下:

var threeSum = function(nums) {
    if (nums.length < 3){
        return [];
    }
    let result = [];
    for (let index = 0; index < nums.length; index++){
        // 当前的值
        let curItem = nums[index];
        // 剩余的值
        let rest = 0 - curItem;
        // 那接下来的问题就是,剩余的数组中,2数之和呗
        for (let nextIndex = index + 1; nextIndex < nums.length; nextIndex++){
            // 最终剩余的值
            let finalNum = rest - nums[nextIndex];
            // 在剩余的数组中查找最后的值
            for (let finalIndex = nextIndex + 1; finalIndex < nums.length; finalIndex++){
                let temp = [];
                if (nums[finalIndex] === finalNum){
                    temp.push([
                        nums[index],
                        nums[nextIndex],
                        nums[finalIndex]
                    ]);
                    result.push(temp);
                }
            }
        }
    }
    return result;
}

上面这3个循环肯定是超时的,所以我们现在要想一下,如何减少一层循环。

AC版本

我们可以先这么想一下,假如有如下数组:

let arr = [-3, -2, -1, 0, 1, 2, 3]

那上面的数组里,每个子项都可以作为子数组的开头,比如 -3作为开头,那么符合条件的子数组可能是:

[-3, 0, 3]

-2作为开头,那么符合条件的子数组就有可能是:

[-2, 0, 2]

那确定了上面的思想后,第一层遍历的含义也就确定了,就是让每个子项都作为子数组的第一项。

接下来,在上一步的基础上,我们可以使用首尾指针的方式来解决这道题目,具体如下:

let left = index + 1;
let right = nums.length - 1;

// 说明此时的 nums[index], nums[left], nums[right]应该push到数组里 
nums[index] + nums[left] + nums[right] === 0

// 如果大于0,说明 right指针应该减小,这样 以后的数组之和 才有可能 === 0
nums[index] + nums[left] + nums[right] > 0

// 如果相加小于0,说明 left指针应该增大,这样 以后的数组之和 才有可能 === 0
nums[index] + nums[left] + nums[right] < 0

特别提醒:

对于上述的3种情况,上述情况的成立需要满足以下规则:
1、index < left < right。
2、nums[index] <= nums[left] <= nums[right]。

为了满足上面的2个规则,我们可以在一开始就对数组进行从小到大的排序。

有了上面的理解,我们的代码就比较好写了:

var threeSum = function(nums) {
    if (nums.length < 3){
        return [];
    }
    
    // 从小到大排序
    nums.sort(
        (a, b) => a - b
    );
    
    let result = [];
    
    // 这里需要解释一下,为什么 index < nums.length - 2 ?
    // 因为我们的子数组长度始终为3,当 index === nums.length - 2时,子数组的长度不可能为3
    for (let index = 0; index < nums.length - 2; index++){
        let left = index + 1;
        let right = nums.length - 1;
        while(left < right){
            let curAllNum = nums[index] + nums[left] + nums[right];
            if (curAllNum === 0){
                // 1、如果满足条件,则加入到数组里
                result.push([
                   nums[index],
                   nums[left],
                   nums[right]
                ]);
                left++;
                right--;
            }
            
            if (curAllNum < 0){
                // 2、说明left应该增大,这样才有可能找到index开头的,并且符合答案的子项
                left++;
            }
            
            if (curAllNum > 0){
                // 3、说明right应该减小,这样才有可能找到index开头的,并且符合答案的子项
                right--;
            }
        }
    }
    
    return result;
}

这一版本,leetcode虽然可以ac,但是无论用时,还是内存消耗情况,它都处于下游。那有没有其他更高效的方案呢?还请各位大佬在评论区里大显身手。

本期的分享到这里就结束了,我们下期再见啦~~

你可能感兴趣的:(javascript,算法)