找矩阵中的最大正方形

问题描述

在一个二维01矩阵中找到全为1的最大正方形，返回其面积。
样例
1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0
返回 4

解题思想

先构造一个辅助矩阵tmp，把matrix矩阵的第一行和第一列元素直接复制；
对于剩下元素，若matrix[i][j]=0, 则tmp[i][j]=0;
若matrix[i][j]=1,则tmp[i][j]= min(tmp[i-1][j],tmp[i-1][j-1],tmp[i][j-1]) +1;
如
matrix=
1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0

则有
tmp=
1 0 1 0 0
1 0 1 1 1
1 1 1 2 2
1 0 0 1 0
tmp[i][j]的值表示的是矩阵中以(i,j)位置为右下角的正方形的边长；
最后遍历tmp矩阵，求出最大值即可得到最大正方形边长

代码

public int maxSquare(int[][] matrix) {
        // write your code here

        int row = matrix.length;  //行大小
        int line = matrix[0].length;  //列大小

        //一个与matrix相同大小的辅助数组
        int[][] tmp = new int[row][line];

        //将matrix的第一行和第一列元素直接存放到
        for(int i=0;i<row;i++){
            tmp[i][0] = matrix[i][0];
        }
        for(int i=0;i<line;i++){
            tmp[0][i] = matrix[0][i];
        }

        for(int i=1;i<row;i++){
            for(int j=1;j<line;j++){
                if(matrix[i][j] == 1){
                    tmp[i][j] = 
                    Math.min(Math.min(tmp[i-1][j],tmp[i][j-1]),tmp[i-1][j-1]) + 1;
                }
                if(matrix[i][j] == 0){
                    tmp[i][j] = 0;
                }

            }
        }

        int max=0;  //记录tmp中最大元素的值（tmp中元素值表示正方形的边长)
        for(int i=0;i<row;i++){
            for(int j=0;j<line;j++){
                if(tmp[i][j] > max){
                    max = tmp[i][j];
                }
            }
        }

        return max*max;
    }