Leetcode - TwoSum

题目描述:

给定一个整数数组和一个目标值，找出数组中和为目标值的两个数。

你可以假设每个输入只对应一种答案，且同样的元素不能被重复利用。

示例:

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

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

这是我第一次写的, 一开始以为没错, 其实是有错误的, 主要是因为两次遍历有可能跟 target 相同, 返回的 index 就会出错, 而且时间复杂度为O(n^2), 效率太低:

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        if len(nums) == 2:
            return [0, 1]
        for i in nums:
            for j in nums:
                if i + j == target:
                    return [nums.index(i), nums.index(j)]


if __name__ == '__main__':
    nums = [2, 3, 4]
    target = 5
    print(Solution().twoSum(nums, target))

后来借鉴了网上大神的代码, 渐渐有点明白了:

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        dict_temp = dict()
        for index, value in enumerate(nums):
            if targte - value in dict_temp:
                return [dict_temp[target - value], index]
            dict_temp[value] = index

微妙之处就在于 if targte - value in dict_temp, 因为 target = nums[index] + nums[index+1] 成立的话, 那么 if targte - value in dict_temp 肯定是在 dict_temp 的键中.