HDU 1198--并查集

http://acm.hdu.edu.cn/showproblem.php?pid=1198

并查集，把每个方块编号，然后有联通的将其并起来，方向的控制及编号的处理比较复杂。

代码：

#include<iostream>
using namespace std;
int farm[11][4]={{1,0,0,1},{1,1,0,0},{0,0,1,1},{0,1,1,0},{1,0,1,0},
{0,1,0,1},{1,1,0,1},{1,0,1,1},{0,1,1,1},{1,1,1,0},{1,1,1,1}};//将A~H的水管方向按照上、右、下、左标记
int dist[4][2]={{-1,0},{0,1},{1,0},{0,-1}},p[3000],vis[3000];
char g[55][55];
int findx(int x)
{
    if(x!=p[x])
        p[x]=p[findx(p[x])];
    return p[x];
}
void merge(int x,int y)
{
    p[findx(x)]=findx(y);
}
int main()
{
    int n,m,i,j,k;
    while(~scanf("%d %d ",&n,&m),n!=-1||m!=-1)
    {
        for(i=0;i<=n*m;i++)
                p[i]=i,vis[i]=0;
        for(i=1;i<=n;i++)
            for(j=1;j<=m;j++)
                scanf(" %c ",&g[i][j]);
        for(i=1;i<=n;i++)
            for(j=1;j<=m;j++)
            {
                if(!vis[(i-1)*m+j])
                {
                    for(k=0;k<4;k++)
                    {
                        int dx=i+dist[k][0];
                        int dy=j+dist[k][1];
                        if(dx<1||dy<1||dx>n||dy>m)
                            continue;
                        int fx;
                        if(k%2==0)
                            fx=(k==0?2:0);
                        else
                            fx=(k==1?3:1);           //控制方向
                        if(farm[g[dx][dy]-'A'][fx]&&farm[g[i][j]-'A'][k]&&!vis[(dx-1)*m+dy])
                        {
                            merge((i-1)*m+j,(dx-1)*m+dy);
                            vis[(i-1)*m+j]=1;
                        }
                    }
                }
            }
            int cnt=0;
            for(i=1;i<=n*m;i++)
                if(p[i]==i)
                    ++cnt;
            printf("%d\n",cnt);
    }
    return 0;
}