【Leetcode/Python】001-Two Sum

题目

给定一个整数数组和一个目标值，找出数组中和为目标值的两个数。
你可以假设每个输入只对应一种答案，且同样的元素不能被重复利用。

示例:

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

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        a = []
        for i in range(len(nums)-1):
            for j in range(i+1, len(nums)):
                if target == (nums[i] + nums[j]):
                    a = [i,j]
                    break
        return a

存在问题

• 直接循环存在重复计算
• 下标可以使用dict的key和value，用enumerate迭代。

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        dict = {}
        for i, num in enumerate(nums):
            if target - num in dict:
                return [d[target - num], i]
            dict[num] = i