元素查找(hash初步)
原创
©著作权归作者所有:来自51CTO博客作者背着代码的蜗牛的原创作品,请联系作者获取转载授权,否则将追究法律责任
元素查找
题目描述:
给出n个正整数,然后有m个询问,每个询问一个整数,询问该整数是否在n个正整数中出现过。
输入描述:
第一行两个整数 n 和m。
第二行n个正整数(1<=n<= 100000)
第三行m个整数(1<=m<=100000)
输出描述:
一共m行,若出现则输出YES,否则输出NO
样例输入:
4 2
2 1 3 4
1 9
样例输出:
YES
NO
数据范围及提示:
所有数据都不超过10^8
重点内容
hash思路:给定一个字符串(或数字串)a,模一个数(一般为质数),令x=a%mod,以x为起点,a为终点建立一条边,查询时扫描某一个点的连边即可。
#include<iostream>
using namespace std;
const int maxn=500010;
int n,m,tot,tmp,head[maxn];
struct node
{
int to;
int next;
}e[maxn];
int get_hash(int x)
{
return x%10007;
}
void add_edge(int u,int v)
{
tot++;
e[tot].to=v;
e[tot].next=head[u];
head[u]=tot;
}
bool find(int u,int v)
{
for(int i=head[u];i;i=e[i].next)
if(e[i].to==v)
return 1;
return 0;
}
int main()
{
int x,y;
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>tmp;
x=get_hash(tmp);
add_edge(x,tmp);
}
for(int i=1;i<=m;i++)
{
cin>>tmp;
x=get_hash(tmp);
if(find(x,tmp))
cout<<"YES"<<endl;
else cout<<"NO"<<endl;
}
return 0;
}