APIO2014 迴文串 迴文自動機

題意

定義\(s\)的一個子串的存在值為這個子串出現的次數乘子串的長度

求\(s\)的所有迴文串的存在值

分析

對\(s\)構建出迴文自動機，\(cnt\)表示當前結點當前的出現次數，那麼類似AC自動機的fail樹，倒序對fail樹的結點求siz即可得到結點的總出現次數

然後乘上len即可

程式碼

#include<bits/stdc++.h>
#define pii pair<int,int>
#define fi first
#define se second
#define eps 1e-9
#define db long double
#define equals(a,b) fabs(a-b) < eps
using namespace std;

typedef long long ll;



inline ll rd(){
	ll x;
	scanf("%lld",&x);
	return x;
}

const int maxn = 3e5 + 5;
char ss[maxn];

namespace pam {
	int sz, tot, last;
	int cnt[maxn], ch[maxn][26], len[maxn], fail[maxn];
	char s[maxn];
	int node(int l) {
  		sz++;
  		memset(ch[sz], 0, sizeof(ch[sz]));
  		len[sz] = l;
  		fail[sz] = cnt[sz] = 0;
  		return sz;
	}
	void clear() {
  		sz = -1;
  		last = 0;
  		s[tot = 0] = '$';
  		node(0);
  		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 (!ch[now][c - 'a']) {
    		int x = node(len[now] + 2);
    		fail[x] = ch[getfail(fail[now])][c - 'a'];
  	 	 	ch[now][c - 'a'] = x;
  		}
  		last = ch[now][c - 'a'];
  		cnt[last]++;
	}
}

int main(){
	scanf("%s",ss + 1);
	int len = strlen(ss + 1);
	pam::clear();
	for(int i = 1;i <= len;i++)
		pam::insert(ss[i]);
	ll ans = 0;
	for(int i = pam::sz;i >= 1;i--)
		pam::cnt[pam::fail[i]] += pam::cnt[i];
	for(int i = 1;i <= pam::sz;i++)
		ans = max(ans,(ll)(pam::cnt[i]) * (pam::len[i]));
	cout << ans;
}