给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
示例 1:
输入:nums = [2,7,11,15], target = 9
输出:[0,1]
解释:因为 nums[0] + nums[1] == 9 ,返回 [0, 1] 。
示例 2:
输入:nums = [3,2,4], target = 6
输出:[1,2]
示例 3:
输入:nums = [3,3], target = 6
输出:[0,1]
提示:
- 2 <= nums.length <= 104
- 109 <= nums[i] <= 109
- 109 <= target <= 109
- 只会存在一个有效答案
暴力枚举不再赘述。
for(int i=0;i<nums.length-1;i++)
for(int j=i+1;j<nums.length;j++)
if(nums[i]+nums[j]==target)
return new int[] {i,j};
return new int[] {};
今天学习到了新的一种方式–查找表法
- 在遍历的同时,记录一些信息,以省去一层循环,这是“以空间换时间"的想法
- 需要记录已经遍历过的数值和它所对应的下标,可以借助查找表实现
- 查找表有两个常用的实现:
- 哈希表
- 平衡二叉搜索树
同时我们不需要维护查找表中的顺序,所以这道题使用哈希表
下面我们来看看这个算法是如何运行的
第一步 6之前没有元素与之对应,将其与对应下标直接存入表中。
第二步 8-3=5,5不在表中,将3和其下标存入哈希表中
第三步 按顺序检索直到遇到某个数与其对应的值在哈希表中已经存储过 ,将两个值的下标返回即可。
下面是代码
int len = nums.length;
// 初始化哈希表时尽量定义好容量,避免扩容时带来的性能消耗
HashMap<Integer, Integer> hashMap = new HashMap<>(len - 1);
hashMap.put(nums[0],0);
for (int i = 1; i < len; i++) {
// 每次都检查对应数在不在表中
if (hashMap.containsKey(target - nums[i])){
return new int[] {i, hashMap.get(target - nums[i])};
}
hashMap.put(nums[i], i);
}
throw new IllegalArgumentException("No two sum solution");
算法结束,下面我们来分析复杂度
-时间复杂度:O(n),这里n为数组的长度。
-空间复杂度:O(n),哈希表里最多需要存n-1个键值对。