算法模板:模拟离散列表(哈希表:拉链式)
模板题
维护一个集合,支持如下几种操作:
-
I x
,插入一个数 x; -
Q x
,询问数 x是否在集合中出现过;
现在要进行 N次操作,对于每个询问操作输出对应的结果。
输入格式
第一行包含整数 N,表示操作数量。
接下来 N 行,每行包含一个操作指令,操作指令为 I x
,Q x
中的一种。
输出格式
对于每个询问指令 Q x
,输出一个询问结果,如果 x在集合中出现过,则输出 Yes
,否则输出 No
。
每个结果占一行。
数据范围
1≤N≤105
−109≤x≤109
输入样例:
5
I 1
I 2
I 3
Q 2
Q 5
输出样例:
Yes
No
题目链接:https://www.acwing.com/problem/content/842/
分析
哈希表分为字符串哈希方式、开放寻址发和拉链法,这里使用拉链法;哈希表的查询可以达到O(1);
插入一个数值时存于一个e[N]数组中,下标确定方式为x mod N ∈(0,N),但是有时候可能存在两个数字同余,使得两个数值的k值一样,这时候就产生了冲突,所以就需要定义一个数组h[N],其中每个元素初始化为-1(memset(h,-1,sizeof(h));)表示链表末端的NULL(这里的链表使用静态链表)每一个h数组的元素都是一个链表的表头,这样一个数组就拉起来了很多链条,每个链条储存一个数字在e数组的下标索引。
在插入元素的时候,计算k值,在表头为h[k]的链表中插入元素,可以是头插法或者是尾插法。
在查询元素时,根据要查询的数值的k值来查询表头为h[k]的链表,从表头开始遍历该链表查询是否有相应元素。
CODE
#include<iostream>
#include<cstring>
using namespace std;
const int N=100003;
int h[N],e[N],ne[N],idx=0;
void insert(int x){//插入静态链表
int k=(x%N+N)%N;//防止x%N为负数,所以需要+N,但是+N会使得k大于N所以还有在最外面加个%N
e[idx]=x;//存储数据
ne[idx]=h[k];
h[k]=idx++;
}
bool find(int x){
int k=(x%N+N)%N;
for(int i=h[k];i!=-1;i=ne[i]){
if(e[i]==x)
return true;
}
return false;
}
int main(){
int n;
scanf("%d",&n);
memset(h,-1,sizeof(h));
while(n--)
{
char op[2];
int x;
scanf("%s%d",op,&x);
if(op[0]=='I'){
insert(x);
}else{
if(find(x))printf("Yes\n");
else printf("No\n");
}
}
return 0;
}