古有六级abandon,现有力扣两数之和。两数之和的“里程碑”式的意义不言而喻呀,作为力扣的第一个题目,不会太难,不然一来就打击了众多像我一样的菜鸡程序员。
废话不多说,来看看题目:
给定一个整数数组
nums
和一个整数目标值target
,请你在该数组中找出 和为目标值target
的那 两个 整数,并返回它们的数组下标。你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。
读完题目,可以发现题目的含义实际上是,从一个数组中找两个数a,b,使得a+b=target,而且数组中只有唯一的一组答案,同时要求每个元素只能用一次。
拿到这个题,刷刷刷,直接一个暴力!
下面是js的暴力解法。
/**
* @param {number[]} nums
* @param {number} target
* @return {number[]}
*/
var twoSum = function(nums, target) {
for(let i = 0;i < nums.length;i++){
for(let j=i+1;j<nums.length;j++){
if(nums[i]+nums[j]===target){
return [i,j]
}
}
}
};
当然,像我这样的菜鸡程序员可能只会以上一种解法,但是大佬们的解法就层出不穷了。下面说一种利用哈希表求解的方法。
利用哈希表求解有一个好处,就是可以用空间换时间。首先遍历一次数组,将每个数都添加到哈希表中,再遍历一次数组,判断target-nums[i]是否在哈希表中,如果是,那就直接返回,不是直接判断下一个(此题条件就是必有解,因此不用考虑太多)。
聪明的读者肯定发现了上述方法的问题,那就是会重复的利用元素。那应该采用什么办法来规避这个问题呢?
其实不需要提前把数组中的值加入哈希表,而是可以仅仅通过一次遍历就解决问题。
talk is cheap, show me the code!
代码如下(是不是简洁多了?时间复杂度也降低了一个量级,果然大佬就是大佬啊):
/**
* @param {number[]} nums
* @param {number} target
* @return {number[]}
*/
var twoSum = function(nums, target) {
let map = new Map();
for(let i = 0;i < nums.length; i++){
if(map.get(target - nums[i])!=null){
return [i,map.get(target-nums[i])];
}
map.set(nums[i],i);
}
};