UOJ #103. 【APIO2014】Palindromes (回文自动机模板题)

http://uoj.ac/problem/103

题意:

给你一个由小写拉丁字母组成的字符串 s。我们定义 s 的一个子串的存在值为这个子串在 s 中出现的次数乘以这个子串的长度。

对于给你的这个字符串 s,求所有回文子串中的最大存在值。

思路:

回文自动机,也叫回文树,但其实并不是树2333,所以以后还是称为回文自动机,缩写为PAM

学会了(?)SAM之后再看PAM真是简单得一逼。

学习笔记之后补。

对于这道题,PAM中一个状态的cnt表示的该状态所表示的回文串出现的次数

len表示的是该状态所表示的回文串的长度

UPDATE:

下面的代码风格太丑了,打算弃用。

更新的代码风格见最后

  1#include <iostream>
  2#include <cstdio>
  3#include <cstring>
  4#include <algorithm>
  5typedef long long LL;
  6#define ri register int
  7using namespace std;
  8#define MAXALP 30
  9#define ms(a,x) memset(a,x,sizeof(a)) 
 10const int N = 3E5+7;
 11struct PAM
 12{
 13    int cnt,len,fail;
 14    int nxt[MAXALP];
 15}st[N];
 16int n, m, sz , last, cur;
 17char s[N];
 18inline int new_node(int x)
 19{
 20    st[sz].len = x; st[sz].cnt = 0;
 21    ms(st[sz].nxt,0);
 22    return sz++;
 23}
 24inline int get_fail(int x, int n)
 25{
 26    while(s[n-st[x].len-1] != s[n]) x = st[x].fail;
 27    return x;
 28}
 29inline void pam_init()
 30{
 31    sz = 0 ;
 32    new_node(0);
 33    new_node(-1);
 34    st[0].fail = 1;
 35    ms(st[0].nxt,0);
 36    last = 0;
 37}
 38void pam_insert( int c,int head)
 39{
 40    cur = get_fail(last,head);
 41    if (!st[cur].nxt[c])
 42    {
 43    int nw = new_node(st[cur].len+2);
 44    st[nw].fail = st[get_fail(st[cur].fail,head)].nxt[c];
 45    st[cur].nxt[c] = nw;
 46    }
 47    last = st[cur].nxt[c];
 48    st[last].cnt++;
 49}
 50int main()
 51{
 52    scanf("%s",s);
 53    n = strlen(s);
 54    pam_init();
 55    for(int i=0;i<n;i++) pam_insert(s[i]-'a',i);
 56    for(int i=sz-1;i>=0;i--) st[st[i].fail].cnt += st[i].cnt;
 57    LL ans = 0;
 58    for(int i=2;i<sz;i++) ans = max(ans, 1LL*st[i].len*st[i].cnt);
 59    cout<<ans<<endl;
 60    return 0;
 61}
 62
 63
 64
 65
 66
 67
 68
 69
 70/* ***********************************************
 71Author :111qqz
 72Created Time :2017年11月14日 星期二 00时13分44秒
 73File Name :103.cpp
 74************************************************ */
 75
 76#include <bits/stdc++.h>
 77#define ms(a,x) memset(a,x,sizeof(a))
 78typedef long long LL;
 79using namespace std;
 80const int N=3E5+7;
 81struct PAM
 82{
 83    int fail,cnt,len;
 84    int nxt[26];
 85}st[N];
 86char S[N];
 87int n,now,sz;
 88void pam_init()
 89{
 90    st[0].fail = st[1].fail = 1;
 91    st[1].len = -1;
 92    sz = 1;
 93}
 94void extend(int c,int pos)
 95{
 96    int p = now;
 97    while (S[pos-st[p].len-1]!=S[pos]) p = st[p].fail;
 98    if (!st[p].nxt[c]){
 99    int np=++sz,q=st[p].fail;
100    st[np].len=st[p].len+2;
101    while (S[pos-st[q].len-1]!=S[pos]) q=st[q].fail;
102    st[np].fail=st[q].nxt[c];
103    st[p].nxt[c] = np;
104    }
105    now=st[p].nxt[c];
106    st[now].cnt++;
107}
108int main()
109{
110    scanf("%s",S);
111    pam_init();
112    for ( int i = 0 ,_=strlen(S); i < _ ; i++)
113        extend(S[i]-'a',i);
114    LL ans = 0 ;
115    for ( int i = sz ; i >= 1 ; i--)
116    {
117        st[st[i].fail].cnt += st[i].cnt;
118        ans = max(ans,1LL*st[i].len*st[i].cnt);
119    }
120    cout<<ans<<endl;
121    return 0;
122}