leetcode——1. 兩數之和

題目描述

給定一個整數數組 nums 和一個目標值 target，請你在該數組中找出和爲目標值的那 兩個 整數，並返回他們的數組下標。

你可以假設每種輸入只會對應一個答案。但是，你不能重複利用這個數組中同樣的元素。

示例:

給定 nums = [2, 7, 11, 15], target = 9
因爲 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

解題思路

• 使用字典保存元素與索引
• 兩次遍歷，該題必定有一個答案
• 根據元素值，直接返回索引，省去查詢過程

代碼

class Solution(object):
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        dict = {}
        for ind,num in enumerate(nums):
            dict[num] = ind
        for i,num in enumerate(nums):
            j = dict.get(target - num)
            if j is not None and i != j:
                return [i,j]

如有錯誤，請批評指正！