ACM编程 计算疫情扩散时间【Java】

题目描述

在一个地图中(地图由n*n个区域组成），有部分区域被感染病菌。 感染区域每天都会把周围（上下左右）的4个区域感染。 请根据给定的地图计算，多少天以后，全部区域都会被感染。 如果初始地图上所有区域全部都被感染，或者没有被感染区域，返回-1

输入描述

一行N*N个数字（只包含0,1，不会有其他数字）表示一个地图，数字间用,分割，0表示未感染区域，1表示已经感染区域 每N个数字表示地图中一行，输入数据共表示N行N列的区域地图。

例如输入1,0,1,0,0,0,1,0,1，表示地图

1,0,1 0,0,0 1,0,1

输出描述

一个整数，表示经过多少天以后，全部区域都被感染 1<=N<200

public static void main(String[] args){
    Scanner sc = new Scanner(System.in);
    String str =  sc.nextLine();

    //判断是否全部感染
    int num = 0;
    if(!str.contains("1")){
        //无感染
        System.out.println(-1);
    }else{
        //有感染
        String[] strArr = str.split(",");
        int n = (int)Math.sqrt(strArr.length);
        int[][] arr = new int [n][n];;

        for (int i=0;i<strArr.length;i++){
            int rowIndex = i/n;
            int columnIndex = i%n;
            arr[rowIndex][columnIndex] = Integer.parseInt(strArr[i]);
        }

        while (!isCheck(arr)){
            arr = gr(arr,n);
            num ++;
        }
    }


    System.out.println(num);


}

/**
 * 判断是否全部感染
 * 1,0,1,0,0,0,1,0,1
 * @param arr
 * @return
 */
public static boolean isCheck(int[][] arr) {
    for (int i = 0; i < arr.length; i++){
        for (int j = 0; j < arr[i].length; j++){
            if (arr[i][j] == 0) {
                return false;
            }
        }
    }
    return true;
}

/**
 * 感染
 * @param arr
 * @return
 */
public static int[][] gr(int[][] arr,int n) {
    int[][] newArr = new int[n][n];
    for (int i = 0; i < arr.length; i++) {
        for (int j = 0; j < arr[i].length; j++) {
            newArr[i][j] = arr[i][j];
        }
    }

    for (int i = 0; i < arr.length; i++) {
        for (int j = 0; j < arr[i].length; j++) {
            // 跳过非感染区
            if (arr[i][j] == 0) {
                continue;
            }
            // 若在范围内，就上下左右感染
            if (i - 1 >= 0) {
                newArr[i - 1][j] = 1;
            }
            if (i + 1 < n) {
                newArr[i + 1][j] = 1;
            }
            if (j - 1 >= 0) {
                newArr[i][j - 1] = 1;
            }
            if (j + 1 < n) {
                newArr[i][j + 1] = 1;
            }
        }
        StringBuilder sb= new StringBuilder();
        for (Integer in:newArr[i]) {
            sb.append(",").append(in);
        }
    }
    return newArr;
}