leetcode-數組-簡單-兩數之和 II - 輸入有序數組

題目

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

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

說明:  返回的下標值（index1 和 index2）不是從零開始的。
          你可以假設每個輸入只對應唯一的答案，而且你不可以重複使用相同的元素。

輸入: numbers = [2, 7, 11, 15], target = 9
輸出: [1,2]
解釋: 2 與 7 之和等於目標數 9 。因此 index1 = 1, index2 = 2 。

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/two-sum-ii-input-array-is-sorted
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

解答

方法一：

這個題目之前已經出現過了，唯一的不同就是返回的下標不同，需要進行把下標進行 +1操作。

這裏展示一個頭尾指針往中間移動的辦法。分析如下：

• 升序排列。
• 先計算頭尾指針指向的數據之和。
• 如果和大於target 那麼尾指針向左移動那麼和就會變小。
• 如果和小於target 那麼頭指針向右移動那麼和就會變大。
• 直到找到和爲target

class Solution {
public:
    vector<int> twoSum(vector<int>& numbers, int target) {
        //共同往中間移動
        auto iter = numbers.begin();
        auto riter = numbers.rbegin();
        while (true) {
            if (*iter + *riter == target) {
                break;
            } else if (*iter + *riter > target) {
                riter++;
            } else {
                iter++;
            }
        }
        int index1 = iter - numbers.begin() + 1;
        int index2 = numbers.size() - (riter - numbers.rbegin());
        return vector<int>{index1, index2};
    }
};

這個代碼雖然能跑通測試用例，但是使用 while(true)不太合理，因爲如果不存在不存在target的話，指針會越界。

可以加上一個判斷邏輯。

 while (iter != numbers.end() && riter != numbers.rend()) {}