题目:
给定一个整数数组 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
只会存在一个有效答案
进阶:你可以想出一个时间复杂度小于 O(n2) 的算法吗?
这道题最简单的做法应该就是利用双重循环遍历nums,查找出两数相加为target的下标
即:
for(int i =0;i<nums.length;i++){
for(int j=i; j<nums.length; j++){
if(nums[i]+nums[j]==target ){
return new int[]{i,j};
}
}
}
但是这种时间复杂度是O(n^2) 的代码肯定是不过关的,所以必须要将时间压缩在O(n2)之内。也就是说最多只能遍历一次数组,这样的话 公式可以看为 target-nums[n1] = n2
问题主要就在如何存储n2,这时我想到可以使用红黑树进行存储,由于java的hashMap中已经实现了红黑数算法,所以可以直接使用map存储n2,即:
Map<Integer,Integer> map = new HashMap<Integer,Integer>();
int[] sums = new int[2];
for(int i=0;i<nums.length;i++){
int sum = target - nums[i];
if(map.containsKey(sum)){
sums[0] = map.get(sum);
sums[1] = i;
return sums;
}
map.put(nums[i],i);
}
完美实现所需时间在O(n2)之内。