leetcode的第2题两数相加怎么写?
时间: 2024-11-18 07:17:20 浏览: 20
java-leetcode题解之第2题两数相加.zip
LeetCode 第2题 "两数之和"(Two Sum),是一个经典的查找问题,通常涉及到哈希表数据结构的应用。在 Python 中,你可以使用如下的思路解决:
```python
def twoSum(nums, target):
# 创建一个字典用于存储每个数字及其索引
num_dict = {}
for index, num in enumerate(nums):
complement = target - num
# 如果目标值减去当前数值已经在字典中,说明找到了一对和为目标的数
if complement in num_dict:
return [num_dict[complement], index]
# 否则,将当前数值及其索引添加到字典中
num_dict[num] = index
# 测试函数
nums = [2, 7, 11, 15]
target = 9
print(twoSum(nums, target)) # 输出:[0, 1]
```
这个函数通过遍历输入数组,查找目标值与当前元素之差是否已经存在字典中。如果存在,则找到这对数;否则,将当前元素及其索引存入字典继续搜索。
阅读全文