2020网易笔试第四题

题目描述

英雄与怪兽。更新二维数组中的值
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
 */