蓝桥杯 暗恋问题解析

试题 算法训练 暗恋

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
　　同在一个高中，他却不敢去找她，虽然在别人看来，那是再简单不过的事。暗恋，是他唯一能做的事。他只能在每天课间操的时候，望望她的位置，看看她倾心的动作，就够了。操场上的彩砖啊，你们的位置，就是他们能够站立的地方，他俩的关系就像砖与砖之间一样固定，无法动摇。还记得当初铺砖的工人，将整个操场按正方形铺砖（整个操场可视为R行C列的矩阵，矩阵的每个元素为一块正方形砖块），正方形砖块有两种，一种为蓝色，另一种为红色。我们定义他和她之间的“爱情指标”为最大纯色正方形的面积，请你写一个程序求出“爱情指标”。
输入格式
　　第一行两个正整数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
数据规模和约定
　　40%的数据R,C<=10;
　　70%的数据R,C<=50;
　　100%的数据R,C<=200;

问题分析

首先看到这题，我第一反应是考虑用dp和二维数组前缀和之类的方法，但是仔细一想。
问题中没有求和，也只有0和1两个元素，用前缀和显然不合适，dp也很难找到入手点（能做，但是下面的方法显然是更简单）

既然目标区域为正方形，那么计算量上显然减小了很多。
这就让朴素算法成为了可能。

我们两层for循环遍历每个节点，用一个check函数来表示每个点正方形的最大边长，最后存储最大的边长就好啦~

check函数的逻辑也很好写，从2开始遍历，递归。如果满足则遍历是否满足3，4，5……以此类推。注意点是，这里递归最好进行算法优化，否则容易超时。

优化的方法是，以半径2为例，当2满足时，在计算3时，我们只需要遍历半径2所不包含的部分就好了，而不需要重新一一检查所有元素是否相同。时间复杂度从（o^2)变为了(o).

PS：也可以不使用函数递归，改用循环，虽然递归思路清晰，但是递归过程中反复传递地址会造成时间浪费，所以不使用递归会更快。

最后代码如下：

#include <iostream>
using namespace std;


int r, c, ma = 1;
int a[200][200];

int check(int x, int y,int le=2)
{
	int poi= x + le, poi2= y + le;
	if (poi<=r && poi2 <=c)
	{
		for (int i = x; i < poi-1; i++)
		{	
				if (a[i][poi2-1] != a[x][y])
				{
					return le - 1;
				}
		}
		for (int j = y; j < poi2; j++)
		{
			if (a[poi - 1][j] != a[x][y])
			{
				return le - 1;
			}
		}
	}
	else
	{
		return le - 1;
	}
	return check(x, y, le + 1);
}

int main()
{
	int ram;
	cin >> r >> c;
	for (int i = 0; i < r; i++)
	{
		for (int j = 0; j< c; j++)
		{
			cin >> a[i][j];
		}
	}

	for (int i = 0; i < r; i++)
	{
		for (int j = 0; j < c; j++)
		{
			ram = check(i, j);
			if (ram > ma)
			{
				ma = ram;
			}
		}
	}
		cout << ma * ma;
	return 0;
}