DFS红与黑

1:红与黑
查看 提交 统计 提问
总时间限制: 1000ms 内存限制: 65536kB
描述
有一间长方形的房子，地上铺了红色、黑色两种颜色的正方形瓷砖。你站在其中一块黑色的瓷砖上，只能向相邻的黑色瓷砖移动。请写一个程序，计算你总共能够到达多少块黑色的瓷砖。
输入
包括多个数据集合。每个数据集合的第一行是两个整数W和H，分别表示x方向和y方向瓷砖的数量。W和H都不超过20。在接下来的H行中，每行包括W个字符。每个字符表示一块瓷砖的颜色，规则如下
1）‘.’：黑色的瓷砖；
2）‘#’：白色的瓷砖；
3）‘@’：黑色的瓷砖，并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。
当在一行中读入的是两个零时，表示输入结束。
输出
对每个数据集合，分别输出一行，显示你从初始位置出发能到达的瓷砖数(记数时包括初始位置的瓷砖)。
样例输入
6 9
….#.
…..#
……
……
……
……
……

@…#

.#..#.
0 0
样例输出
45

解析：
这道题就是很简单的一个求极大连通子图的节点个数

#include <iostream>
#include <cstring>
using namespace std;
void dfs(int i, int j);

int graph[25][25];
int color[25][25];
int w;
int h;
int origin_i;
int origin_j;
int area;

int main(int argc, const char * argv[]) {
    char a;

    while (true) {
        cin >> w >> h;
        if((w == 0)&&(h ==0))
            break;
        else{
            memset(graph, 0, sizeof(graph));
            memset(color, 0, sizeof(color));
            area = 0;
            for(int i = 1; i <= h; ++i)
                for(int j = 1; j <= w; ++j){
                    cin >> a;
                    if(a == '.' )
                        graph[i][j] = 1;
                    else if(a == '#')
                        graph[i][j] = 0;
                    else{
                        graph[i][j] = 1;
                        origin_i = i;
                        origin_j = j;
                        //origin_i,origin_j为起点
                    }
                }
            dfs(origin_i, origin_j);
            cout << area << endl;
        }
    }
    return 0;
}


void dfs(int i, int j){
    if(color[i][j] == 1)
        return;
    area ++;
    color[i][j] = 1;
    if( graph[i-1][j] == 1)
        dfs(i-1,j);
    if( graph[i+1][j] == 1)
        dfs(i+1,j);
    if( graph[i][j-1] == 1 )
        dfs(i,j-1);
    if( graph[i][j+1] == 1)
        dfs(i,j+1);
}