BZOJ1511: [POI2006]OKR-Periods of Words

1511: [POI2006]OKR-Periods of Words

Time Limit: 5 Sec  Memory Limit: 64 MB
Submit: 330  Solved: 200
[Submit][Status][Discuss]

Description

一 个串是有限个小写字符的序列,特别的,一个空序列也可以是一个串. 一个串P是串A的前缀, 当且仅当存在串B, 使得 A = PB. 如果 P A 并且 P 不是一个空串,那么我们说 P 是A的一个proper前缀. 定义Q 是A的周期, 当且仅当Q是A的一个proper 前缀并且A是QQ的前缀(不一定要是proper前缀). 比如串 abab 和 ababab 都是串abababa的周期. 串A的最大周期就是它最长的一个周期或者是一个空串(当A没有周期的时候), 比如说, ababab的最大周期是abab. 串abc的最大周期是空串. 给出一个串,求出它所有前缀的最大周期长度之和.

Input

第一行一个整数 k ( 1 k 1 000 000) 表示串的长度. 接下来一行表示给出的串.

Output

输出一个整数表示它所有前缀的最大周期长度之和.

Sample Input

8
babababa

Sample Output

24

HINT

Source

 

【题解】

之前有n - nxt[n]是最小循环节

只需要不断往前跳nxt，跳到不能再跳

还是n - nxt

递推即可

我记得形成的东西好像叫KMP树来着？还是nxt树？还是fail树?

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <queue>
#include <vector>
#include <cmath> 
#define min(a, b) ((a) < (b) ? (a) : (b))
#define max(a, b) ((a) > (b) ? (a) : (b))
#define abs(a) ((a) < 0 ? (-1 * (a)) : (a))
inline void swap(long long &a, long long &b)
{
    long long tmp = a;a = b;b = tmp;
}
inline void read(long long &x)
{
    x = 0;char ch = getchar(), c = ch;
    while(ch < '0' || ch > '9') c = ch, ch = getchar();
    while(ch <= '9' && ch >= '0') x = x * 10 + ch - '0', ch = getchar();
    if(c == '-') x = -x;
}

const long long INF = 0x3f3f3f3f;
const long long MAXN = 2000000 + 10;

char s[MAXN];
long long nxt[MAXN], cnt[MAXN], len; 

int main()
{
    read(len);scanf("%s", s);
    nxt[0] = -1;
    for(register long long i = 1, j = -1;i < len;++ i)
    {
        while(j >= 0 && s[i] != s[j + 1]) j = nxt[j];
        if(s[i] == s[j + 1]) ++ j;
        nxt[i] = j;
    }
    long long sum = 0;
    for(register long long i = 0;i < len;++ i)
    {
        long long now = nxt[i];
        if(now != -1 && cnt[now] != -1) now = cnt[now];
        if(now != -1) sum += i - now;
        cnt[i] = now;
    }
    printf("%lld", sum);
    return 0;
}