华为OD2023(A卷)基础题8【开心消消乐】

​
【开心消消乐】

给定一个N行M列的二维矩阵，矩阵中每个位置的数字取值为0或1。矩阵示例如：
1100
0001
0011
1111
现需要将矩阵中所有的1进行反转为0，规则如下：
1） 当点击一个1时，该1便被反转为0，同时相邻的上、下、左、右，以及左上、左下、右上、右下8 个方向的1（如果存在1）均会自动反转为0；
2）进一步地，一个位置上的1被反转为0时，与其相邻的8个方向的1（如果存在1）均会自动反转为0；

按照上述规则示例中的矩阵只最少需要点击2次后，所有值均为0。请问，给定一个矩阵，最少需要点击几次后，所有数字均为0？

Java代码：

import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner;

public class OdAb8 {
    public static int matrixFlip(int[][] matrix) {
        int m = matrix.length; // 矩阵行数
        if (m == 0) return 0;
        int n = matrix[0].length; // 矩阵列数
        boolean[][] visited = new boolean[m][n]; // 定义标记数组用于记录某个位置是否被访问过
        int count = 0; // 记录需要点击的次数

        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (matrix[i][j] == 1 && !visited[i][j]) { // 如果当前位置为1且未被访问过
                    bfs(matrix, visited, i, j); // 进行广度优先搜索并将经过的位置全部标记为已访问
                    count++; // 点击次数加一
                }
            }
        }
        return count; // 返回需要点击的次数
    }

    public static void bfs(int[][] matrix, boolean[][] visited, int i, int j) {
        Queue<int[]> queue = new LinkedList<>(); // 利用队列来进行广度优先搜索
        visited[i][j] = true; // 将当前位置标记为已访问
        queue.offer(new int[]{i, j}); // 将当前位置推入队列

        while (!queue.isEmpty()) {
            int[] cur = queue.poll(); // 取出队首位置
            int x = cur[0], y = cur[1];
            matrix[x][y] = 0; // 将当前位置置为0

            // 对当前位置的8个相邻位置进行访问
            for (int r = -1; r <= 1; r++) {
                for (int c = -1; c <= 1; c++) {
                    if (r == 0 && c == 0) continue; // 跳过当前位置本身
                    int nx = x + r, ny = y + c;
                    if (nx >= 0 && nx < matrix.length && ny >= 0 && ny < matrix[0].length // 判断边界
                            && matrix[nx][ny] == 1 && !visited[nx][ny]) { // 判断是否为1且未被访问过
                        visited[nx][ny] = true; // 将其标记为已访问
                        queue.offer(new int[]{nx, ny}); // 将其推入队列
                    }
                }
            }
        }
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        int m = scanner.nextInt();

        int[][] matrix = new int[n][m];
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                matrix[i][j] = scanner.nextInt();
            }
        }
        System.out.println(matrixFlip(matrix)); // 输出最少需要点击的次数，即2
    }
}