资质|假设和结论|问题4

简介

在编写代码时，有时需要根据某些特定的条件来执行不同的操作，这就需要用到假设和结论。假设通常由if-else语句实现，而结论则通过return语句返回相应的结果。

在这个主题下，我们将探讨如何使用假设和结论来解决问题4。

问题4

给定一个整数数组，找到其中两个数使它们相加得到一个特定的目标数。

例如，如果数组为[2, 5, 7, 11, 15]，目标数为9，则答案为[2, 7]或[5, 7]。

解决方案

我们可以使用一个字典来存储已经遍历过的数字，字典的键是数字本身，值是数字在列表中的索引。在遍历列表时，我们检查目标数与当前数字的差值是否已经在字典中出现过。如果是，就返回这两个数在数组中的索引，即为所求。如果不是，则把当前数字和它在数组中的索引添加到字典里，继续遍历。如果没有找到符合条件的两个数，则返回空列表。

这个算法的时间复杂度为$O(n)$，因为我们只需要遍历整个数组一次。

class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        dic = {}
        for i, num in enumerate(nums):
            if target - num in dic:
                return [dic[target-num], i]
            dic[num] = i
        return []

结论

通过这个例子，我们学习了如何使用假设和结论来解决问题。在这个算法中，我们假设目标数与当前数字的差值已经在字典中出现过，如果在字典中存在这样的键，则返回相应的值。如果不在字典中，则将当前数字与它的索引添加到字典中。最终的结论是，如果找到(符合条件的)数字对，则返回它们在数组中的索引，否则返回空列表。