1. 程式人生 > >領釦(LeetCode)兩數之和II - 輸入有序陣列 個人題解

領釦(LeetCode)兩數之和II - 輸入有序陣列 個人題解

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

函式應該返回這兩個下標值 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 }