ZJOI 2009 对称的正方形 ST+Manacher

题目：

Orez很喜欢搜集一些神秘的数据，并经常把它们排成一个矩阵进行研究。最近，Orez又得到了一些数据，并已经把它们排成了一个n行m列的矩阵。通过观察，Orez发现这些数据蕴涵了一个奇特的数，就是矩阵中上下对称且左右对称的正方形子矩阵的个数。 Orez自然很想知道这个数是多少，可是矩阵太大，无法去数。只能请你编个程序来计算出这个数。

简要思路：

首先在数之间加上0，矩阵变为(n*2-1)*(m*2-1)，于是就只要找边长为奇数的正方形了。

不妨设left[i][j]为以（i，j）为正方形中心时，仅考虑左边方向所能延伸的最长长度，right[][]，up[][]，down[][]定义类似；设lx[i][j]为以i，j为中心的，水平方向上的最长回文串长度的一半，ly[][]定义类似；设f[i][j]为以（i，j）为中心的合法正方形个数。则可得到以下结论（证明略）:

1）f[i][j] = min{left[i][j], right[i][j], up[i][j], down[i][j]}；

2）left[i][j] = max{k | min{ly[i][j-k...j]} >= j - k}，right[][]，up[][]，down[][]求法类似；

3）i 一定时，j - left[i][j]随j增加不减，right[][]，up[][]，down[][]性质类似；

4）若i, j同偶，则（i，j）对答案的贡献为(f[i][j] + 1) / 2；若i,j同奇，则其贡献为f[i][j] / 2 + 1；否则没有贡献。

结合以上结论，最小值用ST解决，回文串用马拉车解决即可。

代码：

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>

#define For(i,j,k) for(int i = j;i <= k;i++)
#define Forr(i,j,k) for(int i = j;i >= k;i--)
#define Set(i,j) memset(i, j, sizeof(i))

const int N = 2010;

using namespace std;

int n, m;
int A

, v
;
int lx

, ly

;
int Min
[12], Log
, f

;

int Read(){
char c = getchar();
while(c > '9' || c < '0') c = getchar();
int x = c - '0';
for(c = getchar();'0' <= c && c <= '9';c = getchar()) x = x * 10 + c - '0';
return x;
}

void Manacher(int *len, int tot){
int k = 0;
For(i,1,tot){
int p = k + len[k];
if(p > i && len[2 * k - i] < p - i) len[i] = len[2 * k - i];
else{
if(p > i) len[i] = p - i;
while(i + len[i] < tot && i - len[i] > 1 && v[i + len[i] + 1] == v[i - len[i] - 1]) ++len[i];
k = i;
}
}
}

void RMQ(int x

, int k, int tot){
Set(Min, 0x3f);
For(i,1,tot) Min[i][0] = x[i][k];
For(j,1,11)
For(i,1,tot - (1 << j) + 1) Min[i][j] = min(Min[i][j-1], Min[i+(1<<(j-1))][j-1]);
}

int Query(int L, int R){
int x = Log[R - L + 1];
return min(Min[L][x], Min[R-(1<<x)+1][x]);
}

int main(){
n = Read(), m = Read();
For(i,2,N-1) Log[i] = Log[i>>1] + 1;
For(i,1,n) For(j,1,m) A[i*2-1][j*2-1] = Read();
n = n * 2 - 1, m = m * 2 - 1;
For(i,1,n){
For(j,1,m) v[j] = A[i][j];
Manacher(lx[i], m);
}
For(i,1,m){
For(j,1,n) v[j] = A[j][i];
Manacher(ly[i], n);
}
Set(f, 0x3f);
For(i,1,n){
RMQ(ly, i, m);
int v = 1;
For(j,1,m){
while(v < j && Query(v, j) < j - v) ++v;
f[i][j] = min(f[i][j], j - v);
}
v = m;
Forr(j,m,1){
while(v > j && Query(j, v) < v - j) --v;
f[i][j] = min(f[i][j], v - j);
}
}
For(i,1,m){
RMQ(lx, i, n);
int v = 1;
For(j,1,n){
while(v < j && Query(v, j) < j - v) ++v;
f[j][i] = min(f[j][i], j - v);
}
v = n;
Forr(j,n,1){
while(v > j && Query(j, v) < v - j) --v;
f[j][i] = min(f[j][i], v - j);
}
}
int Ans = 0;
For(i,1,n) For(j,1,m)
if((i & 1) && (j & 1)) Ans += (f[i][j] >> 1) + 1;
else if(!(i & 1) && !(j & 1)) Ans += (f[i][j] + 1) >> 1;
printf("%d\n", Ans);
return 0;
}