两数之和 II - 输入有序数组

Posted by zhouqian on Monday, July 25, 2022

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

给你一个下标从 1 开始的整数数组 numbers ,该数组已按 非递减顺序排列 ,请你从数组中找出满足相加之和等于目标数 target 的两个数。如果设这两个数分别是 numbers[index1] 和 numbers[index2] ,则 1 <= index1 < index2 <= numbers.length 。

以长度为 2 的整数数组 [index1, index2] 的形式返回这两个整数的下标 index1 和 index2。

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

你所设计的解决方案必须只使用常量级的额外空间。


方案一:双指针分别指向头尾,通过对比指针指向值的和和 target 移动指针。

func twoSum(numbers []int, target int) []int {
    // 定义双指针指向头尾
    l, r := 0, len(numbers) - 1
    for l < r {
        // 计算指针指向元素和
        switch sum := numbers[l] + numbers[r]; {
            case sum < target: // 如果和小于 target 则需要右移指向较小数的指针,使其和增大
                l++
            case sum > target: // 如果和大于 target 则需要左移指向较大数的指针,使其和减小
                r--
            default: 
                return []int{l+1, r + 1}
        }
    }
    return nil
}
  • 时间复杂度:O(n),因为需要移动双指针来计算结果,而移动总次数不会超过 n,所以时间复杂度为 O(n)。
  • 空间复杂度:O(1),因为只使用了常数级别的变量,所以空间复杂度为 O(1)。