leetcode 221. Maximal Square [dp]

ljf_study

于 2019-09-24 10:20:04 发布

阅读量138

点赞数

分类专栏： leetcode 文章标签： dp max_square

本文链接：https://blog.youkuaiyun.com/ljf_study/article/details/101268249

版权

leetcode 专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种解决二维矩阵中寻找包含全部1的最大正方形问题的方法。通过动态规划算法，我们可以在O(m*n)的时间复杂度内找到该正方形，并返回其面积。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given a 2D binary matrix filled with 0’s and 1’s, find the largest square containing only 1’s and return its area.
Example:

Input:

1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0

Output: 4

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