题目描述
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
题解
方法一:嵌套for循环
嵌套for循环是最直接的方法,每个元素跟其它元素相加,看结果是不是target,是就返回。
public static int[] findTwoNum1(int[] nums, int target) {
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 null;
}
复杂度分析:
- 时间复杂度:O(n²),每层for循环,会尝试去查找n次,共有两层for循环进行嵌套,所以会尝试查找 n * n次。
- 空间复杂度:O(1)
方法二:两遍哈希表
两遍哈希表的思路也比较好理解:第一遍,先将nums数组中所有的值和下标添加到哈希表中(nums[i] 为key,i 为value);第二遍,检查每个元素所对应的目标元素(target - nums[i])是否存在于表中。
注意,这里要判断一下从哈希表中找到的元素不能是自己。
public static int[] findTwoNum2(int[] nums, int target) {
Map<Integer,Integer> map = new HashMap<Integer, Integer>(nums.length);
for(int i =0; i< nums.length; i++){
map.put(nums[i],i);
}
Integer index;
for(int i =0; i< nums.length; i++){
index = map.get(target - nums[i]);
if(null != index && i != index){
return new int[] {i,index};
}
}
return null;
}
复杂度分析:
- 时间复杂度:O(n),每个for循环的时间复杂度为O(n),两个for循环不嵌套,并且从map中查询的时间复杂度为O(1),所以最终时间复杂度为O(n)。
- 空间复杂度:O(n),所需的额外空间取决于哈希表中存储的元素数量,该表中存储了 n 个元素。
方法三:一遍哈希表
“两遍哈希表” 相较于 “嵌套for循环” 的时间复杂度已经得到了优化,但是,通过分析我们发现,我们完全可以不用进行两次for循环,我们可以在往哈希表中插入数据的同时就进行比较。
public static int[] findTwoNum3(int[] nums, int target) {
Map<Integer,Integer> map = new HashMap<Integer, Integer>(nums.length);
Integer index;
for(int i =0; i< nums.length; i++){
index = map.get(target - nums[i]);
if(null != index){
return new int[] {index,i};
}
map.put(nums[i],i);
}
return null;
}
复杂度分析:
- 时间复杂度:O(n),for循环的时间复杂度为O(n),从map中查询的时间复杂度为O(1),所以最终时间复杂度为O(n)。
- 空间复杂度:O(n),所需的额外空间取决于哈希表中存储的元素数量,该表中存储了 n 个元素。