题目描述:
有一间长方形的房子,地上铺了红色、黑色两种颜色的正方形瓷砖。 你站在其中一块黑色的瓷砖上,只能向相邻(上下左右四个方向)的黑色瓷砖移动。
请写一个程序,计算你总共能够到达多少块黑色的瓷砖。输入格式:
输入包括多个数据集合。 每个数据集合的第一行是两个整数 W和 H,分别表示 x 方向和 y 方向瓷砖的数量。 在接下来的 H 行中,每行包括 W个字符。每个字符表示一块瓷砖的颜色,规则如下:
1)‘.’:黑色的瓷砖; 2)‘#’:红色的瓷砖; 3)‘@’:黑色的瓷砖,并且你站在这块瓷砖上。该字符在每个数据集合中唯一出现一次。
当在一行中读入的是两个零时,表示输入结束。输出格式:
对每个数据集合,分别输出一行,显示你从初始位置出发能到达的瓷砖数(记数时包括初始位置的瓷砖)。
数据范围:
1≤W,H≤20
输入样例:
输出样例:
45
题目为Flood Fill算法,可以用bfs或dfs求解。
一、bfs
伪代码:
while 队列非空
{
取出对头元素
枚举队头元素的4个邻格
if(格子是陆地并且未被开发)
标记当前格子为已开发
将新格插入队列
}
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Queue;
import java.util.LinkedList;
public class Main{
static int m,n;
static char[][] g = new char[25][25];//保存字符
static Queue<int[]> q = new LinkedList<>();//队列存储当前坐标
public static void main(String[] args)throws Exception{
BufferedReader read = new BufferedReader(new InputStreamReader(System.in));
while(true){
String[] str1 = read.readLine().split(" ");
m = Integer.parseInt(str1[0]);
n = Integer.parseInt(str1[1]);
if(m == 0 || n == 0) break;
int x = 0,y = 0;//表示人所在位置的坐标
for(int i = 0;i < n;i++){
String str2 = read.readLine();
for(int j = 0; j < m; j++){
g[i][j] = str2.charAt(j);
if(g[i][j] == '@'){
x = i;
y = j;
}
}
}
System.out.println(bfs(x,y));
}
}
public static int bfs(int x,int y){
q.add(new int[]{x,y});//把当前坐标加入到队列
g[x][y] = '#'; //把当前坐标设为#,表示已经走过
int res = 0;
int dx[] = {-1,0,1,0}; int dy[] = {0,1,0,-1};
while(q.size() != 0){
int[] poll = q.poll(); //把当前坐标出队
res++;
for(int i = 0;i<4;i++){
int a = poll[0] + dx[i], b = poll[1] + dy[i]; //表示当前坐标向上右下左走
//如果走出这一步出了边界,或者遇到重复元素,就跳过此次循环
if(a < 0 || a >= n || b < 0 || b >= m ||g[a][b] != '.') continue;
g[a][b] = '#';
q.add(new int[]{a,b});
}
}
return res;
}
}
二、
伪代码:
dfs(x,y)
{
将(x,y)标记为已开发
枚举(x,y)的4个邻格
if(邻格可走)
dfs(邻格)
}
import java.io.BufferedReader;
import java.io.InputStreamReader;
public class Main{
static int m,n;
static char[][] g = new char[25][25];//保存字符
static int[] dx = {-1,0,1,0};
static int[] dy = {0,1,0,-1};
public static void main(String[] args)throws Exception{
BufferedReader read = new BufferedReader(new InputStreamReader(System.in));
while(true){
String[] str1 = read.readLine().split(" ");
m = Integer.parseInt(str1[0]);
n = Integer.parseInt(str1[1]);
if(m == 0 || n == 0) break;
int x = 0,y = 0;//表示人所在位置的坐标
for(int i = 0;i < n;i++){
String str2 = read.readLine();
for(int j = 0; j < m; j++){
g[i][j] = str2.charAt(j);
if(g[i][j] == '@'){
x = i;
y = j;
}
}
}
System.out.println(dfs(x,y));
}
}
public static int dfs(int x,int y){
int res = 1;//刚开始的第一个格也算
g[x][y] = '#';
for(int i = 0;i < 4;i++){
int a = x + dx[i];
int b = y + dy[i];
if(a >= 0 && a < n && b >=0 && b < m && g[a][b] == '.')
res += dfs(a,b);
}
return res;
}
}