LeetCode 221. 最大正方形 【DP】

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

示例 1：

输入：matrix = [[“1”,“0”,“1”,“0”,“0”],[“1”,“0”,“1”,“1”,“1”],[“1”,“1”,“1”,“1”,“1”],[“1”,“0”,“0”,“1”,“0”]]
输出：4

示例 2：

输入：matrix = [[“0”,“1”],[“1”,“0”]]
输出：1

示例 3：

输入：matrix = [[“0”]]
输出：0

提示：

m == matrix.length
n == matrix[i].length
1 <= m, n <= 300
matrix[i][j] 为 '0' 或 '1'

dp[i][j]代表以i，j为右下角的正方形的最大边长。
up[i][j]代表i，j的上方有多少个连续的1。
lef[i][j]代表该位置左边有多少个连续的1.

class Solution {
public:
    static const int maxn = 1e3 + 10;
    int up[maxn][maxn] = {0};
    int lef[maxn][maxn] = {0};
    int dp[maxn][maxn] = {0};
    void out(int n, int m) {
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < m; j++) {
                cout << dp[i][j] << " ";
            }
            cout << endl;
        }
        cout << endl;
    }
    int maximalSquare(vector<vector<char>>& matrix) {
        int n = matrix.size(), m = matrix[0].size();
        int ans = 0;
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < m; j++)
                if(j == 0 || matrix[i][j] == '0')
                    lef[i][j] = matrix[i][j] - '0';
                else
                    lef[i][j] = lef[i][j - 1] + 1;
        }
        for(int i = 0; i < n; i++) {
            for(int j = 0; j < m; j++) {
                if(matrix[i][j] == '1') ans = 1;
                if(i == 0 || matrix[i][j] == '0')
                    up[i][j] = matrix[i][j] - '0';
                else
                    up[i][j] = up[i - 1][j] + 1;
            }
        }
        for(int i = 0; i < n; i++)  dp[i][0] = matrix[i][0] - '0';
        for(int i = 0; i < m; i++) dp[0][i] = matrix[0][i] - '0';
        for(int i = 1; i < n; i++) {
            for(int j = 1; j < m; j++) {
                if(matrix[i][j] == '1')
                    dp[i][j] = min(min(up[i][j], lef[i][j]), dp[i - 1][j - 1] + 1);//状态转移
                    ans = max(ans, dp[i][j]);
            }
        }
        // out( n, m);
        return ans * ans;
    }
};