给定一个整数数组 nums
和一个整数目标值 target
,请你在该数组中找出 和为目标值 target
的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9 输出:[0,1] 解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
提示:
2 <= nums.length <= 104
-10^9 <= nums[i] <= 10^9
-10^9 <= target <= 10^9
- 只会存在一个有效答案
就不考虑暴力遍历了,直接考虑进阶时间复杂度小于O(n^2)的算法。
使用哈希表实现,时间复杂度O(n)。
一次遍历,将当前值与目标的差值作为key,当前下标作为value存入hashmap。
在遍历过程中判断,当前值是否已经存在于hashmap中,若已存在,说明已经找到了两个数和为target,返回当前下标及hashmap中对应key的value。
class Solution {public int[] twoSum(int[] nums, int target) {HashMap<Integer,Integer> map = new HashMap<>();for(int i = 0;i < nums.length;i++){if(map.containsKey(nums[i])){return new int[]{map.get(nums[i]),i};}map.put(target - nums[i],i);}return new int[]{0};}
}
算法:
- 创建一个哈希表
map
,其中键是数组中的数字与目标差值,值是该数字的索引。 - 遍历数组
nums
:- 如果
map
中包含nums[i]
,则返回map.get(nums[i])
和i
,表示这两个数字的索引。说明找到了两个数和为target。 - 如果
map
中不包含nums[i]
,则将target - nums[i]
作为键和i
作为值添加到map
中。
- 如果
- 如果遍历数组后没有找到目标和,则返回
[0]
。
**时间复杂度:**O(n),其中 n 是数组 nums
的长度。
**空间复杂度:**O(n),因为哈希表 map
最多存储 n 个键值对。