給定一個已按照升序排列 的有序數組,找到兩個數使得它們相加之和等於目標數。
函數應該返回這兩個下標值 index1 和 index2,其中 index1 必須小於 index2。
說明:
- 返回的下標值(index1 和 index2)不是從零開始的。
- 你可以假設每個輸入只對應唯一的答案,而且你不可以重複使用相同的元素。
示例:
輸入: numbers = [2, 7, 11, 15], target = 9
輸出: [1,2]
解釋: 2 與 7 之和等於目標數 9 。因此 index1 = 1, index2 = 2 。
我的代碼
class Solution {
public:
vector<int> twoSum(vector<int>& numbers, int target) {
int i,j;
int index1,index2;
vector<int> a;
for(i=0;i<numbers.size();i++)
{
int tmp = target - numbers[i];
for(j=i+1;j<numbers.size();j++)
{
if(numbers[j] > tmp) break;
else if(numbers[j]==tmp)
{
index1 = i+1;
index2 = j+1;
}
}
}
a.push_back(index1);
a.push_back(index2);
return a;
}
};
大佬的代碼
方法一:使用二分搜索法,時間複雜度爲O(nlogn)
:
class Solution:
def _BinarySearch(self, numbers, target):
l = 0
r = len(numbers) - 1
while l <= r:
mid = (l + r) // 2
if target == numbers[mid]:
return mid
elif target < numbers[mid]:
r = mid - 1
else:
l = mid + 1
return -1
def twoSum(self, numbers, target):
"""
:type numbers: List[int]
:type target: int
:rtype: List[int]
"""
nums_len = len(numbers)
for i in range(nums_len):
dif = target - numbers[i]
dif_index = self._BinarySearch(numbers[i + 1:], dif)
if dif_index != -1:
return [i + 1, dif_index + i + 2]
return []
方法二:
首先判斷首尾兩項的和是不是target,如果比target小,那麼我們左邊+1位置的數(比左邊位置的數大)再和右相相加,繼續判斷。如果比target大,那麼我們右邊-1位置的數(比右邊位置的數小)再和左相相加,繼續判斷。我們通過這樣不斷放縮的過程,就可以在O(n)的時間複雜度內找到對應的座標位置。(這和快速排序的思路很相似),這時的時間複雜度爲O(n)
。
class Solution:
def twoSum(self, numbers, target):
"""
:type numbers: List[int]
:type target: int
:rtype: List[int]
"""
l = 0
r = len(numbers) - 1
while l < r:
if numbers[l] + numbers[r] == target:
return [l + 1, r + 1]
elif numbers[l] + numbers[r] < target:
l += 1
else:
r -= 1
return []