1、题目描述
维护一个集合,支持如下几种操作:
- “I x”,插入一个数x;
- “Q x”,询问数x是否在集合中出现过;
现在要进行N次操作,对于每个询问操作输出对应的结果。
输入格式
第一行包含整数N,表示操作数量。
接下来N行,每行包含一个操作指令,操作指令为”I x”,”Q x”中的一种。
输出格式
对于每个询问指令“Q x”,输出一个询问结果,如果x在集合中出现过,则输出“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
2、分析
使用链地址法 或者 开放寻址法
3、代码
(1)链地址法
数组 + 单链表 实现
//两种方法:1.链地址法 2.开放地址法
import java.util.*;
import java.io.*;
public class Main{
static int N = 100003; //取质数
static int[] h = new int[N]; //拉链法的哈希数组
static int[] e = new int[N], ne = new int[N]; //单链表,ne[i]保存点的下标位置
static int idx = 0; //当前用到哪个索引
//单链表头插法
static void insert(int x) {
int k = (x % N + N) % N;
e[idx] = x;
ne[idx] = h[k];
h[k] = idx;
idx ++;
}
static boolean 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;
}
public static void main(String[] args) throws Exception{
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
int n = Integer.parseInt(br.readLine());
Arrays.fill(h, -1); //初始化
for(int i = 0;i < n;i ++) {
String line = br.readLine();
String[] splits = line.split(" ");
String op = splits[0];
int x = Integer.parseInt(splits[1]);
if("I".equals(op)) insert(x);
else {
boolean flag = find(x);
if(flag) System.out.println("Yes");
else System.out.println("No");
}
}
}
}
(2)开放寻址法
一般开一个数组容量取数据范围的2-3倍的一个质数
//两种方法:1.链地址法 2.开放地址法
import java.util.*;
import java.io.*;
public class Main{
static int N = 300003; //数组容量取数据范围的2-3倍的一个质数
static int[] h = new int[N];
static int find(int x) {
int k = (x % N + N) % N;
while(h[k] != -1 && h[k] != x) {
k ++;
}
return k;
}
public static void main(String[] args) throws Exception{
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
int n = Integer.parseInt(br.readLine());
Arrays.fill(h, -1);
for(int i = 0;i < n;i ++) {
String line = br.readLine();
String[] splits = line.split(" ");
String op = splits[0];
int x = Integer.parseInt(splits[1]);
int idx = find(x);
if("I".equals(op)) {
h[idx] = x;
}
else {
if(h[idx] == x) System.out.println("Yes");
else System.out.println("No");
}
}
}
}