BZOJ2342：双倍回文（回文自动机+树上差分）

题面
题意：给你一个串，求出它的最长子串
满足该子串有两个长度相等且为偶数的回文串拼接而成

该串显然是回文串，就对应回文自动机上的一个状态

若某个状态的长度为4的倍数
且存在某个祖先的长度为其一半
则该状态可以贡献答案

因为祖先即回文后缀和前缀
长度为其一半的前缀和后缀都是回文串
显然符合题意

我的做法是树上差分

#include <iostream>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <ctime>
#include <cstdio>
#include <cstdlib>
#include <cstring>

using namespace std;
#define mmst(a, b) memset(a, b, sizeof(a))
#define mmcp(a, b) memcpy(a, b, sizeof(b))

typedef long long LL;

const int N=500500;

int n,ans;
int son[N][26],pre[N],len[N],cnt,last;
int to[N],nex[N],head[N],cur;
bool ok[N];
char s[N];

void add(int u,int v)
{
    to[++cur]=v;
    nex[cur]=head[u];
    head[u]=cur;
}

void dfs(int x)
{
    if(ok[len[x]])
    ans=max(ans,len[x]);
    if(len[x]%2==0)
    ok[len[x]<<1]=1;
    for(int h=head[x];h;h=nex[h])
    dfs(to[h]);
    if(len[x]%2==0)
    ok[len[x]<<1]=0;
}

void Insert(int pos)
{
    int x=last,ch=s[pos]-'a';
    while(s[pos-len[x]-1]!=s[pos])
    x=pre[x];
    if(!son[x][ch])
    {
        len[++cnt]=len[x]+2;
        int y=pre[x];
        while(s[pos-len[y]-1]!=s[pos])
        y=pre[y];
        pre[cnt]=son[y][ch];
        son[x][ch]=cnt;
    }
    last=son[x][ch];
}

int main()
{
    pre[0]=pre[1]=cnt=1;
    len[1]=-1;

    cin>>n;
    scanf("%s",s+1);
    for(int i=1;i<=n;i++)
    Insert(i);

    add(1,0);
    for(int i=2;i<=cnt;i++)
    add(pre[i],i);

    dfs(1);
    cout<<ans<<endl;

    return 0;
}