题目描述
维护一个集合,支持如下几种操作:
“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
思路
模拟hash列表
哈希列表有多种
一般方法有拉链法,开放寻址法,字符前缀hash(针对字符串)
拉链法
初始化hash列表
memset(h, -1, sizeof h);
表内所有元素指向节点-1
假设表大小为5, 插入1,2,…,7,8
如图所示
拉链法 c++代码
#include <cstring>
#include <iostream>
using namespace std;
const int N = 100010;
int h[N], e[N], ne[N], idx;
void insert(int x){
//c++不同于其他语言 负数的余数为负数 -3 % 2 = -1,-1无法作为数组下标
int k = (x % N + N) % N; //负数为-1e9 + 1e5仍然是很大的负数,用-1e9 % 1e5 + 1e5保证整数
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 == 'I') insert(x);
else{
if(find(x)) puts("Yes");
else puts("No");
}
}
return 0;
}