维护一个集合,支持如下几种操作:
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
方法一:
拉链法散列表。复习一下用数组+idx存储各种数据结构的方法。。。
#include <bits/stdc++.h>
using namespace std;
const int N = 100010;
int n, e[N], idx, h[N], ne[N]; // e存放数据
void insert(int x) {
int hx = (x % N + N) % N; // +N防止是负数
e[idx] = x;
ne[idx] = h[hx]; // 头插法
h[hx] = idx++;
}
int query(int x) {
int hx = (x % N + N) % N;
int d = h[hx];
while (true) {
if (d == -1) return 0;
if (e[d] == x) return 1;
d = ne[d];
}
}
int main() {
char op[2];
int x;
memset(h, -1, sizeof h);
scanf("%d", &n);
while (n--) {
scanf("%s%d", op, &x);
if (op[0] == 'I') {
insert(x);
} else if (op[0] == 'Q') {
if (query(x)) puts("Yes");
else puts("No");
}
}
}