LeetCode.167 两数之和 II - 输入有序数组

719 阅读1分钟

这是我参与8月更文挑战的第7天,活动详情查看:8月更文挑战

题目描述:

167. 两数之和 II - 输入有序数组 - 力扣(LeetCode) (leetcode-cn.com)

给定一个已按照 升序排列  的整数数组 numbers ,请你从数组中找出两个数满足相加之和等于目标数 target

函数应该以长度为 2 的整数数组的形式返回这两个数的下标值。numbers 的下标 从 1 开始计数 ,所以答案数组应当满足 1 <= answer[0] < answer[1] <= numbers.length

你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。

示例一

输入:numbers = [2,7,11,15], target = 9
输出:[1,2]
解释:2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2

示例二

输入: numbers = [2,3,4], target = 6
输出: [1,3]

示例三

输入: numbers = [-1,0], target = -1
输出: [1,2]

提示:

  • $2 <= numbers.length <= 3 * 10^4$
  • -1000 <= numbers[i] <= 1000
  • numbers递增顺序 排列
  • -1000 <= target <= 1000
  • 仅存在一个有效答案

思路分析

大名鼎鼎的 两数之和 中我们已经写过2种解法

暴力枚举哈希表法

这2个解法也可适用于本题

代码就不再重复贴了

优化: 我们再看一下题目发现,标题中明确说明了 有序数组 那么就针对这个特性我们可以得到时间复杂度和空间复杂度更优的解法。

双指针

这解法其实还是看了官解得到的

捋一捋思路吧

使用双指针,一个指针指向值较小的元素,一个指针指向值较大的元素。指向较小元素的指针从头向尾遍历,指向较大元素的指针从尾向头遍历。

如果两个指针指向元素的和 sum == target,那么得到要求的结果; 如果 sum > target,移动较大的元素,使 sum 变小一些; 如果 sum < target,移动较小的元素,使 um 变大一些。

数组中的元素最多遍历一次,时间复杂度为 O(N)O(N)。只使用了两个额外变量,空间复杂度为 O(1)O(1)

AC代码

class Solution {
    fun twoSum(numbers: IntArray, target: Int): IntArray {
        var left = 0
        var right = numbers.lastIndex
        while (true) {
            (numbers[left] + numbers[right]).let {
                if (it > target) {
                    right--
                }
                if (it < target) {
                    left++
                }
                if (it == target) {
                    return intArrayOf(left + 1, right + 1)
                }
            }
        }
    }
}

总结

其实刚看完官解的时候确定有点疑惑,这样简单的移动真的不会漏掉或错过正确答案吗?

看了官解和精选解算是稍微看明白了,希望二刷的时候没有忘记。

参考

两数之和 II - 输入有序数组 - 两数之和 II - 输入有序数组 - 力扣(LeetCode) (leetcode-cn.com)

两数之和 II - 输入有序数组 - 双指针 O(N) 时间复杂度 - 两数之和 II - 输入有序数组 - 力扣(LeetCode) (leetcode-cn.com)

一张图告诉你 O(n) 的双指针解法的本质原理(C++/Java) - 两数之和 II - 输入有序数组 - 力扣(LeetCode) (leetcode-cn.com)