百练2815:城堡问题（DFS）

描述

     1   2   3   4   5   6   7  
   #############################
 1 #   |   #   |   #   |   |   #
   #####---#####---#---#####---#
 2 #   #   |   #   #   #   #   #
   #---#####---#####---#####---#
 3 #   |   |   #   #   #   #   #
   #---#########---#####---#---#
 4 #   #   |   |   |   |   #   #
   #############################
           (图 1)

   #  = Wall   
   |  = No wall
   -  = No wall

图1是一个城堡的地形图。请你编写一个程序，计算城堡一共有多少房间，最大的房间有多大。城堡被分割成mn(m≤50，n≤50)个方块，每个方块可以有0~4面墙。
输入程序从标准输入设备读入数据。第一行是两个整数，分别是南北向、东西向的方块数。在接下来的输入行里，每个方块用一个数字(0≤p≤50)描述。用一个数字表示方块周围的墙，1表示西墙，2表示北墙，4表示东墙，8表示南墙。每个方块用代表其周围墙的数字之和表示。城堡的内墙被计算两次，方块(1,1)的南墙同时也是方块(2,1)的北墙。输入的数据保证城堡至少有两个房间。输出城堡的房间数、城堡中最大房间所包括的方块数。结果显示在标准输出设备上。
样例输入

4 
7 
11 6 11 6 3 10 6 
7 9 6 13 5 15 5 
1 10 12 7 13 7 5 
13 11 10 8 10 12 13 

样例输出

5
9

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
using namespace std;
int n, m;//行数，列数
int room[51][51];//保存房间信息
int color[51][51];//标记数组，用来检查一个房间是否已经走过
int roomNum = 0, maxArea = 0;//总的房间数和最大房间面积
int area = 0;//当前正在走的房间的面积
void dfs(int i,int j)//对当前位置做dfs，会得到该房间所在的联通分量
{
    //cout << "dfs(" << i << "," << j << ")" << endl;
    if (color[i][j])//为旧点
    {
        //cout << "(" << i << "," << j << ")是旧点" << endl;
        return;
    }
    //cout << "(" << i << "," << j << ")不是旧点" << endl;
    area++;
    //标记为旧点
    color[i][j] = roomNum;
    //按西北东南的顺序做dfs，本题不用考虑边界问题，因为输入保证了四周都有墙，dfs一定会在到达边界的时候停下来
    if (!(room[i][j] & 1))//西边没有墙往西走
    {
        //cout << "(" << i << "," << j << ")西边没有墙往西走" << endl;
        dfs(i, j - 1);
    }

    if (!(room[i][j] & 2))//北边没有墙往北走
    {
        //cout << "(" << i << "," << j << ")北边没有墙往北走" << endl;
        dfs(i - 1, j);
    }

    if (!(room[i][j] & 4))//东边没有墙往东走
    {
        //cout << "(" << i << "," << j << ")东边没有墙往东走" << endl;
        dfs(i, j + 1);
    }

    if (!(room[i][j] & 8))//南边没有墙往南走
    {
        //cout << "(" << i << "," << j << ")南边没有墙往南走" << endl;
        dfs(i + 1, j);
    }

}
int main()
{
    cin >> n >> m;
    memset(room,0,sizeof(room));
    memset(color, 0, sizeof(color));
    for (int i = 1; i <= n; ++i)
        for (int j = 1; j <= m; ++j)
            cin >> room[i][j];

    for (int i = 1; i <= n; ++i)
        for (int j = 1; j <= m; ++j)
        {
            if (!color[i][j])//这个房间没有被走过
            {
                roomNum++;
                area = 0;
                dfs(i,j);
                maxArea = max(maxArea,area);
            }
        }
    cout << roomNum  << endl;
    cout << maxArea  << endl;
    return 0;
}