洛谷 3649 「APIO2014」回文串

发表于 | 分类于 |

学了学 PAM，就不写学习笔记了，因为比较简单。

PAM 模板题，建出 PAM 然后在 Fail 树上统计出现次数即可。
注意到 PAM 的 Fail 树一个拓扑序即为状态建立的顺序。

代码： 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 3e5;
int n;
char s[N + 5];
long long ans;
namespace PAM
{
    struct node
	{
        int ch[26];
        int fa,len,sz;
    } pam[N + 5];
    int las = 1,tot = 1;
    inline int init()
	{
        pam[1].len = -1,pam[0].fa = 1;
        return 0;
    }
    int Init = init();
    void insert(char *s,int i)
	{
        int cur = las,x = s[i] - 'a';
        for(;s[i - pam[cur].len - 1] ^ s[i];cur = pam[cur].fa);
        if(!pam[cur].ch[x])
        {
            int p = ++tot,q = pam[cur].fa;
            pam[p].len = pam[cur].len + 2;
            for(;s[i - pam[q].len - 1] ^ s[i];q = pam[q].fa);
            pam[p].fa = pam[q].ch[x],pam[cur].ch[x] = p;
        }
        ++pam[las = pam[cur].ch[x]].sz;
    }
    inline void build()
	{
        for(register int i = tot;i > 2;--i)
            pam[pam[i].fa].sz += pam[i].sz,ans = max(ans,(long long)pam[i].len * pam[i].sz);
    }
}
int main()
{
    scanf("%s",s + 1),n = strlen(s + 1);
    for(register int i = 1;i <= n;++i)
        PAM::insert(s,i);
    PAM::build();
    printf("%lld\n",ans);
}