作者:手机用户2502880237 | 来源:互联网 | 2023-06-12 18:41
篇首语:本文由编程笔记#小编为大家整理,主要介绍了两数之和 II - 输入有序数组 --二分查找相关的知识,希望对你有一定的参考价值。 题目给定一个已按照升序排列的有序数组,找到两个数使得它们相加之和
篇首语:本文由编程笔记#小编为大家整理,主要介绍了两数之和 II - 输入有序数组 --二分查找相关的知识,希望对你有一定的参考价值。
题目
给定一个已按照升序排列的有序数组,找到两个数使得它们相加之和等于目标数。
函数应该返回这两个下标值 index1 和 index2,其中 index1 必须小于 index2。
说明:
- 返回的下标值(index1 和 index2)不是从零开始的。
- 你可以假设每个输入只对应唯一的答案,而且你不可以重复使用相同的元素。
示例:
输入: numbers = [2, 7, 11, 15], target = 9
输出: [1,2]
解释: 2 与 7 之和等于目标数 9 。因此 index1 = 1, index2 = 2 。
思想
使用双指针,一个指针指向值较小的元素,一个指针指向值较大的元素。指向较小元素的指针从头向尾遍历,指向较大元素的指针从尾向头遍历。
- 如果两个指针指向元素的和sum == target ,那么得到要求的结果;
- 如果sum > target, 我们移动较大的元素, 使sum变小一些;
- 如果sum
数组中的元素最多遍历一次,时间复杂度为 O(N)。只使用了两个额外变量,空间复杂度为 O(1)。
代码
func twoSum(_ numbers: [Int], _ target: Int) -> [Int] {
if numbers.count <= 0 {return []}
var low = 0
var high = numbers.count - 1
var resultArr = [Int]()
while low < high {
let sum = numbers[low] + numbers[high]
if sum == target {
//题目要求是第几个数据,但是数组下标为0开始
//numbers = [2, 7, 11, 15], target = 9 ,题目要求输出[1,2],而对应的数组为[0,1]
//所以low + 1,high + 1
resultArr.append(low + 1)
resultArr.append(high + 1)
return resultArr
} else if sum < target {
low += 1
} else {
high -= 1
}
}
return []
}
结果