没错这是一道简单题 (:зゝ∠)
问题描述:Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution, and you may not use the same element twice.
example:
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].
解题思路:
这道题最简单的解题思路就是,在遍历到 i 位置的数时,再把 i 前面的数遍历一遍,如果 i 位置和它前面 j 位置的数的和为target,则返回。但是这样做时间复杂度是O(n^2)。
我的思路是从头到尾只遍历一次数组。先设置一个map,用来存储已经遍历过的数字,从0位置开始遍历数组,如果在map中找到了target-num[i],表明在这个 i 位置的数字前面的有与 i 位置和为target的数,将其返回,如果在map中不存在,那么就将现在 i 位置的数的下标和数值放入map中。这样的时间复杂度只有O(n)。
代码如下:
public static int[] twoSum(int[] nums, int target) {
Map<Integer, Integer> map = new HashMap<>();
int[] result = new int[2];
for (int i = 0; i < nums.length; i++) {
if (map.containsKey(target - nums[i])) {
// 如果map中有对应的值
result[0] = Math.min(i, map.get(target - nums[i]));
result[1] = Math.max(i, map.get(target - nums[i]));
return result;
} else {
// 如果map不存在对应的值
map.put(nums[i], i);
}
}
return result;
}