题意:给一个字符矩阵,求最大对称等腰直角三角形面积。
做法:首先,旋转对称,做四次。每次先用二分hash预处理出点(i,j)向左和向上延伸的最长长度len[i][j],然后dp[i][j]=min(len[i][j]+dp[i-1][j-1]+2),所有dp值取最大就是边长,算一下得到面积。
代码:
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef unsigned long long ull;
char g[505][505];
char g2[505][505];
int n, m;
int M;
int dp[505][505];
int len[505][505];
ull pren[505][505];
ull prem[505][505];
ull wei[505];
void rotate()
{
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++){
g2[j][n-i+1]=g[i][j];
}
swap(n, m);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++){
g[i][j]=g2[i][j];
}
/*for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
printf("%c",g[i][j]);
}
printf("\n");
}
printf("\n");*/
}
void solve()
{
for(int i=1;i<=n;i++){
ull tmp=0;
pren[i][0]=0;
for(int j=1;j<=m;j++){
tmp*=131;
tmp+=g[i][j];
pren[i][j]=tmp;
}
}
for(int j=1;j<=m;j++){
ull tmp=0;
prem[0][j]=0;
for(int i=1;i<=n;i++){
tmp*=131;
tmp+=g[i][j];
prem[i][j]=tmp;
}
}
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
dp[i][j]=0;
int l=1, r=min(i, j);
int tag=-1;
while(l<=r){
if(r-l<=1){
if(pren[i][j]-pren[i][j-r]*wei[r]==prem[i][j]-prem[i-r][j]*wei[r])tag=r;
else tag=l;
break;
}
int mid=(l+r)>>1;
if(pren[i][j]-pren[i][j-mid]*wei[mid]==prem[i][j]-prem[i-mid][j]*wei[mid])l=mid;
else r=mid;
}
len[i][j]=tag;
}
}
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
dp[i][j]=min(len[i][j], dp[i-1][j-1]+2);
M=max(dp[i][j], M);
}
}
}
int main()
{
wei[0]=1;
for(int i=1;i<=500;i++)wei[i]=wei[i-1]*131;
int t;
scanf("%d", &t);
for(int tt=1;tt<=t;tt++){
scanf("%d%d", &n, &m);
for(int i=1;i<=n;i++)scanf("%s", g[i]+1);
M=0;
solve();
rotate();
solve();
rotate();
solve();
rotate();
solve();
printf("%d\n", M*(M+1)/2);
}
}