HDOJ 1198 Farm Irrigation

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1198

题目大意：有一块n×m大小的农田，被分隔成许多小的正方形，每个正方形中都装有水管，一共有11种水管，分别用字母A~K标明。正方形的中心可以有水源，水可以沿着水管流到其他的正方形上，问至少需要多少个水源，保证农田全部被灌溉。

                         

题目思路：顺着水管连通的方向进行搜索，一次搜索需要一个水源，搜索过的地方标记，最后搜索的次数就是水源的个数。

AC代码：

//  HDOJ
//  1198.cpp
/*
    ID: Firwaless
    LANG: C++
    TASK: Farm Irrigation
*/

#include <cstdio>
#include <cstring>

/* 上下左右四个方向，注意顺序 */
const int dir[4][2] = {-1, 0, 1, 0, 0, -1, 0, 1};

/* 每个水管在上下左右四个方向上是否有出口，按照字母ASCII码大小依次打表，注意顺序 */
const bool rel[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[52][52];

void dfs(int x, int y)
{
    int i, a, b;
    char c = g[x][y];
    
    g[x][y] = '*';
    
    /* 搜索四个方向 */
    for (i = 0; i < 4; i++)
    {
        a = x + dir[i][0];
        b = y + dir[i][1];
        
        /* 先判断边界和是否遍历过，再判断水管的当前方向和要搜索的下一个水管的相反方向是否连通 */
        if (g[a][b] != '*' && rel[c - 'A'][i] && rel[g[a][b] - 'A'][i ^ 1])
        {
            dfs(a, b);
        }
    }
}

int main()
{
    int i, j, n, m, ans;
    
    while (scanf("%d%d%*c", &n, &m), n + m + 2)
    {
        ans = 0;
        memset(g, '*', sizeof(g));
        for (i = 1; i <= n; i++)
        {
            for (j = 1; j <= m; j++)
            {
                g[i][j] = getchar();
            }
            getchar();
        }
        for (i = 1; i <= n; i++)
        {
            for (j = 1; j <= m; j++)
            {
                if (g[i][j] != '*')
                {
                    ans++;
                    dfs(i, j);
                }
            }
        }
        printf("%d\n", ans);
    }
    return 0;
}