P1387 最大正方形

P1387 最大正方形

题目描述

在一个 $n\times m$ 的只包含 $0$ 和 $1$ 的矩阵里找出一个不包含 $0$ 的最大正方形，输出边长。

输入格式

输入文件第一行为两个整数 $n,m(1\le n,m\le 100)$，接下来 $n$ 行，每行 $m$ 个数字，用空格隔开，$0$ 或 $1$。

输出格式

一个整数，最大正方形的边长。

样例

样例输入

4 4
0 1 1 1
1 1 1 0
0 1 1 0
1 1 0 1

样例输出

2

开始思路

用每个最大正方形的右下角元素来记录其大小，对于一个元素，就按照其左上角元素的大小去检查横竖两条边界是否有0的存在，用flg标记，如果没有，那么当前元素就在左上角元素的基础上+1

问题代码（90分）

#include <bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
int n, m, i, j, a[105][105], ma, k;
signed main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> m;
    for (i = 1; i <= n; i++)
        for (j = 1; j <= m; j++)
            cin >> a[i][j];
    for (i = 1; i <= n; i++) {
        for (j = 1; j <= m; j++) {
            int flg = 1;
            k = a[i - 1][j - 1];    //左上角为代表的小正方形边长
            for (; k >= 0; k--) {
                if (a[i - k][j] == 0 || a[i][j - k] == 0) {     //检查横竖两条边界
                    flg = 0;
                    break;
                }
            }
            if (flg == 1)  a[i][j] += a[i - 1][j - 1];  //如果边界没问题，该位置在左上角的基础上再+1
            ma = max(ma, a[i][j]);
            //cout << "a["<<i<<"]["<<j<<"] = " << a[i][j] << endl;    //测试
        }
    }
    cout << ma;
    return 0;
}

反思

卡了一个点，下载数据来看并不是一个极小或极大的特殊点，所以不是特判的问题。经请教大佬后得知，我们不能仅仅看左上角推到底的那个数，如果推到底不成立，那么比它小的也要一个个看。

测试样例

4 4 
1 1 0 0
1 1 1 1
0 1 1 1
0 1 1 1

数组情况

1 1 0 0
1 2 1 1
0 1 2 2
0 1 2 2

很明显，最右下角a[4][4]应该为3，但现在却为2，证实了上面的反思

AC代码

#include <bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
int n, m, i, j, a[105][105], ma, k, flg;
signed main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> m;
    for (i = 1; i <= n; i++)
        for (j = 1; j <= m; j++)
            cin >> a[i][j];
    for (i = 1; i <= n; i++) {
        for (j = 1; j <= m; j++) {
            for (k = a[i - 1][j - 1]; k >= 1; k--){    //左上角为代表的小正方形边长,但要左上角从大到小试一遍，不然卡90分
                flg = 1;
                for (int tmp = k; tmp >= 0; tmp--) {
                    if (a[i - tmp][j] == 0 || a[i][j - tmp] == 0) {     //检查横竖两条边界
                        flg = 0;
                        break;
                    }
                }
                if (flg == 1) {
                    a[i][j] += k;   //如果边界没问题，该位置在左上角的基础上再+1
                    break;
                }
            }
            ma = max(ma, a[i][j]);
            //cout << "a["<<i<<"]["<<j<<"] = " << a[i][j] << endl;    测试
        }
    }
    cout << ma;
    return 0;
}

最后附上思考笔记一张