华为OD机试 - 疫情扩散时间计算 - 矩阵（Java 2024 C卷 200分）

在这里插入图片描述

专栏导读

本专栏收录于《华为OD机试（JAVA）真题（A卷+B卷+C卷）》。

刷的越多，抽中的概率越大，每一题都有详细的答题思路、详细的代码注释、样例测试，发现新题目，随时更新，全天CSDN在线答疑。

一、题目描述

在一个地图中（地图有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

四、解题思路

将输入字符串转换为一维数组；
将一维数组转换为二维矩阵；
定义感染区域队列arrQueue；
- 将一维数组转换为二维矩阵；
- 将感染区域加入队列；
判断特殊情况；
记录未感染区域数量；
记录四个方向的偏移量；
记录感染天数；
当队列不为空且还有未感染区域时，进行循环；
- 取出队首元素；
- 获取队首元素的坐标；
- 记录感染天数；
- 遍历四个方向；
  - 定义新的横坐标；
  - 定义新的纵坐标；
  - 判断边界；
  - 如果该区域未感染；
    - 未感染区域数量减一；
    - 标记该区域已感染；
    - 将该区域加入队列；
- 返回感染天数；
返回感染天数；

五、Java算法源码

package com.guor.od;import java.util.*;public class OdTest {public static void main(String[] args) {Scanner sc = new Scanner(System.in);String input = sc.nextLine();List<Integer> map = new ArrayList<>();int pos = 0;String token = "";// 将输入字符串转换为一维数组while ((pos = input.indexOf(",")) != -1) {token = input.substring(0, pos);map.add(Integer.parseInt(token));input = input.substring(pos + 1);}map.add(Integer.parseInt(input));// 输出感染天数System.out.println(getResult(map));}public static int getResult(List<Integer> map) {int n = (int) Math.sqrt(map.size());// 将一维数组转换为二维矩阵int[][] matrix = new int[n][n];// 用队列存储感染区域Queue<int[]> arrQueue = new LinkedList<>();for (int i = 0; i < n; i++) {for (int j = 0; j < n; j++) {// 将一维数组转换为二维矩阵matrix[i][j] = map.get(i * n + j);// 将感染区域加入队列if (matrix[i][j] == 1) {arrQueue.offer(new int[]{i, j});}}}// 判断特殊情况if (arrQueue.isEmpty() || arrQueue.size() == map.size()) {return -1;}// 记录未感染区域数量int healthyNum = map.size() - arrQueue.size();// 记录四个方向的偏移量int[][] offSets = {{-1, 0}, {1, 0}, {0, -1}, {0, 1}};// 记录感染天数int day = 0;// 当队列不为空且还有未感染区域时，进行循环while (!arrQueue.isEmpty() && healthyNum > 0) {// 取出队首元素int[] arr = arrQueue.poll();// 获取队首元素的坐标int x = arr[0];int y = arr[1];// 记录感染天数day = matrix[x][y] + 1;// 遍历四个方向for (int[] offset : offSets) {// 新的横坐标int x_new = x + offset[0];// 新的纵坐标int y_new = y + offset[1];// 判断边界if (x_new < 0 || x_new >= n || y_new < 0 || y_new >= n) {continue;}// 如果该区域未感染if (matrix[x_new][y_new] == 0) {// 未感染区域数量减一healthyNum--;// 标记该区域已感染matrix[x_new][y_new] = day;// 将该区域加入队列arrQueue.offer(new int[]{x_new, y_new});}}}return day - 1; // 返回感染天数}
}