洛谷 P3367 并查集 【模板题】

题目描述

如题，现在有一个并查集，你需要完成合并和查询操作。

输入输出格式

输入格式：

第一行包含两个整数N、M，表示共有N个元素和M个操作。

接下来M行，每行包含三个整数Zi、Xi、Yi

当Zi=1时，将Xi与Yi所在的集合合并

当Zi=2时，输出Xi与Yi是否在同一集合内，是的话输出Y；否则话输出N

输出格式：

如上，对于每一个Zi=2的操作，都有一行输出，每行包含一个大写字母，为Y或者N

输入输出样例

输入样例#1： 

4 7
2 1 2
1 1 2
2 1 2
1 3 4
2 1 4
1 2 3
2 1 4

输出样例#1： 

N
Y
N
Y

说明

时空限制：1000ms,128M

数据规模：

对于30%的数据，N<=10，M<=20；

对于70%的数据，N<=100，M<=1000；

对于100%的数据，N<=10000，M<=200000。

注意：

并查集合并时要合并并查集的“根”。

注意使用递归进行路径压缩。

以下AC代码： 645ms, 788KB

#include <iostream>
using namespace std;

const int maxn = 10001;
int N;

int p[maxn];

int ffind(int x)
{
    return p[x] == x? x : p[x] = ffind(p[x]);
}

void init(int n)
{
    for(int i = 1; i <= n ;i++)
        p[i] = i;

}

void uunion(int u, int v)
{
    p[ffind(u)] = ffind(v);
}


int main()
{

    cin >> N;
    int T;
    cin >> T;
    init(N);
    int cmd;
    int u,v;
    while(T--)
    {
        cin >> cmd;
        cin >> u >> v;
        if(cmd == 1)
        {
            uunion(u,v);
        }
        else if(cmd == 2)
        {
            int x = ffind(u);
            int y = ffind(v);
            if(x == y) cout << "Y" << endl;
            else cout << "N" << endl;

        }
    }
    return 0;
}