1. 程式人生 > >Python3&刷題之1. 兩數之和

Python3&刷題之1. 兩數之和

1. 兩數之和

給定一個整數陣列 nums 和一個目標值 target,請你在該陣列中找出和為目標值的 兩個 整數。

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

示例:

給定 nums = [2, 7, 11, 15], target = 9

因為 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

 

#two-sum
class Solution(object):

    def twoSum(self,nums,target):
        dict = {}
        for i ,num in enumerate(nums):
            if num in dict:
                return [dict[num],i]
            else:
                dict[target - num] = i

if __name__ == '__main__':
    nums = [2, 7, 11, 15]
    target = 9
    test = Solution().twoSum(nums,target)
    print(test)

 

連結:

https://leetcode-cn.com/problems/two-sum/description/