Dfs之城堡问题(POJ 2815)

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

这个代码的技巧多一点，可以学到更多的东西

#include <iostream>
#include <string.h>
#include <stdio.h>
using namespace std;
int **nData;
int **nMark;
int m,n;
bool is(int x,int y)
{
    if(x>=0&&x<m&&y>=0&&y<n)
        return true;
    else
        return false;
    
}
void mark(int x,int y,int nID){
    if(!is(x,y))             //由于我是数组零开始的所以必须要bool判断一下，否则越界。。
        return ;
    if(nMark[x][y]) return;	//already marked
    nMark[x][y]=nID;
    if((nData[x][y]&0x01)==0)	//goto left
    {mark(x,y-1,nID);	}
    if((nData[x][y]&0x02)==0)	//goto up
    {mark(x-1,y,nID);	}
    if((nData[x][y]&0x04)==0)	//goto right
    {mark(x,y+1,nID);	}
    if((nData[x][y]&0x08)==0)	//goto down
    {mark(x+1,y,nID);	}
}
void work(){
    int i,j;
    scanf("%d%d",&m,&n);//m--row, n--col
    nData=new int * [m];//m row 的一位数组
    nMark=new int * [m];
    for(i=0;i<m;i++){
        nData[i]=new int [n];
        for(j=0;j<n;j++) scanf("%d",&nData[i][j]);
        nMark[i]=new int [n];
        memset(nMark[i],0,sizeof(int)*n);
    }
    int curID=1;	//the first room id is 1
    for(i=0;i<m;i++){
        for(j=0;j<n;j++){
            if(nMark[i][j]==0)
            {mark(i,j,curID);curID++;}
        }
    }
    #ifdef debug
    for(int i=0;i<m;++i)
    {
        for(int j=0;j<n;++j)
            cout<<nMark[i][j];
        cout<<endl;
    }
    #endif
    int *nCount = new int [curID];
    memset(nCount,0,sizeof(int)*curID);
    for(i=0;i<m;i++){
        for(j=0;j<n;j++)
        {
            nCount[nMark[i][j]]++;//count every room size}
        }
    }
      #ifdef debug
        for(int i=1;i<curID;i++)
            cout<<nCount[i]<<endl;
       #endif
        int nMax = -1;
        for(i=1;i<curID;i++)
        {if(nCount[i]>nMax) nMax=nCount[i];}
        printf("%d\n%d\n",curID-1,nMax);
        for(i=0;i<m;i++)
        {  delete[]nData[i];    delete[]nMark[i];           }
        delete[]nCount;  delete[]nData;  delete[]nMark;
}
int main()
    {
    work();
    return 0;
}

下面这一种，简洁明快，数组是从1开始的，所以在碰边界的时候不会有越界问题，不会Runtime error!

#include <iostream>
#include <stack>
#include <cstring>
using namespace std;
int R,C;  //行列数
int rooms[60][60];
int color[60][60]; //房间是否染色过的标记
int maxRoomArea = 0, roomNum = 0;
int roomArea;
void Dfs(int i,int k) {
	if( color[i][k] )  	
		return;
	++ roomArea;
	color [i][k] = roomNum;
	if( (rooms[i][k] & 1) == 0 ) Dfs(i,k-1);   //向西走
	if( (rooms[i][k] & 2) == 0 ) Dfs(i-1,k);   //向北 
	if( (rooms[i][k] & 4) == 0 ) Dfs(i,k+1);  //向东
	if( (rooms[i][k] & 8) == 0 ) Dfs(i+1,k);  //向南
}
int main()  {
	cin >> R >> C;
	for( int i = 1;i <= R;++i)
		for ( int k = 1;k <= C; ++k)
			cin >> rooms[i][k];
	memset(color,0,sizeof(color));
	for( int i = 1;i <= R; ++i)
		for( int k = 1; k <= C; ++ k) {
			if( !color[i][k] ) {
				++ roomNum ;   	roomArea = 0; 
				Dfs(i,k); 
				maxRoomArea = max(roomArea,maxRoomArea);
			}
		}
	cout << roomNum << endl;
	cout << maxRoomArea << endl;
}