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.
Example: Given nums = [2, 7, 11, 15], target = 9, Because nums[0] + nums[1] = 2 + 7 = 9, return [0, 1].
LeetCode第一题,用两层循环的暴力算法可以解出,时间复杂度O(n^2)
public int[] twoSum(int[] nums, int target) {
int[] result = new int[2];
for (int i = 0; i < nums.length - 1; i++) {
for (int j = i + 1; j < nums.length; j++) {
if (nums[i] == target - nums[j]) {
result[0] = i;
result[1] = j;
}
}
}
return result;
}
第二种方法使用HashMap,先查看哈希表中是否存在nums,若没有,则将target-nums[i]压入哈希表,若有,则输出当前nums的index和输入的target-nums在哈希表中的value。时间复杂度为O(n)。
public int[] twoSum(int[] nums, int target) {
int[] result = new int[2];
Map<Integer, Integer> map = new HashMap<Integer, Integer>();
for (int i = 0; i < nums.length; i++) {
if (map.containsKey(nums[i])) {
result[0] = map.get(nums[i]);
result[1] = i;
break;
} else {
map.put(target - nums[i], i);
}
}
return result;
}