【华为OD机试题】学生方阵 Java代码实现

题目描述

学校组织活动，将学生排成一个矩形方阵。 请在矩形方阵中找到最大的位置相连的男生数量。 这个相连位置在一个直线上，方向可以是水平的，垂直的，成对角线的或者呈反对角线的。 注：学生个数不会超过10000

输入

输入的第一行为矩阵的行数和列数，接下来的n行为矩阵元素，元素间用”,”分隔。

输出

输出一个整数，表示矩阵中最长的位置相连的男生个数。

样例输入 

3,4
F,M,M,F
F,M,M,F
F,F,F,M

样例输出 

3

import java.util.Scanner;

public class Main {
    
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String[] str = sc.nextLine().split(",");
        int r = Integer.parseInt(str[0]);
        int c = Integer.parseInt(str[1]);
        
        char[][] matrix = new char[r][c];
        for (int i = 0; i < r; i++) {
            String[] row = sc.nextLine().split(",");
            for (int j = 0; j < c; j++) {
                matrix[i][j] = row[j].charAt(0);
            }
        }
        int maxConnected = dfs(matrix);
        
        // 输出结果
        System.out.println(maxConnected);
    }
    
    private static int dfs(char[][] matrix) {
        int r = matrix.length;
        int c = matrix[0].length;
        int[][][] dp = new int[r][c][4];
        int maxConnected = 0;
        
        for (int i = 0; i < r; i++) {
            for (int j = 0; j < c; j++) {
                if (matrix[i][j] == 'M') {
                    // 初始化当前位置的最大相连数量为1
                    for (int k = 0; k < 4; k++) {
                        dp[i][j][k] = 1;
                    }
                    
                    // 更新最大相连数量
                    if (i > 0) {
                        dp[i][j][0] += dp[i-1][j][0];
                    }
                    
                    if (j > 0) {
                        dp[i][j][1] += dp[i][j-1][1];
                    }
                    
                    if (i > 0 && j > 0) {
                        dp[i][j][2] += dp[i-1][j-1][2];
                    }
                    
                    if (i > 0 && j < c - 1) {
                        dp[i][j][3] += dp[i-1][j+1][3];
                    }
                    
                    for (int k = 0; k < 4; k++) {
                        maxConnected = Math.max(maxConnected, dp[i][j][k]);
                    }
                }
            }
        }
        
        return maxConnected;
    }
        
    
}