题目描述
英雄与怪兽。更新二维数组中的值
n*m的地图上,每个位置值为0或1,0表示怪兽,1表示英雄,对于每个英雄求出离他最近的怪兽的距离是多少,矩阵中每个位置离上下左右的距离都是1,若当前位置是怪兽,输出0即可,题目保证至少存在一只怪兽
输入:第一行输入两个正整数 ,n,m
接下来n行,每行输入m个数,从{0,1}中取值。表示每个位置的属性,其中1<=m,n<=1000
例子:
3 3
1 0 1
0 1 0
1 0 1
输出更新后的矩阵。输出n行,每行m个用空格分隔的数字表示答案。注意行末没有空格
例子
1 0 1
0 1 0
1 0 1
解题思路
1、深度优先遍历,返回四个方向的最小值
2、用一个数组tmp标记是否访问过
3、如果访问到边界值或者已经访问过的,则将路径长度置为n*m
自测代码(不知道能不能100%AC)
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int m = sc.nextInt();
int[][] num = new int[n][m];
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
num[i][j]=sc.nextInt();
}
}
int[][] tmp = new int[n][m];
for(int i=0;i<n;i++){
for(int j=0;j<m;j++){
if (num[i][j]==0){
System.out.print(0);
}else{
System.out.print(find(num,i,j,n,m,tmp));
}
if (j==m-1){
System.out.println();
}else{
System.out.print(" ");
}
}
}
System.out.println();
}
public static int find(int[][] num,int i,int j,int n,int m,int[][] tmp){
if ((i<0)||(j<0)||(i>=n)||(j>=m)||tmp[i][j]==1) return 1000;
else if (num[i][j]==0) return 0;
else{
tmp[i][j]=1;
int num1=1+find(num,i+1,j,n,m,tmp);
int num2=1+find(num,i-1,j,n,m,tmp);
int num3=1+find(num,i,j+1,n,m,tmp);
int num4=1+find(num,i,j-1,n,m,tmp);
tmp[i][j]=0;
return Math.min(num1,Math.min(num2,Math.min(num3,num4)));
}
}
}
/*
3 3
1 0 1
0 1 0
1 0 1
3 3
1 1 1
1 1 1
1 1 0
3 3
1 1 1
1 0 1
1 1 1
*/