Farm Irrigation HDU - 1198 （并查集）

Farm Irrigation

HDU - 1198

题意：给11种管道，问草地最少需要打多少个井才可以全部灌溉。

把每种管道的状态用二进制表示一下，然后对每一块草地，判断能否和上面或者左面的草地的管道连接。

然后并查集搞一下。

#include <bits/stdc++.h>
using namespace std;
const int maxn=55;

int g[12]={10,9,6,5,12,3,11,14,7,13,15};
int f[maxn*maxn];
char p[maxn][maxn];
int n,m;

int gf(int x){
    return x==f[x]?f[x]:f[x]=gf(f[x]);
}
void uni(int a,int b){
    int pa=gf(a),pb=gf(b);
    f[pa]=pb;
}
void check(int i,int j){
    int id=i*m+j;
    if(i>0&&((g[p[i][j]-'A']>>3)&1)&&((g[p[i-1][j]-'A']>>2)&1)) {
        uni(id,id-m);
    }
    if(j>0&&((g[p[i][j]-'A']>>1)&1)&&((g[p[i][j-1]-'A'])&1)) {
        uni(id,id-1);
    }
    return ;
}

int main(){
   // freopen("in.txt","r",stdin);
    while(scanf("%d%d",&n,&m)&&(n!=-1&&m!=-1)){
        for(int i=0;i<n;i++){
            scanf("%s",p[i]);
            for(int j=0;j<m;j++){
                int id=i*m+j;
                f[id]=id;
            }
        }
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                check(i,j);
            }
        }
        n=n*m;
        int cnt=0;
        for(int i=0;i<n;i++) if(gf(i)==i) cnt++;
        printf("%d\n",cnt);
    }
    return 0;
}