深搜：POJ城堡问题

深搜：POJ城堡问题

问题描述

• 描述

1 2 3 4 5 6 7
#############################
1 # | # | # | | #
#####—#####—#---#####—#
2 # # | # # # # #
#—#####—#####—#####—#
3 # | | # # # # #
#—#########—#####—#---#
4 # # | | | | # #
#############################
(图 1)
#= Wall
| = No wall
- = No wall
(图坏了，去poj里看)

图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

问题分析

• 深搜问题的范式

Dfs(v) {
	if( v 访问过)
		return;
	将v标记为访问过;
	对和v相邻的每个点u: Dfs(u);
}
int main() {
	while(在图中能找到未访问过的点 k)
		Dfs(k);
}

• 目前问题
  对每一个方块都进行深度搜索，将搜索到的方块进行颜色标记，标记过的方块跳过。
  最后统计颜色（房间数量）以及数量最多的颜色（最大的房间占据的方块数）

• C++代码

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

int C, R;
int room[60][60];
int color[60][60];
int max_room_size = 0;
int room_num = 0;
int room_size = 0;

void dfs(int i, int j){
    if(color[i][j]){
        return;
    }
    
    room_size++;
    color[i][j] = room_num;
    if((room[i][j]&1) == 0){
        dfs(i, j-1);  // 西墙
    }
    if((room[i][j]&2) == 0){
        dfs(i-1, j);  // 北墙
    }
    if((room[i][j]&4) == 0){
        dfs(i, j+1);  // 东墙
    }
    if((room[i][j]&8) == 0){
        dfs(i+1, j);  // 南墙 
    }

}

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

    for(int i=0;i<R;i++){
        for(int j=0;j<C;j++){
            if(!color[i][j]){
                room_num++;
                room_size = 0;
                dfs(i, j);
                max_room_size = max(room_size, max_room_size);
            }
        }
    }

    cout << room_num << endl;
    cout << max_room_size << endl;
    
    return 0;
}