题目链接:1.两数之和

初看题目相信大家都能想到枚举的做法,简单来说把数组里面的所有值,均两两组合相加。若结果与target相等,则将两个数字的下标返回即可。
class Solution {public int[] twoSum(int[] nums, int target) {for(int i = 0; i < nums.length; i++){for(int j = i + 1; j < nums.length; j++){if(nums[i] + nums[j] == target){return new int[] {i, j};}}}return new int[] {0, 0};}
}
提交后,虽然能AC,但我们可以发现距离最优解仍然有很长的距离。

同时我们再对我们的代码分析,可以得出我们当前实现代码的时间复杂度为 O(n2)O(n^2)O(n2)。那我们能不能把时间复杂度优化为 O(n)O(n)O(n)呢。
回看我们的逻辑,我们的本质是为 x 寻找 target - x,那我们现在要做的就是减少 寻找 target - x 的复杂度。对于此目的,我们要采取的方案就是哈希表,先在哈希表中寻找 target - x 是否存在,若存在则直接把 x 和 target - x 的下标返回。若不存在,则将 x 加入哈希表中。
大致逻辑确定了,我们再继续处理一些细节。首先是这个哈希表怎么存储,通过前文分析,我们得知我们需要存储 数字 和 下标,而且我们还要通过 数字 来判断该数字是否存在哈希表中。相信大家已经能想到了,我们要采用的是 HashMap, 其中 key 用来存数字的值,这样我们就可以通过 map.containsKey 来判断该数字是否存在与哈希表中,value 用来存下标,这样返回下标的时候我们可以直接通过 key 获取,而不是再去数组里面找一遍了。
可能又有小伙伴会有这样的疑问,需要判断哈希表里获取的数字的下标是否与我当前 x 的下标一致,若 x = target - x 不判断会不会返回把这一个数字当成两个数字返回呢。其实这是不需要担心的,因为我们在哈希表中寻找 target - x 的时候,并没有将 x 存入哈希表中,所以这时获取的 target - x 一定不是我们当前的 x。
因为我们采用的哈希表, map.containsKey 的时间复杂度为 O(1)O(1)O(1),那么我们这样实现的整体时间复杂度即为O(n)O(n)O(n),具体代码实现如下:
class Solution {public int[] twoSum(int[] nums, int target) {Map map = new HashMap<>();for(int i = 0; i < nums.length; i++){if(map.containsKey(target - nums[i])) {return new int[] {map.get(target-nums[i]),i};}map.put(nums[i], i);}return new int[] {0, 0};}
}
下一篇:阴风来西北,愁随回鹘