1. 程式人生 > >Leetcode刷題 167. 兩數之和 II

Leetcode刷題 167. 兩數之和 II

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

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

說明:

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

示例:

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

解答:思路類似於兩數之和 1

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        # 用len()方法取得nums列表長度
        n = len(nums)
        # 建立一個空字典
        d = {}
        for x in range(n):
            a = target - nums[x]
            # 字典d中存在nums[x]時
            if nums[x] in d:
                return d[nums[x]], x+1
            # 否則往字典增加鍵/值對
            else:
                d[a] = x+1
                # 邊往字典增加鍵/值對,邊與nums[x]進行對比