维护一个集合,支持如下几种操作:
“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
#include <bits/stdc++.h>
using namespace std;
const int n=1e5+30;//离散化
int a[n];
int N;
string t;
const int flag=1e9+50;//这个在uscao那个奶牛那道题也可以用到这个思路
//那道题是将所有点的陨石降落时间初始化为一个大于上限的值然后取MIn,
int find(int x){
int k=(x%n+n)%n;//locate&&防止RE,保证合法性
while(a[k]!=flag&&a[k]!=x){//proof:如果在这之前填过一定会被找到
k++;
if(k==n){
k=0;
}
}
return k;//要么是可以填坑的位置or 已经是x的位置
}
int x;
int main(){
cin>>N;
for(int i=0;i<n;i++){
a[i]=flag;
}
while(N--){
cin>>t>>x;
if(t=="I"){
a[find(x)]=x;
}
else{
if(a[find(x)]==x){
puts("Yes");
}
else{
puts("No");
}
}
}
return 0;
}
update:
刚才回寝室路上又想到了一种情况,就是发生冲突的情况,如果两个不同的值取Mod映射到同一个下标上,这个算法的成立性与否,but想了一会,这个算法依然是对的,就是即便映射到同一个下标上,我们在找到下一个被flag标记过的坑之前,我们依然会遍历搜一遍,一定能找到。