原题地址:https://leetcode.com/problems/two-sum/
题目的要求是:从数组中找出两个数,使其和为特定值target。例如nums = [2, 7, 11, 15], target = 9,因为2+7=9,返回[0, 1]。
最直接的做法(洋气一点叫Brute Force)就是从第一个数开始,一个一个往后找呗,但是作为一道面试题怎么可能这么简单呢,面试官肯定会问:你的算法时间复杂度是O(n2),有没有办法降到O(n)呢?可以反过来想一想:给你一个数x,要求你从数组里找到另一个数使得它们的和为target,换句话说,其实不就是要在数组里找到一个值为(target - x)的数嘛!我们可以把(target - x)称为“补数”,引入一个哈希表存储这些“补数”,这是一种空间换时间的做法。(一个需要注意的地方是要把自己排除在外)
public class Solution {
public int[] twoSum(int[] nums, int target) {
int[] result = new int[2];
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
int complement = target - nums[i];
if (map.containsKey(nums[i]) && map.get(nums[i]) != i) {
result[0] = map.get(nums[i]);
result[1] = i;
break;
}
map.put(complement, i);
}
return result;
}
}