“21天好习惯”第一期－20

学习内容：算法学习（并查集、模拟散列表）

1.并查集

题目：

一共有 nn 个数，编号是 1∼n1∼n，最开始每个数各自在一个集合中。

现在要进行 mm 个操作，操作共有两种：

1. M a b，将编号为 aa 和 bb 的两个数所在的集合合并，如果两个数已经在同一个集合中，则忽略这个操作；
2. Q a b，询问编号为 aa 和 bb 的两个数是否在同一个集合中；

输入格式

第一行输入整数 nn 和 mm。

接下来 mm 行，每行包含一个操作指令，指令为 M a b 或 Q a b 中的一种。

输出格式

对于每个询问指令 Q a b，都要输出一个结果，如果 aa 和 bb 在同一集合内，则输出 Yes，否则输出 No。

每个结果占一行。

数据范围

1≤n,m≤1051≤n,m≤105

输入样例：

4 5
M 1 2
M 3 4
Q 1 2
Q 1 3
Q 3 4

输出样例：

Yes
No
Yes

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 100010;

int p[N];
int n,m;

int find(int x)
{
    if(p[x]!=x)
        p[x]=find(p[x]);
    return p[x];
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i = 0;i<n;i++)
        p[i]=i;
    while(m--)
    {
        char op[2];
        int a,b;
        scanf("%s%d%d",op,&a,&b);
        if(op[0]=='M')
            p[find(a)]=find(b);
        else
            if(find(a)==find(b))
                printf("Yes\n");
            else
                printf("No\n");
    }
}

2.模拟散列表

题目：

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

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

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

输入格式

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

接下来 NN 行，每行包含一个操作指令，操作指令为 I x，Q x 中的一种。

输出格式

对于每个询问指令 Q x，输出一个询问结果，如果 xx 在集合中出现过，则输出 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

代码：

首先需要确定用于取模的值，根据数学经验，该值最好是比N大的第一个素数

#####求比N大的第一个素数
void num(int x)
{
    for(int i=x;;i++)
    {
        bool flag=true;
        for(int j=2;j*j<=i;j++)
        {
            if(i%j==0)
            {
                flag=false;
                break;
            }
        }
        if(flag)
        {
            cout << i << endl;
            break;
        }
    }
}

这里分成了两种方法进行模拟

（1）拉链法

#####拉链法
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

int const N = 100003;

int h[N],e[N],ne[N],idx;

void insert(int x)
{
    int k = (x%N+N)%N;
    e[idx]=x;
    ne[idx]=h[k];
    h[k]=idx++;
}

bool 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;
}

int main()
{
    int n;
    cin >> n;
    memset(h,-1,sizeof h);
    while(n--)
    {
        char op[2];
        int x;
        scanf("%s%d",op,&x);
        if(op[0]=='I')
            insert(x);
        else
            if(find(x))
                cout << "Yes" << endl;
            else
                cout << "No" << endl;
    }
    return 0;
}

（2）开放寻址法

#####开放寻址(纸)法
又称厕所找坑位法。
全员厕所霸王，蹲下就不起来。
人走了还要占着坑，贴张告示不让别人蹲。
坑位被占了，就继续找坑，找几下就能有。
找到厕所底再回厕所头来找，有丰富的坑位不怕找不着。
时间复杂度为O(1)
根据经验，N的值要开2~3倍。
要删除一个点的值时不能直接删，只能加上一个标记，否则将后面的值会丢失。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

int const N = 200003,null = 0x3f3f3f3f;


int h[N];

int find(int x)
{
    int k = (x%N+N)%N;
    while(h[k]!=0x3f3f3f3f&&h[k]!=x)
    {
        k++;
        if(k == N)
            k=0;
    }
    return k;
}

int main()
{
    int n;
    cin >> n;
    memset(h,0x3f,sizeof h);
    while(n--)
    {
        char op[2];
        int x;
        scanf("%s%d",op,&x);
        int k = find(x);
        if(op[0]=='I')
            h[k]=x;
        else
            if(h[k]==x)
                cout << "Yes" << endl;
            else
                cout << "No" << endl;
    }
    return 0;
}