洛谷 4555.最长双回文串

对正串反串分别建 PAM,求出以每个位置为结尾 / 开头的最长回文子串即可。

代码:

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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1e5;
int n;
char s[N + 5],t[N + 5];
int l[N + 5],r[N + 5],ans;
namespace PAM1
{
struct node
{
int ch[26];
int fa,len;
} 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;
}
las = pam[cur].ch[x];
}
}
namespace PAM2
{
struct node
{
int ch[26];
int fa,len;
} 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;
}
las = pam[cur].ch[x];
}
}
int main()
{
scanf("%s",s + 1),n = strlen(s + 1);
for(register int i = 1;i <= n;++i)
PAM1::insert(s,i),t[i] = s[n - i + 1],l[i] = PAM1::pam[PAM1::las].len;
for(register int i = 1;i <= n;++i)
PAM2::insert(t,i),r[n - i + 1] = PAM2::pam[PAM2::las].len;
for(register int i = 1;i < n;++i)
ans = max(ans,l[i] + r[i + 1]);
printf("%d\n",ans);
}