给定一个整数数组 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]
方法一:暴力枚举
两层循环遍历数组,对每一个i,在第二层循环找找相应的target-i,记录下标即可。时间复杂度O(n^2) 空间复杂度O(1)
方法二:哈希表
使用哈希表进行查找,对数组进行一次遍历,对于每一个i,首先从哈希表中寻找有无对应的target-i,如果有,则返回二者的下标。如果哈希表中没有对应的target-i,则将i加入到哈希表中,当继续遍历数组遍历到target-i时,则会在哈希表中找到已经保存的i。
时间复杂度O(n),空间复杂度O(n)。
import java.util.HashMap;
import java.util.Map;
public class Main {
//两层循环遍历数组,对每一个i,在第二层循环找找相应的target-i,记录下标即可。
//时间复杂度O(n^2) 空间复杂度O(1)
public static int[] twoSum(int[] nums, int target) {
int[] res = new int[2];
for(int i = 0; i < nums.length; i++){
for(int j = i+1; j < nums.length; j++){
if (nums[j] == target - nums[i]){
res[0] = i;
res[1] = j;
return res;
}
}
}
return new int[0];
}
//使用哈希表进行查找,对数组进行一次遍历,对于每一个i,首先从哈希表中寻找有无对应的target-i,如果有,则返回二者的下标。
//如果哈希表中没有对应的target-i,则将i加入到哈希表中,当继续遍历数组遍历到target-i时,则会在哈希表中找到已经保存的i、
public static int[] twoSumHash(int[] nums, int target) {
int[] res = new int[2];
Map<Integer, Integer> map = new HashMap<Integer, Integer>();
for(int i = 0; i < nums.length; i++){
if(map.containsKey(target - nums[i])){
res[0] = map.get(target - nums[i]);
res[1] = i;
return res;
}
map.put(nums[i], i);
}
return new int[0];
}
public static void main(String[] args) {
int[] arr = new int[]{2, 7, 11, 15};
int target = 18;
int[] res = twoSum(arr, target);
int[] res2 = twoSumHash(arr, target);
System.out.println(res[0]+" "+res[1]);
System.out.println(res2[0]+" "+res2[1]);
}
}