Palindrome Mouse 迴文自動機

link

---

考慮每個點對答案貢獻爲除他本身外，所在fail鏈上節點個數和迴文樹上該節點祖先節點個數。
這樣寫會有兩個問題
1.重複節點
2.時間複雜度
考慮遞推，該節點的貢獻爲其父親節點貢獻+它所延伸出fail鏈的個數+1

---

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;;
const int SZ = 26;///字符集
const int maxn = 1e5 + 6;
struct PAM {
    struct PamNode{int fail,trans[SZ],sz,len,num;}pam[maxn];
    int tot;char s[maxn];
    void init() {
        tot=1;
        pam[0].fail=1;pam[0].len=0;
        pam[1].fail=1;pam[1].len=-1;
        memset(pam[0].trans,0,SZ*sizeof (int));
        memset(pam[1].trans,0,SZ*sizeof (int));
    }
    inline int newnode(int len) {
        tot++;
        memset(pam[tot].trans,0,SZ*sizeof (int));
        pam[tot].len=len;pam[tot].fail=0;
        pam[tot].sz=0;pam[tot].num=0;
        return tot;
    }
    inline int getfail(int i,int u) {
        while(s[i-pam[u].len-1]^s[i]) u=pam[u].fail;
        return u;
    }
    inline int append(int i,int u) {
        int c=s[i]-'a';
        int fa=getfail(i,u);
        u=pam[fa].trans[c];
        if(!u) {
            int z=newnode(pam[fa].len+2);
            int w=getfail(i,pam[fa].fail);
            pam[z].fail=pam[w].trans[c];
            u=pam[fa].trans[c]=z;///注意這裏要後更新，否則上面getfail時可能導致死循環
            pam[z].num=pam[pam[z].fail].num+1;
/*還有一點我們可以指出, 這裏屬於z代表的迴文子串第一次出現, 之前讀入s[1,..,i-1]都是沒有出現過該回文子串
的, 所以可以維護出每種迴文子串第一次出現的索引.只是這裏沒做而已.
*/
        }
        pam[u].sz++;return u;
    }
    void calu() {
        for(int i=tot,fail;i>1;i--) {
            fail=pam[i].fail;
            pam[fail].sz+=pam[i].sz;
        }
    }
    ll ans=0,dp[maxn],vis[maxn];
    int jmp(int u) {
        int cnt=0;
        while(u>1&&!vis[u]) {
            vis[u]=1;
            u=pam[u].fail;cnt++;
        }
        return cnt;
    }
    void clr(int u,ll kk) {
        for(int i=1;i<=kk;i++) {
            vis[u]=0;
            u=pam[u].fail;
        }
    }

    void dfs(int u,int fa) {
        ll kk=jmp(u);
        if(u>1) dp[u]=kk-1;
        if(fa>1) dp[u]+=dp[fa]+1;
        ans+=dp[u];
        for(int i=0;i<26;i++) {
            if(pam[u].trans[i]) {
                dfs(pam[u].trans[i],u);
            }
        }
        clr(u,kk);
    }
    ll cont() {
        ans=0;
        dfs(0,0);dfs(1,1);
        return ans;
    }
}pa;
int n,_,cs=0;
int main() {
    for(scanf("%d",&_);_;_--) {
        scanf("%s",pa.s+1);
        n=strlen(pa.s+1);
        pa.init();
        for(int i=1,last=0;i<=n;i++) {
            last=pa.append(i,last);
        }
        printf("Case #%d: %lld\n",++cs,pa.cont());
    }
}