P1387 最大正方形

P1387 最大正方形

题目描述

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

输入格式：

输入文件第一行为两个整数n,m（1<=n,m<=100），接下来n行，每行m个数字，用空格隔开，0或1.

输出格式：

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

样例输入 Copy

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

样例输出 Copy

2

思路

因为最近想学过前缀和（其实是不会DP），想了一下如果一个正方形的区间内的和为边长的平方那么这一定是一个全是1的正方形，只要知道二维表中的前缀和公式这题还是很简单的。

AC代码

#include<bits/stdc++.h>
using namespace std;
int s[105][105];
int a[105][105];
int main(){
	int n,m;
	int flag = 0;
	cin >> n >> m;
	memset(a,0,sizeof(a));
	memset(s,0,sizeof(s));
	for(int i = 1 ; i <= n ; i ++){
		for(int j = 1 ; j <= m ; j ++){
			cin >> a[i][j];
			if(a[i][j]) flag = 1;
		}
	}
	
	for(int i = 1 ; i <= n ; i ++){
		for(int j = 1 ; j <= m ; j ++){
			s[i][j] = a[i][j] + s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1];
		}
	}
/*	cout << endl;
	for(int i = 1 ; i <= n ; i ++){
		for(int j = 1 ; j <= m ; j ++){
			cout << s[i][j] << " "; 
		}
		cout << endl;
	}
*/	
	int Max = max(n,m);
	int ans;
	int Index = 1;
	for(int i = 1 ; i < Max ; i ++){
		for(int j = 1 ; j <= n - i ; j ++){
			for(int k = 1 ; k <= m - i ; k ++){
				int x1 = j,x2 = j + i;
				int y1 = k,y2 = k + i;
				ans=s[x2][y2]-s[x1-1][y2]-s[x2][y1 - 1]+s[x1-1][y1-1];
				if(ans == (i + 1) * (i + 1)) Index = i + 1;
//				cout << "X1 : " << x1 << " Y1 : " << y1 << " X2 : " << x2 << " Y2 : " << y2 << " ans : " << ans << endl;
			}
		}
	}
	if(flag == 0) cout << 0 << endl;
	else{
		cout << Index << endl;
		
	}
/*	
for(int i = 1 ; i <= n ; i ++){
		for(int j = 1 ; j <= m ; j ++){
			cout << s[i][j] << " ";
		} 
		cout << endl;
	}
	*/
}

加粗样式