1321-全为1面积最大正方形Ⅲ

题目描述：
PIPI有一个n*m的01矩阵，它现在想在矩阵中找到一个全由1构成且面积最大的正方形，请你输出这个最大正方形的面积。
输入：
单组输入。
第一行两个正整数n,m,n<=3000,m<=3000.
接下来n行，每行m个数字，数字仅由0和1组成。
输出：
输出一个整数，代表最大全1正方形的面积。
样例输入：
3 5
1 0 1 1 1
1 1 1 1 1
1 1 1 1 1
样例输出：
9

题解代码如下：

#include <bits/stdc++.h>
using namespace std;
const int N = 3005;
int s[N][N];
int n, m;


bool judge(int len) {
	for (int a = 1; a + len - 1 <= n; a++) {
		for (int b = 1; b + len - 1 <= m; b++) {
			int c = a + len - 1, d = b + len - 1;
			int sum = s[c][d] - s[c][b - 1] - s[a - 1][d] + s[a - 1][b - 1];
			if (sum == len * len) return true;
		}
	}
	return false;
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);

	cin >> n >> m;
	int x;

	// 前缀和初始化
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= m; j++) {
			cin >> x;
			s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + x;
		}
	}

	// 二分正方形边长
	int l = 1, r = min(n, m);
	while (l < r) {
		int mid = (l + r + 1) >> 1;
		if (judge(mid)) l = mid;
		else r = mid - 1;
	}

	cout << r * r << endl;
	return 0;
}