[BZOJ-3676] && [CodeForces-245H]：回文树的基础应用

题面：

BZOJ---3676

题意：

求出所给字符串的所有本质不同的回文串的数量及其长度

分析：

没啥好分析的，就是一道回文树的裸题，建议阅读：回文树【处理一类回文串问题的强力工具】

说说我的一些理解：

回文树的每个节点都表示一个本质不同的回文串，和AC自动机类似，也有一个fail指针，指向当前不包括自己的最长的回文后缀所在的节点，每加入一个字符，通过跳fail指针去找和这个字符匹配的最长回文串；最后有多少个节点就有多少个本质不同的回文串（当然要减去两个根节点），再从fail树的叶子节点往上跳，统计出每个本质不同回文的数量（因为较长的回文串一定包含了较短的回文串后缀），下面代码中的①②的位置不能交换，否则fail指针就可能指向自己，导致死循环

代码：

#include <bits/stdc++.h>
 
using namespace std;
const int maxn = 3e5+55;
struct PalindromicTree{
    int p,n,last,fail[maxn],tr[maxn][26],s[maxn],len[maxn],cnt[maxn];
    int newnode(int Len){
        for(int i = 0;i < 26; ++i) tr[p][i] = 0;
        len[p] = Len; cnt[p] = 0;
        return p++; 
    }
    void init(){
         p = n = last = 0;
         newnode(0);newnode(-1);
         fail[0] = 1;s[0] = -1;
    }
    int getfail(int x){
        while(s[n-len[x]-1] != s[n]) x = fail[x];
        return x;
    }
    void add(int x){
        s[++n] = x;
        int cur = getfail(last);
        if(!tr[cur][x]){
            int now = newnode(len[cur]+2);
            fail[now] = tr[getfail(fail[cur])][x]; //①
            tr[cur][x] = now;                      //②              
        } 
        last = tr[cur][x];
        cnt[last]++;
    }
    void count(){
        for(int i = p-1;i > 0; --i) cnt[fail[i]] += cnt[i];
    }
}T;
char ss[maxn];
int main(){
    scanf("%s",ss);
    int len = strlen(ss);
    T.init();
    for(int i = 0;i < len; ++i) T.add(ss[i]-'a');
    T.count();
    long long ans = 0LL; 
    for(int i = 0;i < T.p; ++i) ans = max(ans,1LL*T.len[i]*T.cnt[i]);
    printf("%lld",ans);
    return 0;
}

 

题面：

CodeForces---245H

题意：

给出一个长度最大5000的字符串和1e6次的查询，查询所给区间的回文串的数量

分析：

回文自动机num数组的应用，num[now] = num[fail[now]] + 1，可以求出以第x个字符为回文串的结尾的回文串的数量，因为now节点是以s[x]结尾的回文串，fail[now]也是以s[x]结尾的回文串，因为fail[now]是now的后缀；并且这些回文串都是本质不同的

那么我们就可以统计前缀和得到当前前缀的回文串的个数，此题从每个的起点都跑一遍，预处理出答案即可

代码：

#include <bits/stdc++.h>

using namespace std;
typedef long long LL;
const int mod = 1e9+7;
const int maxn = 5e3+53;
struct PalindromicTree{
    int last,tot,n,s[maxn],len[maxn],fail[maxn],num[maxn],tr[maxn][27];
    int newnode(int Len){
        for(int i = 0;i < 27; ++i) tr[tot][i] = 0;
        len[tot] = Len;
        num[tot] = 0;
        return tot++;
    }
    void init(){
        last = tot = n = 0;
        newnode(0),newnode(-1);
        fail[0] = 1; s[0] = -1;
    }
    int getfail(int x){
        while(s[n-len[x]-1] != s[n]) x = fail[x];
        return x;
    }
    int add(int x){
        s[++n] = x;
        int cur = getfail(last);
        if(!tr[cur][x]){
            int now = newnode(len[cur]+2);
            fail[now] = tr[getfail(fail[cur])][x];
            tr[cur][x] = now;
            num[now] = num[fail[now]]+1; 
        } 
        last = tr[cur][x];
        return num[last];        //返回以当前节点结尾的回文串的数量
    }
}T;
char ss[maxn];
int q,l,r,sum[maxn][maxn];
int main(){
    scanf("%s %d",ss+1,&q);
    int len = strlen(ss+1);
    for(int i = 1;i <= len; ++i){
        T.init();
        for(int j = i; j <= len; ++j){
            sum[i][j] = sum[i][j-1] + T.add(ss[j]-'a');
        }
    }
    while(q--){
        scanf("%d %d",&l,&r);
        printf("%d\n",sum[l][r]);
    }
    return 0;
}

 

题面：

最长公共回文子串

题意：

求所给两个串的最长公共回文子串

分析：

先将第一个串建一颗回文树，标记每个节点来自第一个串，又从这颗树的偶根出发跑一遍第二个串，如果达到了第一个串的节点，取最大长度即可，这种方法也可以求几个串的最长公共回文子串

代码:

#include <bits/stdc++.h>
 
using namespace std;
const int maxn = 3e5+55;
struct PalindromicTree{
    int p,n,ans,last,fail[maxn],tr[maxn][26],s[maxn],len[maxn],cnt[maxn];
    int newnode(int Len){
        for(int i = 0;i < 26; ++i) tr[p][i] = 0;
        len[p] = Len; cnt[p] = 0;
        return p++; 
    }
    void init(){
         p = n = last = ans = 0;
         newnode(0);newnode(-1);
         fail[0] = 1;s[0] = -1;
    }
    int getfail(int x){
        while(s[n-len[x]-1] != s[n]) x = fail[x];
        return x;
    }
    void add(int x,int op){
        s[++n] = x;
        int cur = getfail(last);
        if(!tr[cur][x]){
            int now = newnode(len[cur]+2);
            fail[now] = tr[getfail(fail[cur])][x];
            tr[cur][x] = now;
        } 
        last = tr[cur][x];
        cnt[last] |= op;
        if(cnt[last] == 3) ans = max(ans,len[last]);
    }
    void count(){
        for(int i = p-1;i > 0; --i) cnt[fail[i]] += cnt[i];
    }
}T;
char ss[maxn];
int main(){
    while(~scanf("%s",ss)){
        T.init();
        int len = strlen(ss);
        for(int i = 0;i < len; ++i) T.add(ss[i]-'a',1);
        scanf("%s",ss);
        len = strlen(ss);
        T.n = 0,T.last = 0;         
        for(int i = 0;i < len; ++i) T.add(ss[i]-'a',2);
        cout << T.ans << '\n';
    }
    return 0;
}