LintCode 631.最大矩阵Ⅱ

给出一个只有 0 和 1 组成的二维矩阵。
找出最大的一个子矩阵，使得这个子矩阵的主对角线元素均为 1 ，其他元素均为 0 。

你可以认为所求的矩阵一定是一个方阵。
主对角线指的是从左上角到右下角的一条对角线。

动态规划
初始化两个二维数组 分别记录每个点及其上面0的个数喝每个点及其左边0的个数
dp数组每个点代表当前最大矩阵的边长
用矩阵的右下角作为定位点 dp数组每个为1的点上面和左面0的个数和左上的值 三者最小值+1为当前点的最大矩阵

public class Solution {
    /**
     * @param matrix: a matrix of 0 an 1
     * @return: an integer
     */
    public int maxSquare2(int[][] matrix) {
        int m = matrix.length;
        int n = matrix[0].length;
        if(matrix == null || m == 0 || n == 0){
            return 0;
        }
        int[][] up = new int[m][n];
        int[][] left = new int[m][n];
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(i == 0 || j == 0){
                    up[i][j] = matrix[i][j] == 0 ? 1 : 0;
                    left[i][j] = matrix[i][j] == 0 ? 1 : 0;
                }
                else{
                    if(matrix[i][j] == 0){
                        left[i][j] = left[i-1][j] + 1;
                    }
                    else{
                        left[i][j] = 0;
                    }
                    if(matrix[i][j] == 0){
                        up[i][j] = up[i][j-1] + 1;
                    }
                    else{
                        up[i][j] = 0;
                    }
                }
            }
        }
        int[][] res = new int[m][n];
        int max = 0;
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                if(i == 0 || j == 0){
                    res[i][j] = matrix[i][j];
                }
                else{
                    if(matrix[i][j] == 1){
                        res[i][j] = Math.min(res[i-1][j-1], Math.min(up[i][j-1], left[i-1][j])) + 1;
                    }
                    else{
                        res[i][j] = 0;
                    }
                }
                max = Math.max(res[i][j], max);
            }
        }
        return max*max;
    }
}