题目
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例1:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
解法
第一种解法通过暴力,固定i,移动j扫描一遍数组,每次判断是否条件nums[i]+nums[j] == target
,如果满足则保存,然后再i+1固定,j从头到尾再扫描一遍,以此类推,时间复杂度为O(n^2)。
第二种解法,使用空间换时间的思想,nums[i]作为key,i作为value保存到map,每次扫描判断map中是否有符合map.containsKey(target-nums[i])
的值,如果有保存,时间复杂度为O(n),空间复杂度为O(n)。
代码如下所示
public int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> posMap = new HashMap<>(nums.length);
for (int i = 0; i < nums.length; i++) {
if (posMap.containsKey(target - nums[i])) {
return new int[] {i, posMap.get(target - nums[i])};
}
posMap.put(nums[i], i);
}
return new int[2];
}