HDU 1198(并查集)

原创于 2018-04-25 19:53:23 发布 · 172 阅读

0 ·

CC 4.0 BY-SA版权

数据结构专栏收录该内容

103 篇文章

订阅专栏

本文介绍了一种使用并查集算法解决管道连接问题的方法。通过定义特定的数据结构和操作，确保了可以高效地判断任意两个管道节点是否连通，并计算整个网络中独立连通区域的数量。适用于需要解决类似迷宫或管道布局连通性问题的场景。

#include <bits/stdc++.h>

using namespace std;

//二进制表示北东南西。1表示有，0表示没有。
const int pipetype[] = {
     9      // A 1001
    ,12    // B 1100
    ,3      // C 0011
    ,6      // D 0110
    ,10    // E 1010
    ,5      // F 0101
    ,13    // G 1101
    ,11    // H 1011
    ,7      // I 0111
    ,14    // J 1110
    ,15    // K 1111
};

const int N=50;
int f[N][N];            //爹
char farm[N][N+1];      //农厂。
int m,n,cnt;

void init(int row,int col)
{
    for(int i=0;i<row;i++)
        for(int j=0;j<col;j++)
            f[i][j]=i*n+j;
    cnt=row*col;
}

int Find(int row,int col)
{
    return (f[row][col]==row*n+col)?f[row][col]:f[row][col]=Find(f[row][col]/n,f[row][col]%n);
}

void Union(int row1,int col1,int row2,int col2)
{
    int p1=Find(row1,col1);
    int p2=Find(row2,col2);
    if(p1!=p2)
    {
        f[p2/n][p2%n]=p1;               //靠左原则。
        cnt--;
    }
}

void judge(int row,int col)
{
    //与左侧联通
    if(col>0&&(pipetype[farm[row][col]-'A']&1)&&((pipetype[farm[row][col-1]-'A']>>2)&1))
        Union(row,col,row,col-1);
    //与上方联通
    if(row>0&&((pipetype[farm[row][col]-'A']>>3)&1)&&((pipetype[farm[row-1][col]-'A'])>>1)&1)
        Union(row,col,row-1,col);
}

int main()
{
    ios::sync_with_stdio(false);cin.tie(0);

    while(~scanf("%d%d",&m,&n)&&n>0)
    {
        init(m,n);
        for(int i=0;i<m;i++)
            scanf("%s",farm[i]);
        for(int i=0;i<m;i++)
            for(int j=0;j<n;j++)
                judge(i,j);
        printf("%d\n",cnt);
    }
    return 0;
}