LeetCode221 最大正方形

LeetCode221 最大正方形

网址

在一个由 0 和 1 组成的二维矩阵内，找到只包含 1 的最大正方形，并返回其面积。

class Solution {
public:
    int maximalSquare(vector<vector<char>>& matrix) {
        /**
        dp[i][j]表示以第i行第j列为右下角所能构成的最大正方形边长, 则递推式为: 
dp[i][j] = 1 + min(dp[i-1][j-1], dp[i-1][j], dp[i][j-1]); matrix[i][j]=1
        **/
        if(matrix.empty()) return 0;
        int rows=matrix.size(),rols=matrix[0].size(),ans=0;
        // int dp[matrix.size(),matrix[0].size()];
        vector< vector<int> >dp(rows,vector<int>(rols));
        // 初始化dp
        for(int i=0;i<rols;++i){
            dp[0][i]=matrix[0][i]-'0';
            ans = max(ans,dp[0][i]);
        }
        for(int i=0;i<rows;++i){
            dp[i][0]=matrix[i][0]-'0';
            ans=max(ans,dp[i][0]);
        }
        for(int i=1;i<rows;++i){
            for(int j=1;j<rols;++j){
                if(matrix[i][j]=='1'){
                    dp[i][j] = 1 + min(dp[i-1][j-1], min(dp[i-1][j], dp[i][j-1]));
                }
                else{
                    dp[i][j]=0;
                }
                if(ans<dp[i][j]) ans=dp[i][j];
            }
        }
        return ans*ans;
    }
};