[hihocoder1066]无间道之并查集

问题描述

就是并查集的简单应用。详情请参考 hihocoder官网。

算法详解

每个节点存储两个数值，其父亲节点的位置以及类别内部节点的个数。其中父子节点位于一个类别，而类别内部节点数只对根节点有意义。

1. 寻找根节点

根节点的位置可以用于区分两个节点的类别。所以寻找根节点是个十分常见的操作，这里使用递归实现，同时加入短路，减少平均复杂度。

int find_root(int i){
    if (nodes[i].parent == -1)
        return i;
    return nodes[i].parent = find_root(nodes[i].parent);
}

2. 结合两个类别

题目中结合两个节点，其实也就是结合这两个节点所在的类别。鉴于每个节点只能有一个父亲节点，以保障树形结构，所以最终是需要结合两个类别对应的根节点。而为了使得树更加平衡，加入树的大小作为指标，使得较小的树作为较大树根节点的子节点。

void combine(int r1,int r2){
    if (r1 == r2)
        return;
    if (nodes[r1].num < nodes[r2].num)
        swap(r1,r2);
    nodes[r2].parent = r1;
    nodes[r1].num += nodes[r2].num;
    nodes[r2].num = 0;
}

全部代码

#include <iostream>
#include <map>
#include <string>

using namespace std;

struct Node{
    int parent;
    int num;

    Node():parent(-1),num(1) {}
    Node(const Node &n):parent(n.parent),num(n.num) {}
    ~Node() {}
};

Node nodes[1000000];
int find_root(int i){
    if (nodes[i].parent == -1)
        return i;
    return nodes[i].parent = find_root(nodes[i].parent);
}

void combine(int r1,int r2){
    if (r1 == r2)
        return;
    if (nodes[r1].num < nodes[r2].num)
        swap(r1,r2);
    nodes[r2].parent = r1;
    nodes[r1].num += nodes[r2].num;
    nodes[r2].num = 0;
}

void print_nodes(int cnt){
    cout << "parent:\t";
    for (int i = 0;i < cnt;++i)
        cout << nodes[i].parent << '\t';
    cout << endl;

    cout << "size:\t";
    for (int i = 0;i < cnt;++i)
        cout << nodes[i].num << '\t';
    cout << endl;
}

map<string,int> str2int;

int main()
{
    int N,cnt=0;
    cin >> N;
    int op;
    string str1,str2;

    for (int i = 0;i < N;++i){
        cin >> op >> str1 >> str2;
        if (str2int.find(str1) == str2int.end())
            str2int.insert(pair<string,int>(str1,cnt++));
        if (str2int.find(str2) == str2int.end())
            str2int.insert(pair<string,int>(str2,cnt++));
        if (op == 0)
            combine(find_root(str2int[str1]),find_root(str2int[str2]));
        else
            cout << ((find_root(str2int[str1])==find_root(str2int[str2]))?"yes":"no") << endl;
    }
    return 0;
}