Python LeetCode(1.两数之和)

Python LeetCode(1.两数之和)

题目：给定一个整数数组 nums 和一个目标值 target，请你在该数组中找出和为目标值的那 两个 整数，并返回他们的数组下标。

你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例:

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

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

Solution 1：(遍历列表，对于当前的数字，我们需要找的另外一个数字为target-current_num,即在除了当前数字之外的新列表中寻找另外一个数字是否存在，返回两数的index)

class Solution_1():
    def twoSum(self, nums, target):
        '''
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        '''
        for i in nums:
            j = target - i
            start_index = nums.index(i)
            next_index = start_index + 1
            temp_nums = nums[next_index:]
            if j in temp_nums:
                return [start_index, next_index + temp_nums.index(j)]

solution_1 = Solution_1()
print(solution_1.twoSum([2, 7, 1, 9], 11))

[0, 3]

Solution 2:（新建一个字典，遍历列表，判断target-current_num是否在字典中，如果不在，就把当前数字和其index存入字典；若在，则返回字典中在的数字的index以及current_num的index）

class Solution_2():
    def twoSum(self, nums, target):
        '''
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        '''
        dict_nums = {}
        for i, num in enumerate(nums):
            if target - num not in dict_nums:
                dict_nums[num] = i
            else:
                return [dict_nums[target - num], i]

solution_2 = Solution_2()
print(solution_2.twoSum([10, 12, 4, 8], 18))

[0, 3]