一道关于动态规划法的题。
矩阵中最长的连续1线段 :
一个矩阵中的元素只由0和1组成,对于给定的矩阵,试求矩阵中最长的连续1线段 ,方向可以是横向的、竖向的、斜向的。
public int longestLine(int[][] M) {
if(M.length==0)return 0;
int[][][] dp=new int[M.length][M[0].length][4];
int max=1;
for(int i=0;i<M.length;i++) {
for(int j=0;j<M[0].length;j++) {
if(M[i][j]==0)continue;
dp[i][j][0]=j==0?1:dp[i][j-1][0]+1;
dp[i][j][1]=i==0?1:dp[i-1][j][1]+1;
dp[i][j][2]=(i==0||j==0)?1:dp[i-1][j-1][2]+1;
dp[i][j][3]=(i==0||j==M[0].length-1)?1:dp[i-1][j+1][3]+1;
max=Math.max(max,Math.max(Math.max(dp[i][j][0],dp[i][j][1]),Math.max(dp[i][j][2],dp[i][j][3])));
}
}
return max;
}
动态规划法,用了一个三维数组来针对不同方向的线段,其实也可以想成是多个二维数组。
循环体中用了一些手段可以使代码更简洁。
如果需要对不同方向分别记录,那么设置多维数组是很有必要的。