题目地址()
https://leetcode.cn/problems/two-sum/submissions/
题目描述
给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target 的那 两个 整数，并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
前置知识
- 哈希表
思路
关键点
- 求和转换为求差
- 借助Map结构将数组中每个元素及其索引相互对应
- 以空间换时间，将查找时间从O(N)降低到O(1)
代码
- 语言支持：Java
Java Code:
class Solution {
public int[] twoSum(int[] nums, int target) {
int[] ans = new int[2];
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; ++i) {
map.put(nums[i], i);
}
for (int i = 0; i < nums.length; ++i) {
// 这里注意后面第二条件，一个数只能用一次
if (map.get(target - nums[i]) != null && map.get(target - nums[i]) != i) {
ans[0] = i;
ans[1] = map.get(target - nums[i]);
return ans;
}
}
return null;
}
}
更简洁的写法：
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; ++i) {
if (map.containsKey(target - nums[i])) {
return new int[]{map.get(target - nums[i]), i};
}
map.put(nums[i], i);
}
return null;
}
Go Code:
func twoSum(nums []int, target int) []int {
hashMap := map[int]int{}
for i, v := range nums {
if j, ok := hashMap[target - v]; ok {
return []int{j, i}
}
hashMap[v] = i
}
return nil
}
复杂度分析
令 n 为数组长度。
- 时间复杂度：
- 空间复杂度：
