PoJ1979 Red and Black (DFS)

题目链接：http://poj.org/problem?id=1979

题意：

　　有一个由正方形的黑色砖块和红色砖块组成的矩形广场，一个人站在广场上的某处，他可以选择往四周(上下左右)的黑色砖块上走，问他一共可以走多少个砖块(加上他本身站立的砖块).

　　"."为黑色砖块

　　"#"为红色砖块

　　"@"为人的起始位置

思路：

　　利用DFS 从人的起始位置开始搜索，搜索的过程中直接统计砖块的数量就好。

代码：

#include <iostream>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <stack>
#include <queue>
#include <algorithm>
#include <string>

typedef long long LL;
using namespace std;
const double PI = acos(-1);
const int MAXN = 20;
int visit[MAXN + 3][MAXN + 3];//标记数组
char map[MAXN + 3][MAXN + 3];
int stepX[] = {-1, 0, 0, 1};
int stepY[] = {0, -1, 1, 0};
int cnt;

void DFS(int x, int y) {
    visit[x][y] = 1;
    for(int i = 0; i < 4; i++) {
        int tx = x + stepX[i], ty = y + stepY[i];
        if(tx >=0 && ty >= 0 && !visit[tx][ty] && map[tx][ty] == '.'){
            cnt++; //计数
            visit[tx][ty] = 1;
            DFS(tx, ty);
        }
    }
}

int main() {
    //freopen("input", "r", stdin);
    int m = -1, h = -1;
    while(scanf("%d%d", &m, &h), m || h) {
        memset(visit, 0, sizeof(visit));
        memset(map, 0, sizeof(map));
        for(int i = 0; i < h; i++) scanf("%s", map[i]);
        int stX = -1, stY = -1; //起始坐标
        for(int i = 0; i < h; i++) {
            for(int j = 0; j < m; j++) {
                if(map[i][j]== '@') stX = i, stY = j; 
            }
        }
        cnt = 1;
        DFS(stX, stY);//从起始坐标开始搜索
        printf("%d
", cnt);
    }
    return 0;
}