实现DFS之“农田灌溉”

  • Post author:
  • Post category:其他

这也是一道利用了DFS的题目,先说下我的思路:用一个二维数组记录每个字母所代表的含义(管道方向),用另一个二维数组记录4个方向的变换坐标;随后利用经典的DFS递归遍历即可~(还要注意在方向的处理上前后要保持一致,否则很容易计算出错 :| )


农田灌溉(Farm Irrigation)   

题目描述: 

Benny有一大片农田需要灌溉。农田是一个长方形,被分割成许多小的正方形。每个正方形中都安装了水管。不同的正方形农田中可能安装了不同的水管。一共有11种水管,分别用字母A~K标明,如图1所示。 


Benny农田的地图是由描述每个正方形农田中水管类型的字母组成的矩阵。例如,如果农田的地图为: ADC FJK IHE 

则农田中水管分布如图2所示。 


某些正方形农田的中心有水源,因此水可以沿着水管从一个正方形农田流向另一个正方形农田。如果水可以流经某个正方形农田,则整个正方形农田可以全部灌溉到。 Benny想知道至少需要多少个水源,以保证整个农田都能被灌溉到? 例如,图2所示的农田至少需要3个水源,图中的圆点表示每个水源。  
输入描述:  
输入文件中包含多个测试数据。每个测试数据的第1行为两个整数M和N,表示农田中有M行,每行有N个正方形。接下来有M行,每行有N个字符。字符的取值为’A’~’K’,表示对应正方形农田中水管的类型。当M或N取负值时,表示输入文件结束。如果M和N的值为正数,则其取值范围是1≤M, N≤50。  
输出描述:  
对输入文件中的每个测试数据所描述的农田,输出占一行,为求得的所需水源数目的最小值。  

样例输入:  

2 2

DK

HF

3 3

ADC

FJK

IHE

-1 -1

样例输出: 

2 3

闲话少叙,直接上代码+注释~

Code:

#include<iostream>
using namespace std;

#define MAXN 50
char map[MAXN][MAXN];	//地图
int M, N;				//M:农田行数,N:每行的正方形个数

int dir[4][2] = { {-1, 0}, {1, 0}, {0, -1}, {0, 1} };	//管道坐标变换的四个方向:对应的顺序上下左右。
int dirO[11][4] = {
	{-1, 0, -1, 0}, {-1, 0, 0, 1}, {0, 1, -1, 0},
	{0, 1, 0, 1}, {-1, 1, 0, 0}, {0, 0, -1, 1},
	{-1, 0, -1, 1}, {-1, 1, -1, 0}, {0, 1, -1, 1},
	{-1, 1, 0, 1}, {-1, 1, -1, 1}
	};		//每种农田块对应的4个管道方向(-1为左或上、1为右或下、0为没有)。

void DFS(int x, int y)
{
	int xx, yy;
	char temp = map[x][y];		//暂存当前农田块的形状,便于后面的比较~
	map[x][y] = 'Y';			//覆盖原来农田为已遍历
	for(int i = 0; i < 4; i++)
	{
		//计算下一个方向的坐标。
		xx = x + dir[i][0];		
		yy = y + dir[i][1];
		if(xx<0 || xx>=MAXN || yy<0 || yy>=MAXN) continue;		//越界,跳至下一个方向的判断
		
		//方向所代表的值:上0、下1、左2、右3。
	
		//如果向上走(0)或向左走(2),比如向左走则判断原坐标的“左”管道和下一个方向坐标的“右”管道,方向所代表的值+1
		if(i == 0 || i == 2)	
		{//若相邻两块农田管道能“接上”,即管道方向对应的值之积为-1
			if(map[xx][yy]!='Y' && dirO[temp-'A'][i]*dirO[map[xx][yy]-'A'][i+1]==-1) DFS(xx, yy);
		}
		//否则为向下走(1)或向右走(3),比如向下走则判断原坐标的“下”管道和下一个方向坐标的“上”管道。对应方向所代表的值-1
		else if(map[xx][yy]!='Y' && dirO[temp-'A'][i]*dirO[map[xx][yy]-'A'][i-1]==-1) DFS(xx, yy);
	}
}
int main()
{
	int i, j;
	int count;
	while(scanf("%d%d", &M, &N))
	{
		count = 0;
		if(M < 0 || N < 0) break;
		for(i = 0; i < M; i++) scanf("%s", map[i]);			//以行为单位为map地图赋值
		//遍历地图、递归计数
		for(i = 0; i < M; i++)					
			for(j = 0; j < N; j++)
				if(map[i][j] != 'Y') { DFS(i, j); count++; }
		printf("%d\n", count);
	}
	return 0;
}

运行结果:


如有Bug,欢迎拍砖~ :)


版权声明:本文为u012904198原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。