正方形(矩阵dp)

221.最大正方形(leetcode)

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

输入:

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

输出: 4

思路:我们用动态规划来考虑这个问题,dp[i][j]表示以i,j为右下角,所有数据为1的的最长边长,我们只需要求出所有dp[i][j]来,取最大值的平方就是最大面积。
如果该位置的值是 0,则 dp(i, j) = 0,因为当前位置不可能在由 1 组成的正方形中;

如果该位置的值是 1,则 dp(i, j) 的值由其上方、左方和左上方的三个相邻位置的 dp 值决定。具体而言,当前位置的元素值等于三个相邻位置的元素中的最小值加 11,状态转移方程如下:
dp(i,j)=min(dp(i−1,j),dp(i−1,j−1),dp(i,j−1))+1在这里插入图片描述
代码如下:

int maximalSquare(vector<vector<char>>& matrix) {
        if(matrix.size()==0||matrix[0].size()==0)
        return 0;
        int ans=0;
        int n = matrix.size();
        int m = matrix[0].size();
        vector<vector<int> > dp(n,vector<int>(m)); //构建和matrix一样大的vector
        for(int i =0;i<n;i++)
        for(int j =0;j<m;j++)
        {
            if(matrix[i][j]=='1')
            {
                if(i==0||j==0) dp[i][j] =1; //如果是第一行或者第一列
            else
            {
                dp[i][j]=min(min(dp[i-1][j],dp[i][j-1]),dp[i-1][j-1])+1;
            }
            }
            ans = max(dp[i][j],ans);
        }
        return ans*ans; //结果要区平方
    }

1277.统计全为1的正方形子矩阵

和上述方法相同,但是这里与一个技巧,最大边长为n,则matrix[i][j]就能分成n,n-1,n-2…1,也就是ans直接加n就是该点所能贡献的正方形个数
代码如下:

int countSquares(vector<vector<int>>& matrix) {
        int n =matrix.size();
        int m = matrix[0].size();
        vector<vector<int> > f(n,vector<int>(m));
        int ans=0;
        for(int i=0;i<n;i++)
        for(int j=0;j<m;j++)
        {
            if(matrix[i][j]==1)
            {
                if(i==0||j==0) f[i][j]=1;
                else
                {
                    f[i][j] = min(min(f[i-1][j],f[i][j-1]),f[i-1][j-1])+1;
                }
            }
            ans += f[i][j];
        }
        return ans;
    }
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值