常见面试题 —— 两数之和(拒绝暴利法)
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那两个整数,并返回他们的数组下标。(leetcode上面的一道题目)
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
菜鸟程序员最容易被问到的一个面试题,我就被gg了,因为平常也不怎么用哈希表,结果答题就直接暴力了,暴力法简直low到爆炸想找个洞钻进去。
记录一下这个简单又具有考察性的题目。
解法一:两遍哈希表
时间复杂度:O(n), 我们把包含有 n 个元素的列表遍历两次。 由于哈希表将查找时间缩短到 O(1) ,所以时间复杂度为O(n)。 空间复杂度:O(n), 所需的额外空间取决于哈希表中存储的元素数量,该表中存储了 n 个元素。
思路
使用了两次迭代。
在第一次迭代中,我们将每个元素的值和它的索引添加到表中。
在第二次迭代中,我们将检查每个元素所对应的目标元素(即目标元素与当前元素之差target - nums[i])是否存在于表中。注意,该目标元素不能是 nums[i] 本身!
public static int[] twoSum(int[] nums, int target) { Map<Integer,Integer> map = new HashMap<>(); for (int i = 0; i < nums.length; i++) { // 把数值作为key,位置作为value map.put(nums[i], i); } for (int i = 0; i < nums.length; i++) { // 计算目标值和当前值的差,并且在map中查找这个差是否存在 // containsKey方法判断是否有key(有key-value键值对就会返回true,不存在则返回false) int x = target - nums[i]; if (map.containsKey(x) && map.get(x) != i) { return new int[] { i, map.get(x)}; } } return null; }
解法二:一遍哈希表
时间复杂度:O(n), 我们只遍历了包含有 n 个元素的列表一次。在表中进行的每次查找只花费 O(1) 的时间。 空间复杂度:O(n), 所需的额外空间取决于哈希表中存储的元素数量,该表最多需要存储 n 个元素。
思路
事实证明,我们可以一次性完成。在进行迭代并将元素插入到表中的同时,我们还会回过头来检查表中是否已经存在当前元素所对应的目标元素。如果它存在,那我们已经找到了对应解,并立即将其返回。
public static int[] twoSum(int[] nums, int target) { Map<Integer,Integer> map = new HashMap<>(); for (int i = 0; i < nums.length; i++) { // 算出当前值对应的差值
int x = target - nums[i]; // 先判断key中是否存在这个差值,存在的话就直接返回,不存在的话就增加键值对 if (map.containsKey(x)) { return new int[] {map.get(x), i}; } map.put(nums[i], i); } return null; }
真的是又简单又考察思维方式的题。。。
继续加油叭~