GATE CS Mock 2018 | 第 38 题介绍

这道题要求我们实现一个函数，该函数接收两个参数：一个整数数组和一个目标值。函数应该返回数组中两个元素的下标，这两个元素的和等于目标值。

例如，给定数组 [3, 5, 8, 2] 和目标值 10，函数应该返回 [0, 2]（因为 3 + 8 = 10）。

解题思路

这道题可以用一种叫做“哈希表”的数据结构来解决。哈希表是一种用于维护“键值对”关系的数据结构，它支持常数时间内进行查找、插入、删除等操作。

我们可以遍历一遍整个数组，对于每个元素，都检查一下是否存在另一个元素，使得它们的和等于目标值。为了快速地找到这个“另一个元素”，我们可以把数组中的所有元素都先放到一个哈希表中，然后在哈希表中查找目标值与当前元素的差值是否存在。

代码实现

以下是 Python 代码的实现：

def two_sum(nums, target):
    hashmap = {}
    for i, num in enumerate(nums):
        complement = target - num
        if complement in hashmap:
            return [hashmap[complement], i]
        hashmap[num] = i

该代码首先创建了一个空哈希表 hashmap，然后遍历了一遍数组 nums。对于每个元素 num，我们计算出目标值与它的差值 complement，然后在哈希表中查找该差值是否存在。如果存在，那么说明我们已经找到了符合条件的两个元素，直接返回它们的下标即可。如果不存在，那么我们就把当前元素 num 和它的下标 i 添加到哈希表中，以备后续查找使用。

总结

这道题是一道比较经典的算法题，涉及到了哈希表等多种基础的数据结构和算法知识。掌握这道题的解法可以帮助我们更好地理解和运用哈希表等数据结构，同时也可以提高我们解决实际问题的能力。