题目描述
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出和为目标值的那两个整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
解题思路
使用哈希表(HashMap)存储遍历过的元素及其下标。遍历数组时,检查 target - nums[i] 是否在哈希表中,若存在则找到答案,否则将当前元素存入哈希表。
核心思想:用空间换时间,将查找时间从 O(n) 降到 O(1)。
代码实现
Go
func twoSum(nums []int, target int) []int {
m := make(map[int]int)
for i, num := range nums {
if j, ok := m[target-num]; ok {
return []int{j, i}
}
m[num] = i
}
return nil
}
复杂度分析
- 时间复杂度:O(n),只需遍历一次数组
- 空间复杂度:O(n),哈希表存储最多 n 个元素
