1. 程式人生 > >Input array is sorted(兩數之和 II

Input array is sorted(兩數之和 II

題目描述

給定一個已按照升序排列 的有序陣列,找到兩個數使得它們相加之和等於目標數。

函式應該返回這兩個下標值index1 和 index2,其中 index1 必須小於 index2

說明

  1. 返回的下標值(index1和index2)不是從零開始的。
  2. 你可以假設每個輸入只對應唯一的答案,而且你不可以重複使用相同的元素。

示例 

輸入:numbers = [2, 7, 11, 15], target = 9

輸出:[1, 2] 

解釋:2 與 7 之和等於目標數9。因此 index1 = 1,index2 = 2。

 解題思路

本題同樣可以使用兩數之和I中的雜湊表的方式,但是本題加上了輸入有序這個條件,便可以採用另一種思路。

首先建立左右兩個指標,分別指向陣列的首尾元素,先對右指標進行處理,預設的條件是target = numbers[l] + numbers[r],所以numbers[r] = target - numbers[l],我們對不符合該條件(大於)的元素進行排除,處理之後的右指標指向的元素應該滿足numbers[r] <= target - numbers[l],然後對左右指標之間的元素進行篩選(包括左右指標的元素),當二者之和小於target時,左指標自增1,大於target時,右指標自減1,等於時跳出迴圈,最後按順序返回兩個指標+1。

程式碼 

public int[] twoSum(int[] numbers, int target) {
    int[] result = new int[2];
    int l = 0, r = numbers.length - 1;
    while (r > 0 && numbers[r] > target - numbers[l]) {
        r--;
    }
    while (l < r) {
        if (target == numbers[l] + numbers[r]) break;
        if (target > numbers[l] + numbers[r]) l++;
        else r--;
    }
    result[0] = l + 1;
    result[1] = r + 1;
    return result;
}

執行結果

 

歡迎各位大神指點