程序设计与算法慕课018:红与黑(C++动态规划)

题目描述

有一间长方形的房子,地上铺了红色、黑色两种颜色的正方形瓷砖。你站在其中一块黑色的瓷砖上,只能向相邻的黑色瓷砖移动。请写一个程序,计算你总共能够到达多少块黑色的瓷砖。

输入

包括多个数据集合。每个数据集合的第一行是两个整数W和H,分别表示x方向和y方向瓷砖的数量。W和H都不超过20。在接下来的H行中,每行包括W个字符。每个字符表示一块瓷砖的颜色,规则如下
1)‘.’:黑色的瓷砖;
2)‘#’:红色的瓷砖;(原网站上这里写成了白色)
3)‘@’:黑色的瓷砖,并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。
当在一行中读入的是两个零时,表示输入结束。

输出

对每个数据集合,分别输出一行,显示你从初始位置出发能到达的瓷砖数(记数时包括初始位置的瓷砖)。

样例输入

6 9 
....#. 
.....# 
...... 
...... 
...... 
...... 
...... 
#@...# 
.#..#. 
0 0

样例输出

45

代码

这道题是比较简单的DFS问题,按照老师给的DFS结构,找清楚遍历的条件就可以了。

#include <iostream>
#include <cstring>
#include <string.h>
using namespace std;
int W, H; int sum = 0;
char tfloor[21][21]; //存放地板数据
char label[21][21];  //标记是否走过
void dfs(int m, int n)
{
    
    
	if (label[m][n] == 1) //m为行,n为列
		return;
	//如果这块砖没有走过,那么现在走过了
	label[m][n] = 1; sum += 1; 
	//如果没到上边界且上面一个是'.',就一直往上走。其余三个换了个方向也是同理
	if (m != 0 && tfloor[m - 1][n] == '.') dfs(m - 1, n); 
	if (m != H - 1 && tfloor[m + 1][n] == '.') dfs(m + 1, n);
	if (n != 0 && tfloor[m][n - 1] == '.') dfs(m, n - 1);
	if (n != W - 1 && tfloor[m][n + 1] == '.') dfs(m, n + 1);
}
int main()
{
    
    
	while (cin >> W >> H) 
	{
    
    
		if(W==0 && H==0)
			break;
		int bw = 0, bh = 0; //起点位置
		for (int i = 0; i < H; i++) {
    
    
			for (int j = 0; j < W; j++) {
    
    
				cin >> tfloor[i][j];
				if (tfloor[i][j] == '@') {
    
     //找到起点
					bw = i; bh = j;
					tfloor[i][j] = '.';
				}
			}
		}
		memset(label, 0, sizeof(label));//将所有标记置零
		dfs(bw, bh); 
		cout << sum << endl;
		sum = 0;
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_43390123/article/details/117527417