LeetCode 221. Maximal Square

题解

这道题在luogu上遇过，还有更烦人的类型比如算对角长度的。
dp解之，比较简单 大体思路就是dp[i,j]代表以[i,j]为右下角的方块长度
然后根据dp[i-1,j-1]和两边长度，遍历计算即可。

ps:我写的空间上还可以优化，vec和hor矩阵可以用cot省略。

---

Code

class Solution {
public:
    int maximalSquare(vector<vector<char>>& matrix) {
        if(matrix.empty()) return 0;
        int n=matrix.size();
        int m=matrix[0].size();
        vector<vector<int>> cot(n+1),vec(n+1),hor(n+1);
        for(int i=0;i<=n;i++){ 
            cot[i] = hor[i] = vec[i] = vector<int> (m+1,0);
        }
        
        int res=0;
        for(int i=1;i<=n;i++){
            for(int j=1;j<=m;j++){
                if(matrix[i-1][j-1]=='1'){
                    vec[i][j] = vec[i][j-1]+1;
                    hor[i][j] = hor[i-1][j]+1;
                    
                    cot[i][j] = min(cot[i-1][j-1],min(vec[i][j-1],hor[i-1][j]))+1;
                    
                    res=max(cot[i][j],res);
                }
            }
        }
        return res*res;
    }
};