840. 模拟散列表(哈希表(拉链法))

维护一个集合,支持如下几种操作:

  1. I x,插入一个数 x;
  2. Q x,询问数 x 是否在集合中出现过;

现在要进行 N 次操作,对于每个询问操作输出对应的结果。

输入格式

第一行包含整数 NN,表示操作数量。

接下来 N 行,每行包含一个操作指令,操作指令为 I xQ x 中的一种。

输出格式

对于每个询问指令 Q x,输出一个询问结果,如果 xx 在集合中出现过,则输出 Yes,否则输出 No

每个结果占一行。

数据范围

1≤N≤105
−109≤x≤109

输入样例:

5
I 1
I 2
I 3
Q 2
Q 5

输出样例:

Yes
No

在插入和查询时,数据范围过大,会超时,这里可以避免这个问题 

#include 
#include 

using namespace std;

constexpr int N=1000003;

int n;
int head[N],e[N],ne[N],cnt;

void insert(int x){
    int k=(x%N+N)%N;
    e[cnt]=x;
    ne[cnt]=head[k];
    head[k]=cnt++;
}
bool find(int x){
    int k=(x%N+N)%N;
    for(int i=head[k];i != -1;i=ne[i]){
        if(e[i]==x)
            return true;
    }
    return false;
}
int main(){
    memset(head ,-1,sizeof head);
    scanf("%d",&n);
    while (n--){
        char ch;
        int x;
        cin>>ch>>x;
        if(ch=='I'){
            insert(x);
        }
        else{
            if(find(x)){
                printf("Yes\n");
            }
            else{
                printf("No\n");
            }
        }
    }
}

你可能感兴趣的:(散列表,链表,算法)