题目描述:
给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 的那 两个 整数,并返回它们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
暴力解法
最开始想到的解法就是暴力解法,使用两轮循环,在第一轮循环中先固定一个数,再第二轮循环中,判断遍历到的数是否与固定数的和为target,如果是,则返回两个下标,如果不是,则继续循环。
class Solution {
public int[] twoSum(int[] nums, int target) {
// 下标数组
int[] loc = new int[2];
for (int first = 0; first < nums.length-1; first ++){
for (int last = first + 1; last < nums.length; last ++){
if (nums[first] + nums[last] == target){
loc[0] = first;
loc[1] = last;
break;
}
}
}
return loc;
}
}
但是,暴力解法的时间复杂度为 O ( n 2 ) O\left( {{n^2}} \right) O(n2),时间消耗为12ms,这是比较差的!
哈希表
毫无疑问,肯定是需要一次遍历的,因此,需要将 O ( n 2 ) O\left( {{n^2}} \right) O(n2)降低为 O ( n ) O\left( {{n}} \right) O(n),为此,使用哈希表来做查找。
哈希表hsahMap中的key为数组元素值,value为数组下标。假设当前遍历到的数为number,则根据target求得另一个数annother = target - number,此时的目的是查看数组中是否存在annother。若存在,则直接返回number和annother两个数对应的下标,若是当前还不存在,则将number和对应的数组下标存入hashMap中,以便下次查找。
class Solution {
public int[] twoSum(int[] nums, int target) {
HashMap<Integer, Integer> hashMap = new HashMap<Integer, Integer>();
int annother = 0;
for (int i = 0; i < nums.length; i ++){
annother = target - nums[i];
// 找到这两个数,直接返回
if (hashMap.containsKey(annother))
return new int[]{i, hashMap.get(annother)};
// 未找到,暂存在hashMap中
else
hashMap.put(nums[i], i);
}
return new int[2];
}
}
时间复杂度为 O ( n ) O\left( {{n}} \right) O(n),时间消耗为0ms,内存消耗为38.7MB。