DFS 城堡问题百练2815

输入
程序从标准输入设备读入数据。第一行是两个整数，分别是南北向、东西向的方块数。在接下来的输入行里，每个方块用一个数字(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<bits/stdc++.h>
using namespace std;

int c=0,room[51][51]={0},color[51][51]={0},m,n,square[2500]={0},maxs=0;
void f(int i,int j,char sf)
{
	if(i<=0||i>m||j<=0||j>n) return ;
	if(color[i][j]==1){
		 return ;
	}
	color[i][j]=1;
	square[c]++;
	if(sf!='w'&&!(room[i][j]&1)) f(i,j-1,'e');//no west wall
	if(sf!='n'&&!(room[i][j]&2)) f(i-1,j,'s');
	if(sf!='e'&&!(room[i][j]&4)) f(i,j+1,'w');
	if(sf!='s'&&!(room[i][j]&8)) f(i+1,j,'n');
	if(sf=='t') {
		maxs=max(maxs,square[c]);
		++c;	
	}
}

int main()
{
	scanf("%d%d",&m,&n);
	for(int i=1;i<=m;i++)
	for(int j=1;j<=n;j++)
		scanf("%d",&room[i][j]);
	for(int i=1;i<=m;i++)
	for(int j=1;j<=n;j++)
	f(i,j,'t');
	printf("%d\n%d\n",c,maxs);
	return 0;
}

#include<bits/stdc++.h>
using namespace std;

int c=0,room[51][51]={0},color[51][51]={0},m,n,square=0,maxs=0;
void f(int i,int j)
{
	if(i<=0||i>m||j<=0||j>n) return ;
	if(color[i][j]) return ;
	color[i][j]=1;
	square++;
	if(!(room[i][j]&1)) f(i,j-1);
	if(!(room[i][j]&2)) f(i-1,j);
	if(!(room[i][j]&4)) f(i,j+1);
	if(!(room[i][j]&8)) f(i+1,j);
}

int main()
{
	scanf("%d%d",&m,&n);
	for(int i=1;i<=m;i++)
	for(int j=1;j<=n;j++)
		scanf("%d",&room[i][j]);
	for(int i=1;i<=m;i++)
	for(int j=1;j<=n;j++)
	{
		if(!color[i][j])
		{
			c++;
			square=0;
			f(i,j);
			maxs=max(maxs,square);
		}
	}
	printf("%d\n%d\n",c,maxs);
	return 0;
}