題目
給定一個已按照升序排列 的有序數組,找到兩個數使得它們相加之和等於目標數。
函數應該返回這兩個下標值 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()) {}