Red and Black (找到一个标记一个)

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

题目大意：

有一个长方形的房间，上面铺着正方形的瓷砖。每块瓷砖都是红色或黑色的。一个男人站在一块黑色的瓷砖上。从一个瓦片，他可以移动到四个相邻瓦片中的一个。但是他不能在红瓦上移动，他只能在黑瓦上移动。

编写一个程序，通过重复上面描述的动作来计算他可以到达的黑色方块的数量。

输入

输入由多个数据集组成。数据集以包含两个正整数W和H的行开始;W和H分别是x和y方向上的瓦片数。W和H不超过20。

'.——一块黑瓷砖

“#”——红色瓷砖

@——一个男人在黑色的平铺上(数据集中只出现一次）

思路：

找到起始位置然后进行 dfs，找到一个就标记一个，然后一直走下去

一开始自己想错了，我以为是找最多的可以走的。但是后来感觉根据题目的意思来看就是找到一条路就可以了。哭泣

AC代码：

#include <cstdio>
#include <string>
#include <iostream>
#include <algorithm>
#include <string.h>
#include <math.h>
#include <vector>

using namespace std;

const int maxn = 30;

char ss[maxn][maxn];
int mx[] = {0,0,1,-1};
int my[] = {1,-1,0,0};
int vis[maxn][maxn];
int n,m;
int cnt = 0;

bool check(int i,int j)
{
    if (ss[i][j] == '.' && !vis[i][j])
        return true;
    return false;
}

void dfs(int row,int col)
{
    for (int i=0;i<4;i++)
    {
        int nx = row + mx[i];
        int ny = col + my[i];
        if (check(nx,ny))
        {
            cnt++;
            vis[nx][ny] = 1;
            dfs(nx,ny);
        }
    }

}


int main() {
    while (cin >> m >> n) {
        if (m == 0 && n == 0)
            return 0;
        bool flag = false;
        cnt = 0;
        memset(vis,0, sizeof(vis));
        memset(ss,'#', sizeof(ss));
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= m; j++) {
                cin >> ss[i][j];
            }
        }
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= m; j++) {
                if (ss[i][j] == '@') {
                    dfs(i, j);
                    printf("%d
", cnt + 1);
                    flag = true;
                    break;
                }
            }
            if (flag)
                break;
        }
    }
}