AcWing - 840. 模拟散列表

1、题目描述

维护一个集合，支持如下几种操作：

1. “I x”，插入一个数x；
2. “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");
            }
        }
    }
    
}