NOI2014动物园

NOI2014动物园

题目描述

有一个$num$数组——对于字符串$S$的前$i$个字符构成的子串，既是它的后缀同时又是它的前缀，并且该后缀与该前缀不重叠，将这种字符串的数量记作$num[i]$。
例如$S$为$aaaaa$，则$num[4]=2$。
这是因为$S$的前$4$个字符为$aaaa$，其中$a$和$aa$都满足性质‘既是后缀又是前缀’，同时保证这个后缀与这个前缀不重叠。同理，$num[1]=0$,$num[2]=num[3]=1$,$num[5]=2$。
写一个程序求出num数组呢？
$\prod (i=1\dots n)=(num[i]+1)$ 对$1000000007$取模的结果即可。

输入格式

输入文件的第1行仅包含一个正整数$n$
表示测试数据的组数。 随后$n$行，每行描述一组测试数据。
每组测试数据仅含有一个字符串$S$。数据保证$S$中仅含小写字母。

输出格式

输出文件应包含$n$行
每行描述一组测试数据的答案

样例一

3
aaaaa
ab
abcababc

36
1
32

题解

先用$kmp$求出最大匹配，顺便记下该位置的前后缀匹配个数。
如果把不重叠这个条件去掉，$num[i]$就是问指针要在$next$数组上跳多少次才能跳到$0$。
假设我们处理第$i$位，一路匹配最终匹配到第$j$位，那么$num[i]=num[j]+1$.
要求不重叠，所以先跳到一个不重叠的子串，然后对每个位置求$next$，直到满足前缀长度不超过一半。

代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;

const ll mod = 1e9+7;
const int M = 1e6+10;
ll cnt;
char ch[M];
int next[M], num[M], n, len;

void add(int ai) { cnt *= ai; cnt %= mod;}

void kmp() {
	for(int i = 2, j = 0; i <= len; i++) {
		while(j && (ch[i] != ch[j+1])) j = next[j];
		ch[i] == ch[j+1] ? j++ : j=j;
		next[i] = j;
		num [i] = num[j]+1;
	}
	cnt = 1;
	for(int i = 2, j = 0; i <= len; i++) {
		while(j && ch[i] != ch[j+1]) j = next[j];
		ch[i] == ch[j+1] ? j++ : j=j;
		while((j << 1) > i)	j = next[j];
		add(num[j] + 1);
	}
}

int main() {
	scanf("%d", &n);
	while( n-- ) {
		memset(next, 0, sizeof(next));
		memset(num , 0, sizeof(num ));
		scanf("%s", ch+1);
		len = strlen(ch+1);
		num[1] = 1;  cnt = 1;
		kmp();
		printf("%lld\n", cnt);
	}
	return 0;
}