hdu 1198 Farm Irrigation

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

#include <cstdio>
#include <cstring>
#include <algorithm>
#define maxn 4000
using namespace std;

int b[11][4]={{1,0,1,0},{1,0,0,1},{0,1,1,0},{0,1,0,1},{1,1,0,0},{0,0,1,1},{1,0,1,1},{1,1,1,0},{0,1,1,1},{1,1,0,1},{1,1,1,1}};
char g[60][60];
int n,m;
int f[maxn];

int Find(int x)
{
    if(x==f[x]) return x;
    return f[x]=Find(f[x]);
}

void inti()
{
    for(int i=0; i<=n*m+1; i++) f[i]=i;
}

void union1(int a,int b)
{
    int fa=Find(a);
    int fb=Find(b);
    if(fa!=fb)
        f[fa]=fb;
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        inti();
        if(n==-1&&m==-1) break;
        for(int i=0; i<n; i++)
        {
            scanf("%s",g[i]);
        }
        for(int i=0; i<n; i++)
        {
            for(int j=0; j<m; j++)
            {
                if(b[g[i][j]-'A'][3]==1&&b[g[i][j+1]-'A'][2]==1&&j+1<m)
                {
                    union1(i*m+j,i*m+j+1);
                }
                if(b[g[i][j]-'A'][1]==1&&b[g[i+1][j]-'A'][0]==1&&i+1<n)
                {
                    union1(i*m+j,(i+1)*m+j);
                }
            }
        }
        int cnt=0;
        for(int i=0; i<n*m; i++)
        {
            if(f[i]==i) cnt++;
        }
        printf("%d
",cnt);
    }
    return 0;
}