LeetCode—— 兩數之和 II - 輸入有序數組

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

函數應該返回這兩個下標值 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 []

發佈了28 篇原創文章 · 獲贊 13 · 訪問量 4274
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章