维护一个集合,支持如下几种操作:
I x
,插入一个数 xx;Q x
,询问数 xx 是否在集合中出现过;
现在要进行 NN 次操作,对于每个询问操作输出对应的结果。
输入格式
第一行包含整数 NN,表示操作数量。
接下来 NN 行,每行包含一个操作指令,操作指令为 I x
,Q x
中的一种。
输出格式
对于每个询问指令 Q x
,输出一个询问结果,如果 xx 在集合中出现过,则输出 Yes
,否则输出 No
。
每个结果占一行。
数据范围
1≤N≤1051≤N≤105
−109≤x≤109−109≤x≤109
输入样例:
5
I 1
I 2
I 3
Q 2
Q 5
输出样例:
Yes
No
拉链法
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 100003;
int h[N],e[N],ne[N],idx;
void Insert(int x)
{
int k=(x % N + N )% N;
e[idx] = x;
ne[idx] = h[k];
h[k] = idx++;
}
int 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;
}
寻地址法
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 200003,inf=0x3f3f3f3f;
int h[N];
int find(int x)
{
int k=(x % N + N) % N;
while(h[k]!=inf && h[k] != x)
{
k++;
if(k == N) k=0;
}
return k;
}
int main()
{
int n;
scanf("%d", &n);
memset(h, 0x3f, sizeof h);
while (n -- )
{
char op[2];
int x;
scanf("%s%d",op,&x);
int ans=find(x);
if(*op=='I') h[ans] = x;
else
{
if(h[ans] != inf) puts("Yes");
else puts("No");
}
}
return 0;
}