题目描述
学校组织活动,将学生排成一个矩形方阵。 请在矩形方阵中找到最大的位置相连的男生数量。 这个相连位置在一个直线上,方向可以是水平的,垂直的,成对角线的或者呈反对角线的。 注:学生个数不会超过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;
}
}