leetcode学习日记1-two sum-利用map模拟哈希表达到快速查找
前几天,作为一个计算机菜鸟,才开始刷leetcode。
leetcode第一题,two sum,居然就没那么简单。在追求运行速度的过程中,有应用哈希表的思想。然而,题解只有java版本的。这篇博客就用来以two sum题目为例概括一下如何利用map来模拟哈希表达到快速查找,并提供相应的c++代码。
two sum 题目如下:
给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
我的第一遍程序是暴力搜索,对每一个元素nums[i],从i向后开始搜索,搜索到nums[i]+nums[j]==target则返回,搜索不到则令i自加。
代码如下
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
int i=0;
static vector<int> a(2);
for(i=0;i<nums.size();i++)//第一个数
{
for(int j=i+1;j<nums.size();j++)//第二个数,从第一个数后面开始,避免重复检索
{
if(target==nums[i]+nums[j])
{
a[0]=i;
a[1]=j;
return a;
}
}
}
}
};
结果……不怎么样
- 于是看了一下题解的方法
题解用到了一个叫做哈希表的思想 - 哈希表主要是通过一个函数建立两个量之间的映射,以此来快速检索
- 考虑这道题目,核心问题就是,当我们知道数组中的一个元素时,我们想知道值为(target-该元素值)的数在不在这个数组中。我的第一种暴力搜索算法用枚举法寻找第二个值的存在性,由于每次都要遍历一遍,有很多重复的计算量。我们已知要找的数,希望能直接查找到这个数是否在数组中,而能避免逐个检验。此外,由于输出的需要,我们还要知道该数在数组中的位置。
- 我们可以想象一个映射,即一一对应关系:每一个数组中的数值,与它在数组中的位置一一对应。我们将每一个数组元素以它的数值作为索引(因为查找元素时是以值为参考),将它的位置作为对应的map元素。
- 这样一来,我们只需要检查我们的map中是否有这个索引为该值的元素,如果有,返回其对应位置即可。
- 有没有一点用空间换时间的味道
接下来,附上c++的代码
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
map<int,int> hash;//定义map,其索引和元素都是int
int i;
for(i=0;i<nums.size();i++)//逐一存入map,并同时检索
{
int complement=target-nums[i];//需要检索的值
if(hash.find(complement)!=hash.end())//如果map中找得到该值
{ vector<int> a(2);
a[0]=hash.find(complement)->second;//返回其位置
a[1]=i;
return a;
}
hash[nums[i]]=i;//如果map中找不到该值,把当前数存入map,开始处理下一个数
}
}
};
结果: