题解 | #牛群的活动区域#

牛群的活动区域

https://www.nowcoder.com/practice/eabeca0c6e944a618f8adfed128d847e

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param board char字符型二维数组
     * @return char字符型二维数组
     */
    public char[][] solve (char[][] board) {
        // write code here
        int m = board.length;
        int n = board[0].length;

        for (int i = 0; i < m; i++) {
            dfs(board, i, 0);
            dfs(board, i, n - 1);
        }

        for (int j = 1; j < n - 1; j++) {
            dfs(board, 0, j);
            dfs(board, m - 1, j);
        }

        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (board[i][j] == 'B') {
                    board[i][j] = 'A';
                } else if (board[i][j] == 'C') {
                    board[i][j] = 'B';
                }
            }
        }

        return board;
    }

    private void dfs(char[][] board, int i, int j) {
        int m = board.length;
        int n = board[0].length;

        if (i < 0 || i >= m || j < 0 || j >= n || board[i][j] != 'B') {
            return;
        }

        board[i][j] = 'C';
        dfs(board, i - 1, j);
        dfs(board, i + 1, j);
        dfs(board, i, j - 1);
        dfs(board, i, j + 1);
    }
}

编程语言是Java。

该题考察的知识点是深度优先搜索(DFS)算法和二维数组的遍历与修改。

代码简短的文字解释如下:

  1. 遍历数组的第一列和最后一列,对于每个位置上的'B'字符,调用dfs方法进行标记。
  2. 遍历数组的第一行和最后一行,对于每个位置上的'B'字符,调用dfs方法进行标记。
  3. 再次遍历整个数组,将'B'字符替换为'A'字符,将'C'字符替换为'B'字符。
  4. 返回修改后的数组。
全部评论

相关推荐

05-09 13:22
门头沟学院 Java
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务