[数据结构与算法]leetcode之两数之和(一)

题目内容

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。

示例 1：
输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。
示例 2：
输入：nums = [3,2,4], target = 6
输出：[1,2]
示例 3：
输入：nums = [3,3], target = 6
输出：[0,1]

解题思路

暴力O(n^2)

两层for循环，最暴力的一种方法。

class Solution:
    def twoSum(self, nums, target):
        res = []
        for i in range(len(nums)):
            for j in range(i + 1, len(nums)):
                if nums[i] + nums[j] == target:
                    res.append(i)
                    res.append(j)
        return res

哈希表(O(n))

暴力太慢了，第二层for其实就是找一个数变成target，这一步可以使用hash表来解决

class Solution2:
    def twoSum(self, nums, target):
        my_dict = {}
        res = []
        for i in range(len(nums)):
            my_dict[nums[i]] = i
        for j in range(len(nums)):
            kk = my_dict.get(target - nums[j])
            if kk:
                res.append(j)
                res.append(kk)
                break
        return res