2018-12-13 LeetCode Q1 两数之和

最新推荐文章于 2024-05-28 21:27:00 发布

原创最新推荐文章于 2024-05-28 21:27:00 发布 · 192 阅读

0 ·

CC 4.0 BY-SA版权

Algorithm 同时被 3 个专栏收录

7 篇文章

订阅专栏

Leetcode

5 篇文章

订阅专栏

Python

5 篇文章

订阅专栏

本文详细解析了一道经典算法题“两数之和”的两种解法：暴力遍历和两遍哈希。通过实例展示了如何在给定的整数数组中找到和为目标值的两个整数，并返回它们的数组下标。

1. 两数之和

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

示例:

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

方法1：暴力遍历
思路是从列表中取一个数 $a_k$ ，而后遍历 $a_k$ 后的每一个数 $a_j$ ，如果有：
$a_k+a_j = target$ ，则返回两数的下标。
在下面的代码中要注意 $i n d e x 2$ 的值是从0开始的，因此要得到在原列表中的真实下标，需要变换成 $i n d e x 1 + i n d e x 2 + 1$ ，用时5800ms。

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        for index1,num1 in enumerate(nums):
        	for index2, num2 in enumerate(nums[index1+1:]):
        		if num1 + num2 == target:
        			return [index1, index1+index2+1]

方法2：两遍哈希
此方法的思想是建立一个 $n u m$ 和 $t a r g e t - n u m$ 一一对应的字典，而后用dict的get方法去尝试寻找结果下标，下面的代码在建表时使用了set作为value，用时52ms。

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        # Hash
        dict1 = {}
        for index,num in enumerate(nums):
            dict1[num] = (index,target-num)
        for index1,num in enumerate(nums):
            index2 = dict1.get(target-num, -1)
            if index2 != -1 and index2[0]!=index1:
                return [index1, index2[0]]