HDOJ1198 Farm Irrigation(DFS)

题目链接：点击打开链接

题意：

Benny 有一大片农田需要灌溉。农田是一个长方形，被分割成许多小的正方形。每个正方形 中都安装了水管。不同的正方形农田中可能安装了不同的水管。一共有 11 种水管，分别用字母 A～ K标明，如图 2.11(a)所示。  

Benny 农田的地图是由描述每个正方形农田中水管类型的字母组成的矩阵。 某些正方形农田的中心有水源，因此水可以沿着水管从一个正方形农田流向另一个正方形农 田。如果水可以流经某个正方形农田，则整个正方形农田可以全部灌溉到。 Benny想知道至少需要多少个水源，以保证整个长方形农田都能被灌溉到。

题解：

DFS求连通块。只需更增加一个判断条件即可。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAX=55;
int  vis[MAX][MAX];
char g[MAX][MAX];
int n,m;
struct part
{
    int l,r,u,d;
    part(){};
    part(int l,int r,int u,int d):l(l),r(r),u(u),d(d){};
};
part p[12];
int dirx[4]={0,1,0,-1};
int diry[4]={1,0,-1,0};
void init()
{
    p[0]=part(1,0,1,0);
    p[1]=part(0,1,1,0);
    p[2]=part(1,0,0,1);
    p[3]=part(0,1,0,1);
    p[4]=part(0,0,1,1);
    p[5]=part(1,1,0,0);
    p[6]=part(1,1,1,0);
    p[7]=part(1,0,1,1);
    p[8]=part(1,1,0,1);
    p[9]=part(0,1,1,1);
    p[10]=part(1,1,1,1);
}
int judgein(int x,int y)
{
    return x>=0&&x<n&&y>=0&&y<m;
}
int judgecon(int x1,int y1,int x2,int y2,int i)
{
    int a=g[x1][y1]-'A';
    int b=g[x2][y2]-'A';
    if(i==0) return p[a].r==p[b].l&&p[b].l==1;
    else if(i==1) return p[a].d==p[b].u&&p[b].u==1;
    else if(i==2) return p[a].l==p[b].r&&p[b].r==1;
    else if(i==3) return p[a].u==p[b].d&&p[b].d==1;
}
void dfs(int x,int y)
{
    vis[x][y]=1;
    for(int i=0;i<4;i++)
    {
        int nx=x+dirx[i];
        int ny=y+diry[i];
        if(judgein(nx,ny)&&!vis[nx][ny]&&judgecon(x,y,nx,ny,i))
        {
            vis[nx][ny]=1;
            dfs(nx,ny);
        }
    }
    return ;
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    init();
    while(scanf("%d%d",&n,&m))
    {
        if(n==-1&&m==-1) break;
        getchar();
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                scanf("%c",&g[i][j]);
            }
            getchar();
        }
        int cou=0;
        memset(vis,0,sizeof(vis));
        for(int i=0;i<n;i++)
        {
            for(int j=0;j<m;j++)
            {
                if(!vis[i][j])
                {
                    dfs(i,j);
                    cou++;
                }
            }
        }
        printf("%d\n",cou);
    }
    return 0;
}