【ybt】【字符串 hash 课过 例3】对称正方形

对称正方形

题目链接：对称正方形

---

题目描述

解题思路

这道题很容易想到是哈希，而且是二维哈希。

正着、上下反着、左右反着就表示了所有的情况。

但是这道题的数据大小决定了它用不了 $O(n^3)$ 以上复杂度的算法。

而枚举每个点就需要 $O(n^2)$ 的复杂度。

所以我们只能以每个点为基础向外拓展，二分这个值的大小。

这样的话时间复杂度就是 $O(n^{2}logn)$。

code

#include<iostream>
#include<cstdio>
using namespace std;

const int p_1=10000007ull,p_2=10000009ull;

int n,m,ans;
int a[1010][1010];
int b[1010][1010];
int c[1010][1010];
int p1[1010],p2[1010];

bool check(int mid,int i,int j)
{
	if(i-mid<0||j-mid<0||i>n||j>m||i<0)
		return 0;
	int aa=a[i][j]-a[i][j-mid]*p1[mid]-a[i-mid][j]*p2[mid]+a[i-mid][j-mid]*p1[mid]*p2[mid];
	int tx=n-(i-mid),ty=m-(j-mid);
	int bb=b[tx][j]-b[tx][j-mid]*p1[mid]-b[tx-mid][j]*p2[mid]+b[tx-mid][j-mid]*p1[mid]*p2[mid];
	int cc=c[i][ty]-c[i][ty-mid]*p1[mid]-c[i-mid][ty]*p2[mid]+c[i-mid][ty-mid]*p1[mid]*p2[mid];
	if(aa==bb&&bb==cc)
		return 1;
	return 0;
}

void hashst()
{
	p1[0]=p2[0]=1ull;
	for(int i=1;i<=1001;i++)
		p1[i]=p1[i-1]*p_1,p2[i]=p2[i-1]*p_2;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
		{
			a[i][j]=a[i][j-1]*p_1+a[i][j];
			b[i][j]=b[i][j-1]*p_1+b[i][j];
			c[i][j]=c[i][j-1]*p_1+c[i][j];
		}
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
		{
			a[i][j]=a[i-1][j]*p_2+a[i][j];
			b[i][j]=b[i-1][j]*p_2+b[i][j];
			c[i][j]=c[i-1][j]*p_2+c[i][j];
		}
}

int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
		{
			scanf("%d",&a[i][j]);
			b[n-i+1][j]=a[i][j];
			c[i][m-j+1]=a[i][j];
		}
	hashst();
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
		{
			int l=0,r=min(i,j)+1,maxn=1;
			while(l<=r)
			{
				int mid=(l+r)/2;
				if(check(mid*2-1,i+mid,j+mid))
					l=mid+1,maxn=max(maxn,mid*2-1);
				else
					r=mid-1;
			}
			ans+=(maxn+1)/2;
			maxn=0;
			l=0,r=min(i,j);
			while(l<=r)
			{
				int mid=(l+r)/2;
				if(check(mid*2,i+mid,j+mid))
					l=mid+1,maxn=max(maxn,mid*2);
				else
					r=mid-1;
			}
			ans+=(maxn+1)/2; 
		}
	cout<<ans<<endl;
}