bzoj 4044: [Cerc2014] Virus synthesis 回文树

题意

你要用ATGC四个字母用两种操作拼出给定的串：
1.将其中一个字符放在已有串开头或者结尾
2.将已有串复制，然后reverse，再接在已有串的头部或者尾部
一开始已有串为空。求最少操作次数。
len<=100000

分析

首先有一个结论，就是组成一个偶回文串的最后一步一定是操作二。
这样的话我们可以把回文树搞出来然后dp。
设f[i]表示建出回文串i的最少步数。
f[长度为0的偶回文串]=1
如果i为奇回文串，则f[i]=len[i]
否则的话，有两种转移
一种是f[i]=f[去掉头尾后的串]+1
一种是找到一个最长的不超过len[i]/2的偶回文后缀j，f[i]=f[j]+len[i]/2-len[j]
求出所有的f后，枚举最后一次操作二搞出了哪个偶回文串，设为x，那么ans=min(ans,n-len[x]+f[x])

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=100005;

int n,sz,len[N],f[N],ls,ch[N][4],trans[N],fa[N],fail[N],num[26];
char s[N];

void ins(int c,int n)
{
    int p=ls;
    while (s[n-len[p]-1]!=s[n]) p=fail[p];
    if (!ch[p][c])
    {
        int np=++sz,k=fail[p];len[np]=len[p]+2;fa[np]=p;
        while (s[n-len[k]-1]!=s[n]) k=fail[k];
        fail[np]=ch[k][c];ch[p][c]=np;
        if (s[n]==s[n-len[trans[p]]-1]) trans[np]=ch[trans[p]][c];
        else trans[np]=fail[np];
        while (len[trans[np]]>len[np]/2) trans[np]=fail[trans[np]];
    }
    ls=ch[p][c];
}

void solve()
{
    f[0]=1;
    for (int i=2;i<=sz;i++)
        if (len[i]&1) f[i]=len[i];
        else f[i]=min(f[fa[i]]+1,f[trans[i]]+len[i]/2-len[trans[i]]+1);
    int ans=n;
    for (int i=2;i<=sz;i++) if (!(len[i]&1)) ans=min(ans,f[i]+n-len[i]);
    printf("%d\n",ans);
}

int main()
{
    num[0]=0;num['C'-'A']=1;num['T'-'A']=2;num['G'-'A']=3;
    int T;scanf("%d",&T);
    while (T--)
    {
        scanf("%s",s+1);n=strlen(s+1);
        len[1]=-1;fail[0]=fail[1]=1;sz=ls=1;
        for (int i=1;i<=n;i++) ins(num[s[i]-'A'],i);
        solve();
        for (int i=0;i<=sz;i++)
            for (int j=0;j<4;j++)
                ch[i][j]=0;
    }
    return 0;
}