POJ 2815 城堡问题

题目

总时间限制: 1000ms 内存限制: 65536kB
描述

 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
来源
1164

思路

dfs。对每个房间进行染色，遍历所有结点，寻找最大连通子图。

代码

roomArea = 0
colorNum = 0
R = 0
C = 0
def dfs(i, j, rooms, color):
    global R
    global C
    if i == -1 or j == -1 or i == R or j == C:
        return
    if color[i][j]:
        return
    global roomArea
    roomArea += 1
    global colorNum
    color[i][j] = colorNum
    if (rooms[i][j] & 1) == 0: dfs(i, j - 1, rooms, color)
    if (rooms[i][j] & 2) == 0: dfs(i - 1, j, rooms, color)
    if (rooms[i][j] & 4) == 0: dfs(i, j + 1, rooms, color)
    if (rooms[i][j] & 8) == 0: dfs(i + 1, j, rooms, color)
while True:
    try:
        R = int(input().strip())
        C = int(input().strip())
        rooms = []
        color = []
        for i in range(R):
            rooms.append([int(i) for i in input().strip().split()])
            color.append([0 for i in range(C)])
        colorNum = 0
        maxRoomArea = 0
        for i in range(R):
            for j in range(C):
                if color[i][j] == 0:
                    colorNum += 1
                    roomArea = 0
                    dfs(i, j, rooms, color)
                    maxRoomArea = max(maxRoomArea, roomArea)
        print(colorNum)
        print(maxRoomArea)
    except:
        break