题目描述
维护一个集合,支持如下几种操作:
I x
,插入一个数 x;
Q x
,询问数 x 是否在集合中出现过;
现在要进行 N 次操作,对于每个询问操作输出对应的结果。
输入格式
第一行包含整数 N,表示操作数量。
接下来 N 行,每行包含一个操作指令,操作指令为 I x
,Q x
中的一种。
输出格式
对于每个询问指令 Q x
,输出一个询问结果,如果 x 在集合中出现过,则输出 Yes,否则输出 No。
每个结果占一行。
数据范围
1≤N≤10^5
−109≤x≤10^9
输入样例:
5
I 1
I 2
I 3
Q 2
Q 5
输出样例:
Yes
No
思路
代码示例
拉链法:
#include <iostream>
#include <cstring>
using namespace std;
const int N = 100003;// 这里取稍微大点的质数
// h里面存的是数x对应唯一标识的idx,而ne存储的是唯一标识idx对应的下一个idx;e里面存的是idx对应的值
int h[N], e[N], ne[N], idx;
void insert(int x)
{
int k = (x % N + N) % N;// 这里是为了将负数部分也映射到0 ~ 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;
cin >> n;
memset(h, -1, sizeof h); //将数组中的初始值设置为-1
char op;
int x;
while(n --)
{
cin >> op;
if(op == 'I')
{
cin >> x;
insert(x);
}
else
{
cin >> x;
if(find(x)) cout << "Yes" << endl;
else cout << "No" << endl;
}
}
return 0;
}
开放寻址法(这里遇到一个很吐血的错误,就是这个null的值我设置层0x3f3f3f
而不是0x3f3f3f3f
时就会报超时。这里涉及一个memset函数的坑(就是memset是逐字节进行填充的)然后这题可以用一个超越x范围的数值就可以,我这里用了M = 1e9 + 1 比 x的最大值大一点也能过题。
#include <iostream>
#include <cstring>
using namespace std;
const int M = 1e9 + 1;
const int N = 200003, null = 0xffffffff;// 这里要存更多的数了 h[x] 因为没有 这里的null值大于N
// 和拉链法的区别是h直接存放的是实际值,所以不能用-1来作为h的初始值 然后这里是用的是大于N的一个值
int h[N], n;
int find(int x)//返回应该存储的位置
{
int k = (x % N + N) % N;
while(h[k] != M && h[k] != x) k = (k + 1) % N;
return k;
}
int main()
{
cin >> n;
//memset(h, null, sizeof h);
for(int i = 0; i < N; i ++) h[i] = M;
char op;
int x;
while(n --)
{
cin >> op >> x;
int k = find(x);
if (op == 'I')
{
h[k] = x;
}
else{
if(h[k] != M) cout << "Yes" << endl;
else cout << "No" << endl;
}
}
return 0;
}