51nod1277 字符串中的最大值（KMP）

1277 字符串中的最大值

题目来源： Codility

基准时间限制：1 秒 空间限制：131072 KB 分值: 80 难度：5级算法题

一个字符串的前缀是指包含该字符第一个字母的连续子串，例如：abcd的所有前缀为a, ab, abc, abcd。

给出一个字符串S，求其所有前缀中，字符长度与出现次数的乘积的最大值。

例如：S = "abababa" 所有的前缀如下:

 

"a", 长度与出现次数的乘积 1 * 4 = 4,

"ab",长度与出现次数的乘积 2 * 3 = 6,

"aba", 长度与出现次数的乘积 3 * 3 = 9,

"abab", 长度与出现次数的乘积 4 * 2 = 8,

"ababa", 长度与出现次数的乘积 5 * 2 = 10,

"ababab", 长度与出现次数的乘积 6 * 1 = 6,

"abababa", 长度与出现次数的乘积 7 * 1 = 7.

其中"ababa"出现了2次，二者的乘积为10，是所有前缀中最大的。

Input

输入字符串S, (1 <= L <= 100000， L为字符串的长度)，S中的所有字符均为小写英文字母。

Output

输出所有前缀中字符长度与出现次数的乘积的最大值。

Input示例

abababa

Output示例

10

#include<bits/stdc++.h>using namespace std;#define N 100005#define LL long longchar s[N];LL cnt[N];void compute_prefix(int *Next, char *p){    int n, k;    n = strlen(p);    Next[0] = Next[1] = 0;    k = 0;    for(int i = 2; i <= n; i++)    {        while(k && p[k] != p[i-1])            k = Next[k];        if(p[k] == p[i-1])            k++;        Next[i] = k;    }}void kmp_match(char *text, char *p, int *Next){    int ans = 0, j = 0;    int text_len = strlen(text);    int p_len = strlen(p);    for(int i = 0; i < text_len; i++)    {        while(j && text[i] != p[j])            j = Next[j];        if(text[i] == p[j])            j++;        cnt[j]++;    }}LL solve(){    memset(cnt, 0, sizeof(cnt));    int Next[N];    compute_prefix(Next, s);    kmp_match(s, s, Next);    for(int i = strlen(s); i >= 1; i--)        cnt[Next[i]] += cnt[i];    LL ans = 0;    for(int i = 1; i <= strlen(s); i++)        ans = max(ans, 1LL*cnt[i]*i);    return ans;}int main(){    while(~scanf("%s", s))    {        cout << solve() << endl;    }    return 0;}