BZOJ 3676: [Apio2014]回文串

3676: [Apio2014]回文串

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 904  Solved: 356

Description

考虑一个只包含小写拉丁字母的字符串s。我们定义s的一个子串t的“出现值”为t在s中的出现次数乘以t的长度。请你求出s的所有回文子串中的最大出现值。 

Input

输入只有一行，为一个只包含小写字母(a -z)的非空字符串s。 

Output

输出一个整数，为逝查回文子串的最大出现值。 

Sample Input

【样例输入1】

abacaba

【样例输入2】

www

Sample Output

【样例输出1】

7

【样例输出2】

4

HINT

一个串是回文的，当且仅当它从左到右读和从右到左读完全一样。 

在第一个样例中，回文子串有7个：a，b，c，aba，aca，bacab，abacaba，其中： 

● a出现4次，其出现值为4：1：1=4 

● b出现2次，其出现值为2：1：1=2 

● c出现1次，其出现值为l：1：l=l 

● aba出现2次，其出现值为2：1：3=6 

● aca出现1次，其出现值为1=1：3=3 

●bacab出现1次，其出现值为1：1：5=5 

● abacaba出现1次，其出现值为1：1：7=7 

故最大回文子串出现值为7。 

【数据规模与评分】 

数据满足1≤字符串长度≤300000。

解题：回文树

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn = 300010;
const int N = 26;
struct PalindromicTree {
    int next[maxn][N],fail[maxn],cnt[maxn];
    int s[maxn],len[maxn],last,tot,n;
    int newnode(int length) {
        cnt[tot] = 0;
        memset(next[tot],0,sizeof next[tot]);
        len[tot] = length;
        return tot++;
    }
    int get_fail(int x) {
        while(s[n - len[x] - 1] != s[n]) x = fail[x];
        return x;
    }
    void init() {
        last = tot = n = 0;
        newnode(0);
        newnode(-1);
        s[n] = -1;
        fail[0] = 1;
    }
    void add(int c) {
        c -= 'a';
        s[++n] = c;
        int cur = get_fail(last);
        if(!next[cur][c]) {
            int now = newnode(len[cur] + 2);
            fail[now] = next[get_fail(fail[cur])][c];
            next[cur][c] = now;
        }
        last = next[cur][c];
        cnt[last]++;
    }
    void count() {
        for(int i = tot - 1; i > 1; --i)
            cnt[fail[i]] += cnt[i];
    }
} PT;
char str[maxn];
int main() {
    while(~scanf("%s",str)) {
        PT.init();
        for(int i = 0; str[i]; ++i) PT.add(str[i]);
        PT.count();
        LL ret = 0;
        for(int i = 2; i < PT.tot; ++i)
            ret = max(ret,(LL)PT.len[i]*PT.cnt[i]);
        printf("%lld\n",ret);
    }
    return 0;
}