領釦(LeetCode)兩數之和II - 輸入有序陣列 個人題解
阿新 • • 發佈:2018-11-22
給定一個已按照升序排列 的有序陣列,找到兩個數使得它們相加之和等於目標數。
函式應該返回這兩個下標值 index1 和 index2,其中 index1 必須小於 index2。
說明:
- 返回的下標值(index1 和 index2)不是從零開始的。
- 你可以假設每個輸入只對應唯一的答案,而且你不可以重複使用相同的元素。
示例:
輸入: numbers = [2, 7, 11, 15], target = 9 輸出: [1,2] 解釋: 2 與 7 之和等於目標數 9 。因此 index1 = 1, index2 = 2 。
這道題,第一想法就是暴力解法,從左邊遍歷到右邊,找到剛好能達到Target值的兩個數,然後返回。演算法複雜度O(N^2),所以執行速度也不盡人意
程式碼如下:
class Solution { public int[] twoSum(int[] numbers, int target) { int[] ans = new int[2]; int num1 = 0; int num2 = 0; for (int i = 0; i < numbers.length; i++) { num1 = numbers[i]; for (int j = 0; j < numbers.length; j++) { num2= numbers[j]; if (num1 + num2 == target) { ans[0] = i+1; ans[1] = j+1; return ans; } } } return ans; } }
然後參考了評論,發現了二分做法。
因為是有序陣列,所以設定兩個指示,分別指向最左邊和最右邊,也就是陣列的最大值和最小值,兩個值相加,如果大於target,說明加的最大值太大了,指向最右邊的指示就往左邊移動一位,反之,如果小於target,說明加的最小值太小了,指向最左邊的指示就往右邊移動一位,直到兩個數的相加等於target。不得不說,這種方法雖然之前做題的時候遇到過,但是實際應用時就是不懂用。還是要認真學習啊!
程式碼如下:
1 class Solution { 2 public int[] twoSum(int[] numbers, int target) { 3 int[] ans = new int[2]; 4 int left=0 , right=numbers.length-1; 5 while(left<right) 6 { 7 int sum=numbers[left]+numbers[right]; 8 if(sum<target) 9 left++; 10 else if(sum>target) 11 right--; 12 else { 13 ans=new int[] {left+1,right+1}; 14 break; 15 } 16 } 17 return ans; 18 } 19 }