其他分享
首页 > 其他分享> > 回文自动机(PAM)复习

回文自动机(PAM)复习

作者:互联网

PAM(回文自动机)

应用

构建

模板

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N=300005;
namespace PAM{
    int sz,tot,last; //sz表示当前节点数,tot表示当前加入的字符个数
    int cnt[N],node[N][26],len[N],fail[N];
    char s[N];

    int new_node(int l){
        sz++;
        len[sz]=l;
        return sz;
    }
    void init()
    {
        sz=-1,last=0;
        s[tot=0]='$';
        new_node(0);
        new_node(-1);
        fail[0]=1;
    }
    int getfail(int x){
        while(s[tot-len[x]-1]!=s[tot]) x=fail[x];
        return x;
    }
    void insert(char c){
        s[++tot]=c;
        int now=getfail(last);
        if(!node[now][c-'a']){
            int x=new_node(len[now]+2);
            fail[x]=node[getfail(fail[now])][c-'a'];
            node[now][c-'a']=x;
        }
        last=node[now][c-'a'];
        cnt[last]++;
    }

}
char s[N];
int main()
{
    PAM::init();
    scanf("%s",s);
    for(int i=0;i<s[i];i++){
        PAM::insert(s[i]);
    }
}

image

例题

1.P3649 [APIO2014] 回文串

Problem

给你一个由小写拉丁字母组成的字符串 \(s\)。我们定义 \(s\) 的一个子串的存在值为这个子串在 \(s\) 中出现的次数乘以这个子串的长度。对于给你的这个字符串 \(s\),求所有回文子串中的最大存在值。

Solve

一个节点\(u\)通过\(fail\)指向的节点\(v\),\(v\)所代表的的回文串的出现次数一定包括\(u\)的出现次数,并且由于是按照拓扑序加入的,所以逆序累计出现次数即可。

Code

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N=300005;
namespace PAM{
    int sz,tot,last; //sz表示当前节点数,tot表示当前加入的字符个数
    int cnt[N],node[N][26],len[N],fail[N];
    char s[N];

    int new_node(int l){
        sz++;
        len[sz]=l;
        return sz;
    }
    void init()
    {
        sz=-1,last=0;
        s[tot=0]='$';
        new_node(0),new_node(-1);
        fail[0]=1;
    }
    int getfail(int x){
        while(s[tot-len[x]-1]!=s[tot]) x=fail[x];
        return x;
    }
    void insert(char c){
        s[++tot]=c;
        int now=getfail(last);
        if(!node[now][c-'a']){
            int x=new_node(len[now]+2);
            fail[x]=node[getfail(fail[now])][c-'a'];
            node[now][c-'a']=x;
        }
        last=node[now][c-'a'];
        cnt[last]++;
    }

    ll solve(){
        ll ans=0;
        for(int i=sz;i>=0;i--){
            cnt[fail[i]]+=cnt[i];
        }
        for(int i=1;i<=sz;i++){
            ans=max(ans,1LL*len[i]*cnt[i]);
        }
        return ans;
    }

}
char s[N];
int main()
{
    PAM::init();
    scanf("%s",s);
    for(int i=0;s[i];i++){
        PAM::insert(s[i]);
    }
    printf("%lld\n",PAM::solve());
}

参考

OIwiki

标签:node,sz,now,int,fail,自动机,PAM,回文
来源: https://www.cnblogs.com/Arashimu0x7f/p/16556851.html