【习题·字符串】Milking Grid（区间KMP）

题目描述

每天早晨FARMER JOHN的奶牛都会在挤奶时排成矩形，R行（1<=R<=10,000），C列（1<=C<=75）。我们知道，FARMER JOHN是奶牛专家，他打算写一本关于喂养奶牛的书。他发现，当奶牛按照不同的血统标记以后，整个大矩形就像由很多相同的小矩形拼起来的一样。

请帮助FJ找到面积最小的小矩形，使它能拼出整个大矩形。小矩形的尺寸不一定要整除大矩形的，但是小矩形的排列一定要非常整齐，不允许出现错位之类的情况，即你可以用若干个小矩形拼出一个包含大矩形的矩形。

比如，对于下面的矩形：

ABABA

BABAB

最小的面积应该是4，而不是2。

题解

这道题的主要思路就是，分别求出行和列的公共最小循环节，即每一行（或列）有一个相同的值使得存在大小为这个值的循环节，且使得这个值最小；然后只要两个值相乘即可。

我们可以分别求出行的所有循环节长度，然后数组统计一下即可。

显然，最小循环节$=n-next[i]$;次小循环节$=n-next[next[i]]$…以此类推，我们可以不断通过循环求出并累加在某一个数组上。只要某一个长度的累加次数$=n$(或$m$)，说明这是一个公共循环节、累加一下即可。

代码如下：

#include <bits/stdc++.h>
using namespace std;
int n,m;
char a[20000];
int cnt[20000];
int Next[20000];
char s[20000][100];
void kmp(int N)
{
	for (int i=2,j=0;i<=N;++i)
	{
		while (j>0 && a[i]^a[j+1]) j=Next[j];
		if (a[i]==a[j+1]) j++;
		Next[i]=j;
	}
	int k=Next[N];
	while (k>0)
	{
		cnt[N-k]++;
		k=Next[k];
	}
}
int main(void)
{
	freopen("mgrid.in","r",stdin);
	freopen("mgrid.out","w",stdout);
	int xmin=INT_MAX,ymin=INT_MAX;
	scanf("%d %d",&n,&m);
	for (int i=1;i<=n;++i)
	    scanf("%s",s[i]+1);
	for (int i=1;i<=n;++i)
	{
		for (int j=1;j<=m;++j) a[j]=s[i][j];
		kmp(m);
	}
	for (int i=1;i<=m;++i) 
	    if (cnt[i]==n) 
		{
			ymin=i;
			break;
		} 
	memset(cnt,0,sizeof(cnt));
	memset(Next,0,sizeof(Next));
	for (int i=1;i<=m;++i)
	{
		for (int j=1;j<=n;++j) a[j]=s[j][i];
		kmp(n);
	}
	for (int i=1;i<=n;++i)
	    if (cnt[i]==m)
	    {
	    	xmin=i;
	    	break;
	    }
	int ansy=min(ymin,m);
	int ansx=min(xmin,n);
	printf("%d",ansx*ansy);
	return 0;
}