[蓝桥杯][算法训练VIP]暗恋（二维树状数组）

题目描述
同在一个高中，他却不敢去找她，虽然在别人看 来，那是再简单不过的事。暗恋，是他唯一能做的事。他只能在每天课间操的时候，望望她的位置，看看她倾心的动作，就够了。操场上的彩砖啊，你们的位置，就是他们能够站立的地方，他俩的关系就像砖与砖之间一样固定，无法动摇。还记得当初铺砖的工人，将整个操场按正方形铺砖（整个操场可视为R行C列的矩阵，矩阵的每个元素为一块正方形砖块），正方形砖块有两种，一种为蓝色，另一种为红色。我们定义他和她之间的“爱情指标”为最大纯色正方形的面积，请你写一个程 序求出“爱情指标”。

数据规模和约定
R,C< =200;

输入
第一行两个正整数R和C。
接下来R行C列描述整个操场，红色砖块用1来表示，蓝色砖块用0来表示。
输出
一个数，表示他和她之间的“爱情指标”。
样例输入
5 8
0 0 0 1 1 1 0 1
1 1 0 1 1 1 1 1
0 1 1 1 1 1 0 1
1 0 1 1 1 1 1 0
1 1 1 0 1 1 0 1
样例输出
9
思路：虽然数据量不大，但是纯纯的暴力也不行。我们可以枚举一下正方形的边长，看是否符合定义。但是枚举了边长，如何计算正方形的面积呢？难道要再循环吗？这样有事O(n*n)的时间复杂度。肯定承担不起，因为我们要采用二维树状数组这种数据结构来求一下二维前缀和，这样时间复杂度就降下来了。二维树状数组可以自行百度。
代码如下：

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

const int maxx=2e2+10;
int a[maxx][maxx];
int c[maxx][maxx];
int n,m;

inline int lowbit(int x){return x&-x;}
inline void update(int y,int x,int v)
{
	while(y<maxx)
	{
		int tx=x;
		while(tx<maxx) 
		{
			c[y][tx]+=v;
			tx+=lowbit(tx);
		}
		y+=lowbit(y);
	}
}
inline int query(int x,int y)
{
	int sum=0;
	while(x)
	{
		int ty=y;
		while(ty)
		{
			sum+=c[x][ty];
			ty-=lowbit(ty);
		}
		x-=lowbit(x);
	}
	return sum;
}
inline int check(int &l,int flag)
{
	int x,y,sum,len;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=m;j++)
		{
			if(a[i][j]&&flag)
			{
				while(1)
				{
					x=i+l;
					y=j+l;
					if(x<=n&&y<=m) sum=query(x,y)-query(x,j-1)-query(i-1,y)+query(i-1,j-1);
					else break;
					if(sum==(l+1)*(l+1)) l++;
					else break;
				}
			}
			else if(!a[i][j]&&!flag)
			{
				while(1)
				{
					x=i+l;
					y=j+l;
					if(x<=n&&y<=m) sum=query(x,y)-query(x,j-1)-query(i-1,y)+query(i-1,j-1);
					else break;
					if(sum==0) l++;
					else break;
				}
			}
		}
	}
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	for(int j=1;j<=m;j++) scanf("%d",&a[i][j]);
	for(int i=1;i<=n;i++)
	for(int j=1;j<=m;j++) update(i,j,a[i][j]);
	int len=1;
	int ans1=check(len,1);
	int ans2=check(len,0);
	cout<<len*len<<endl;
	return 0;
}

努力加油a啊，(^o)/~