Day 3 | 三数求和

237 阅读1分钟

题目描述:

LeetCode 地址

标签:数组
难度:简单
描述:

给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target  的那 两个 整数,并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

示例1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1]
示例2:
输入:nums = [3,2,4], target = 6
输出:[1,2]
示例3:
输入:nums = [3,3], target = 6
输出:[0,1]
提示:
  • 2 <= nums.length <= 104
  • 109 <= nums[i] <= 109
  • 109 <= target <= 109
  • 只会存在一个有效答案

题解

题目理解

考察点:数组 + 排序 + 双指针 注意:题目要求“不重复的三元组”,需要处理一下元素重复的情况。

解题思路

  1. 核心思路:排序 + 碰撞双指针

代码

/**
 * @param {number[]} nums
 * @return {number[][]}
 */
var threeSum = function(nums) {
    let result = [];
    nums.sort((a, b) => a - b);
    const len = nums.length;

    for(let i = 0; i < len - 2; i++) {
        let l = i + 1, r = len - 1;
        if (i > 0 && nums[i] === nums[i - 1]) {
            continue;
        }
        while(l < r) {
            if (nums[i] + nums[l] + nums[r] < 0) {
                l++;
                while(l < r && nums[l] === nums[l - 1]) {
                    l++;
                }
            } else if (nums[i] + nums[l] + nums[r] > 0) {
                r--;
                while(l < r && nums[r] === nums[r + 1]) {
                    r--;
                }
            } else {
                result.push([nums[i], nums[l], nums[r]]);
                l++;
                r--;
                while(l < r && nums[l] === nums[l - 1]) {
                    l++;
                }
                while(l < r && nums[r] === nums[r + 1]) {
                    r--;
                }
            }
        }
    }
    return result;
};

算法分析

  1. 时间复杂度:O(n^2)
  2. 空间复杂度:O(logn)

image.png

总结

  1. 数组排序方法
  2. 双向指针,碰撞指针算法
    • 从两端向中间迭代数组。一个指针从始端开始,另一个从末端开始;
    • 终止条件是两个指针相遇;
    • 常用于排序数组。