【LeetCode 算法练习】“两数之和(Two Sum)”问题的 Python3 解答

题目描述

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

示例

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

解题思路

要求的结果是两个索引组成的 List:[aIndex, bIndex],这两个索引对应的值记为 a, b。

遍历列表,假设列表中第1项的值 a 是结果 List 中的一个索引的值,那么用 target 减去 a 后得到的结果 b 就是就是要找的另一个索引的值,列表除掉 a 后的子列表里面,如果 b 存在,问题已解——答案是 a 的索引和 b 的索引组成的列表 [aIndex, bIndex]。

如果不存在,将列表的第二项赋值给 a,继续寻找 b,直到找到为止。

解答

class Solution:
    def twoSum(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """

        i = 0

        for num in nums:
            nextNum = target - num
            j = i + 1
            if nextNum in nums[j:]:
                return [i, nums[j:].index(nextNum) + i + 1]
            i += 1

中文官网的题目地址:两数之和

Python 算法 LeetCode 两数之和 解答 技术 Two Sum
本文更新于