华为OD机试真题 Java 实现【开心消消乐】【2023 B卷 100分】

news2024/7/31 6:32:48

在这里插入图片描述

一、题目描述

给定一个N行M列的二维矩阵，矩阵中每个位置的数字取值为0或1。矩阵示例如：

1 1 0 0
0 0 0 1
0 0 1 1
1 1 1 1

现需要将矩阵中所有的1进行反转为0，规则如下：

当点击一个1时，该1变被反转为0，同时相邻的上、下、左、右，以及左上、左下、右上、右下8个方向的1（如果存在1）均会自动反转为0；
进一步地，一个位置上的1被反转为0时，与其相邻的8个方向的1（如果存在1）均会自动反转为0；
按照上述规则示例中的矩阵只最少需要点击2次后，所有值均为0。请问，给定一个矩阵，最少需要点击几次后，所有数字均为0？

二、输入描述

第一行为两个整数，分别表示矩阵的行数N和列数M，取值范围均为[1, 100]。

接下来N行表示矩阵的初始值，每行均为M个数，取值范围[0,1]。

三、输出描述

输出一个整数，表示最少需要点击的次数。

四、Java算法源码

static int[][] nums;
static int n;//行
static int m;//列

public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    n = sc.nextInt();
    m = sc.nextInt();
    nums = new int[n][m];
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            nums[i][j] = sc.nextInt();
        }
    }
    int ans = 0;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            if (nums[i][j] == 0) {
                continue;
            }
            ans += dfs(i, j);
        }
    }
    System.out.println(ans);
}

public static int dfs(int x, int y) {
    nums[x][y] = 0;
    int[][] direction = new int[][]{{-1, -1}, {-1, 0}, {-1, 1}, {0, -1}, {0, 1}, {1, -1}, {1, 0}, {1, 1}};
    for (int i = 0; i < direction.length; i++) {
        int cx = x + direction[i][0];
        int cy = y + direction[i][1];
        if (0 <= cx && cx < n && 0 <= cy && cy < m && nums[cx][cy] == 1) {
            dfs(cx, cy);
        }
    }
    return 1;
}