哈希表简单算法题:在数组中找到两个数,它们之和等于给定的数字。
程序员文章站
2024-03-15 20:12:18
...
这段时间为了校招找工作开始刷LeetCode,第一道题就是这道题:
两数之和:在数组中找到两个数,它们之和等于给定的数字。
给定一个整数数组 nums 和一个目标值 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};
}
}
}
throw new IllegalArgumentException("There is not such two integer!");
}
}
但是这样做的话,时间空间复杂度都会比较高,看评论的时候发现了用哈希表的方法。哈希表我之前几乎没有用过,这道题正好给了一次机会去简单熟悉一下哈希表的用法。
哈希表是一堆由键和键值组成的数据对的集合,是以索引的方法进行查找。查找时根据键就可以查找到键值,其中键必须是唯一的,键值可以重复。这道题是需要返回所求的两个数的数组下标,所以我们可以以这数值为键,数组下标为键值,建立哈希表求解。代码如下
class Solution {
public int[] twoSum(int[] nums, int target) {
HashMap<Integer,Integer> map = new HashMap<Integer,Integer>();
for(int i=0;i<nums.length;i++){
int leave=target-nums[i];
if(map.containsKey(leave)){
return new int[] {map.get(leave),i};
}
map.put(nums[i],i);
}
throw new IllegalArgumentException("There is not such two integer!");
}
}
其中只用了一次for循环,在循环里实现了哈希表存储数据和哈希表的查询,时间空间复杂度就由之前的O(n²)变成了O(n)。
用到了哈希表的两个方法:containsKey()和get()。第一个是根据键判断哈希表中是否存在这个键,返回值是Boolean。第二个是根据键在哈希表中查找到对应的键值。