DFS第一题(看别人代码 中国大学mooc)
描述
![](https://i-blog.csdnimg.cn/blog_migrate/a9084503721f5ad6b887dc7a5655bad4.png)
图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<stack>
#include<cstring>
using namespace std;
int roomnum=0,maxroomarea=0,roomarea;
int color[100][100],rooms[100][100];
int r,c;
void Dfs(int i,int j)
{
if(color[i][j])
return;
++roomarea;
color[i][j]=roomnum;
if((rooms[i][j]&1)==0) Dfs(i,j-1);
if((rooms[i][j]&2)==0) Dfs(i-1,j);
if((rooms[i][j]&4)==0) Dfs(i,j+1);
if((rooms[i][j]&8)==0) Dfs(i+1,j);
}
int main()
{
#include<stack>
#include<cstring>
using namespace std;
int roomnum=0,maxroomarea=0,roomarea;
int color[100][100],rooms[100][100];
int r,c;
void Dfs(int i,int j)
{
if(color[i][j])
return;
++roomarea;
color[i][j]=roomnum;
if((rooms[i][j]&1)==0) Dfs(i,j-1);
if((rooms[i][j]&2)==0) Dfs(i-1,j);
if((rooms[i][j]&4)==0) Dfs(i,j+1);
if((rooms[i][j]&8)==0) Dfs(i+1,j);
}
int main()
{
cin>>r>>c;
for(int i=1;i<=r;++i)
for(int j=1;j<=c;++j)
cin>>rooms[i][j];
memset(color,0,sizeof(color));
for(int i=1;i<=r;++i)
for(int j=1;j<=c;++j)
if(!color[i][j])
{
++roomnum;
roomarea=0;
Dfs(i,j);
maxroomarea=max(roomarea,maxroomarea);
}
cout<<roomnum<<endl<<maxroomarea<<endl;
for(int i=1;i<=r;++i)
for(int j=1;j<=c;++j)
cin>>rooms[i][j];
memset(color,0,sizeof(color));
for(int i=1;i<=r;++i)
for(int j=1;j<=c;++j)
if(!color[i][j])
{
++roomnum;
roomarea=0;
Dfs(i,j);
maxroomarea=max(roomarea,maxroomarea);
}
cout<<roomnum<<endl<<maxroomarea<<endl;
return 0;
}
}