重刷leetcode-day5-[18. 四数之和]

50 阅读1分钟

【题目】 18. 四数之和

【题目考察】:双指针

【目标复杂度】: O(n^3)

【解法】:

/**
 * @param {number[]} nums
 * @param {number} target
 * @return {number[][]}
 */
var fourSum = function(nums, target) {
     // 定义res存储结果数组
    const res = [];
    // 升序排序
    nums.sort((a, b )=> a - b);
    if (
        // 空
        !nums ||
        // // 数组长度小于4
        nums.length < 4
    ) {
         // 直接返回空数组
        return res;
    }
    
    for (let i = 0; i < nums.length; i++) {

        // 原则:每个循环内都不要和前面用过的数字重复,比如[1,1,2,3,4,5], 多次i循环,两次i都是1,也就是第一位都是1,那么肯定是重复的,因为第一个尝试1的时候已经把第一位是1的所有情况都走了一遍,所以i循环哪去i>0开始去重,因为i是从0开始的,j重i+1开始,因为每一轮都是从i+1开始的
        if (i > 0 && nums[i] === nums[i - 1]) {
            continue;
        }
        
        for (let j = i + 1; j < nums.length; j++) {
            if ((j > i + 1) && nums[j] === nums[j - 1]) {
                continue;
            }

             // 定义左边指针
            let left = j + 1;
             // 定义右指针
            let right = nums.length - 1;
            // 进行while循环判断leftright 条件是left 小于 right 由两边向中间夹,left肯定是比right小的,相等或者大了的就结束
            while(left < right) {
                // 取四个数之和
                const sum = nums[i] + nums[j] + nums[left] + nums[right];
                if (sum === target) {
                    // 如果等于0,那么就把对应的数字放到结果数组中
                    res.push([nums[i], nums[j], nums[left], nums[right]]);
                    // 接着对左边去重
                    while(nums[left] === nums[left + 1]) {
                        left++;
                    }
                    // 对右边去重
                    while(nums[right] === nums[right - 1]) {
                        right--;
                    }
                    // 去重后其实上面哪个数字还是之前那个一样的,所以需要左边右边都移动一位
                    left++;
                    right--;
                } else if (sum < target) {
                    // 如果和小于0,因为是升序排序的,那么表示左边太小了,需要往右移动一点,和就会大一点
                    left++;
                } else {
                    // 如果和大于0,因为是升序排序的,那么表示右边太大了,需要往左移动一点,和就会小一点
                    right--;
                }
            }
        }
    }
    return res;
};