https://www.luogu.org/problemnew/show/P3649
这道题就是一个板子题;
回文自动机这个东西,前几天刚刚听说,看了两天感觉不是怎么好用呢。网上的博客写得也很不错,今天在这里写一个板子。
回文自动机主要就是有两个根,奇数回文串,偶数回文串两个根,fail的和AC自动机差不多,主要注意这些吧。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=300010;
struct Palindromic_Tree{
int nxt[N][30],fail[N],cnt[N];
int num[N],len[N],s[N];
int last,n,p;
int newnode(int l)
{
memset(nxt[p],0,sizeof(nxt[p]));
cnt[p]=num[p]=0;
len[p]=l;
return p++;
}
void init()
{
p=0;
newnode(0);
newnode(-1);
last=n=0;
s[0]=-1;
fail[0]=1;
}
int get_fail(int x)
{
while(s[n-len[x]-1]!=s[n]) x=fail[x];
return x;
}
void add(int c)
{
c-='a';
s[++n]=c;
int cur=get_fail(last);
if(!nxt[cur][c])
{
int now=newnode(len[cur]+2);
fail[now]=nxt[get_fail(fail[cur])][c];
nxt[cur][c]=now;
num[now]=num[fail[now]]+1;
}
last=nxt[cur][c];
cnt[last]++;
}
ll Count()
{
ll ret=0;
for(int i=p-1;i>=0;i--) cnt[fail[i]]+=cnt[i];
for(int i=0;i<p;i++)
ret=max(ret,1LL*cnt[i]*len[i]);
return ret;
}
}pam;
char str[N];
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>str;
int len=strlen(str);
pam.init();
for(int i=0;i<len;i++) pam.add(str[i]);
cout<<pam.Count()<<endl;
return 0;
}
1楼的评论,主要是fail和初始化有点不一样。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 300010;
struct Palindromic_Tree {
int nxt[N][30], fail[N], cnt[N];
int num[N], len[N], s[N];
int last, n, p;
int newnode(int l) {
memset(nxt[p], 0, sizeof(nxt[p]));
cnt[p] = num[p] = 0;
len[p] = l;
return p++;
}
void init() {
p = 0;
newnode(-1);
newnode(0);
n = 0;
last = 1;
s[0] = -1;
fail[1] = 0;
}
int get_fail(int x) {
while (s[n - len[x] - 1] != s[n])
x = fail[x];
return x;
}
void add(int c) {
c -= 'a';
s[++n] = c;
int cur = get_fail(last);
if (!nxt[cur][c]) {
int now = newnode(len[cur] + 2);
if (nxt[get_fail(fail[cur])][c] != 0)
fail[now] = nxt[get_fail(fail[cur])][c];
else fail[now] = 1;
nxt[cur][c] = now;
num[now] = num[fail[now]] + 1;
}
last = nxt[cur][c];
cnt[last]++;
}
ll Count() {
ll ret = 0;
for (int i = p - 1; i >= 0; i--) cnt[fail[i]] += cnt[i];
for (int i = 0; i < p; i++)
ret = max(ret, 1LL * cnt[i] * len[i]);
return ret;
}
} pam;
char str[N];
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin >> str;
int len = strlen(str);
pam.init();
for (int i = 0; i < len; i++) pam.add(str[i]);
cout << pam.Count() << endl;
return 0;
}