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 long
char 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;
}