Input array is sorted(兩數之和 II
阿新 • • 發佈:2018-12-11
題目描述
給定一個已按照升序排列 的有序陣列,找到兩個數使得它們相加之和等於目標數。
函式應該返回這兩個下標值index1 和 index2,其中 index1 必須小於 index2。
說明
- 返回的下標值(index1和index2)不是從零開始的。
- 你可以假設每個輸入只對應唯一的答案,而且你不可以重複使用相同的元素。
示例
輸入: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; }
執行結果
歡迎各位大神指點