[ZJOI2007]棋盘制作

[ZJOI2007]棋盘制作

题目描述

国际象棋是世界上最古老的博弈游戏之一，和中国的围棋、象棋以及日本的将棋同享盛名。据说国际象棋起源于易经的思想，棋盘是一个8*8大小的黑白相间的方阵，对应八八六十四卦，黑白对应阴阳。

而我们的主人公小Q，正是国际象棋的狂热爱好者。作为一个顶尖高手，他已不满足于普通的棋盘与规则，于是他跟他的好朋友小W决定将棋盘扩大以适应他们的新规则。

小Q找到了一张由N*M个正方形的格子组成的矩形纸片，每个格子被涂有黑白两种颜色之一。小Q想在这种纸中裁减一部分作为新棋盘，当然，他希望这个棋盘尽可能的大。

不过小Q还没有决定是找一个正方形的棋盘还是一个矩形的棋盘（当然，不管哪种，棋盘必须都黑白相间，即相邻的格子不同色），所以他希望可以找到最大的正方形棋盘面积和最大的矩形棋盘面积，从而决定哪个更好一些。

于是小Q找到了即将参加全国信息学竞赛的你，你能帮助他么？

输入输出格式

输入格式：

包含两个整数N和M，分别表示矩形纸片的长和宽。接下来的N行包含一个N * M的01矩阵，表示这张矩形纸片的颜色（0表示白色，1表示黑色）。

输出格式：

包含两行，每行包含一个整数。第一行为可以找到的最大正方形棋盘的面积，第二行为可以找到的最大矩形棋盘的面积（注意正方形和矩形是可以相交或者包含的）。

输入输出样例

输入样例#1：

3 31 0 10 1 01 0 0

输出样例#1：

46

说明

对于20%的数据，N, M ≤ 80

对于40%的数据，N, M ≤ 400

对于100%的数据，N, M ≤ 2000

题解：

对于第一问，f[i][j]表示以i，j为右下角的最大正方形那么

if（a[i][j]==a[i-1][j]==a[i][j-1]==a[i-1][j-1]） f[i][j]=min（f[i-1][j-1],f[i-1][j],f[i][j-1]）+1;

else f[i][j]=1;

对于第二问，先预处理出每个点向上和向下的最长连续长度，再n^2，维护一个min_up和min_down若处于行位或无法与前一位构成矩形，则统计ans

代码：

#include<bits/stdc++.h>
using namespace std;

const int max_n = 2010;
const int inf   = 1e9+7;

int f[max_n][max_n];
int up[max_n][max_n],down[max_n][max_n];
int a[max_n][max_n];
int n,m,ans1,ans2;

inline void init()
{
	for(int i=1; i<=n; ++i)
	  f[i][1]=1;
	for(int i=1; i<=m; ++i)
	  f[1][i]=1;
	for(int i=1; i<=n; ++i)
	  for(int j=1; j<=m; ++j)
	    if(a[i][j]==a[i-1][j]) up[i][j]=up[i-1][j]+1;
    for(int i=n; i>=1; --i)
      for(int j=m; j>=1; --j)
        if(a[i][j]==a[i+1][j]) down[i][j]=down[i+1][j]+1;
}

inline void dp1()
{
	for(int i=2; i<=n; ++i)
	  for(int j=2; j<=m; ++j)
	    if(a[i][j]==a[i-1][j] && a[i][j]==a[i][j-1] && a[i][j]==a[i-1][j-1])
	    {
	    	f[i][j]=min(f[i-1][j-1],min(f[i-1][j],f[i][j-1]))+1;
	    	ans1=max(f[i][j],ans1);
		}
		else f[i][j]=1;
	ans1*=ans1;
	for(int i=1; i<=n; ++i)
	{
		int minn1=up[i][1],minn2=down[i][1],tot=1;
		for(int j=2; j<=m; ++j)
		{
			if(a[i][j]==a[i][j-1])
			  minn1=min(minn1,up[i][j]),minn2=min(minn2,down[i][j]),tot++;
			else
			{
				ans2=max(ans2,tot*(minn1+minn2+1));
				tot=1;
				minn1=up[i][j];
				minn2=down[i][j];
			}	
			ans2=max(ans2,tot*(minn1+minn2+1));
		}
	}
}

int main()
{
	scanf("%d%d",&n,&m);
	memset(a,-1,sizeof(a));
	for(int i=1; i<=n; ++i)
	  for(int j=1; j<=m; ++j)
	  {
	  	scanf("%d",&a[i][j]);
	  	if((i+j)&1) a[i][j]^=1;
	  }
	init();
	dp1();
	printf("%d\n%d\n",ans1,max(ans1,ans2));
	return 0;
}