LG P4548 [CTSC2006]歌唱王国（概率生成函数+KMP求border）

题目

设$F(x)={\sum }_{i=0}^{\infty }P(长度为i时结束)x^i$
$G(x)={\sum }_{i=0}^{\infty }P（长度为i时未结束）x^i$
$\because P(长度为i时结束)+P(长度为i时未结束)=P(长度为i-1时未结束)$
$\therefore F(x)+G(x)=G(x)x+1$
$\therefore F^{\prime }(x)+G^{\prime }(x)=G(x)x+G(x)$
$\therefore F^{\prime }(1)=G(1)$
$\because E={\sum }_{i=0}^{\infty }i\ast P(长度为i时结束)=(x{F}^{\prime }(x)){|}_{x=1}=F^{\prime }(1)$
$\therefore E=G(1)$

后面的过程比较抽象，开始说人话：
$Border$是指一个字符串中$s[1....k]=s[n-k+1.....n]$也就是相等的前后缀。
设$s$为名字的字符串。
一个没有结束的串$a$，我们考虑如何让它变成一个结束的串。
显然我们要考虑一个结束的串$b$是怎么从没有结束的串变来的。
我们发现$b$的后面是一个$s$，设去掉$s$后$b$变成了$a$。
那么$F(x)=G(x)\ast \frac{1}{m^{|s|}}\ast x^{|s|}?$
发现可能在加入$s$的一个前缀的时候就构成了$s$。
需要排除这种情况。
发现这种情况需要$s$的前缀是$border$, $a$的后缀是$s$去掉那个$border$后的串。
$F(x)=G(x)\ast \frac{1}{m^{|s|}}\ast x^{|s|}-{\sum }_{s[1...i]isaborder}F(x)\ast {\left(\frac{x}{m}\right)}^{|s|-i}$
我们需要求$G(1)$就令$x=1$，又因为$F(1)=1$
$G(1)=m^{|s|}+{\sum }_{s[1...i]isaborder}{m}^i$
如果我们把原串也看作一个$border$
就是
$E=G(1)={\sum }_{s[1...i]isaborder}{m}^i$
然后KMP求$border$就行，注意这个KMP和一般的不一样

AC Code:

#include<bits/stdc++.h>
#define maxn 100005
#define mod 10000
using namespace std;

int n,t,pw[maxn],a[maxn],nxt[maxn];

int main(){
	scanf("%d%d",&n,&t);
	pw[0] = 1;
	for(int i=1;i<=100000;i++) pw[i] = pw[i-1] * 1ll * n % mod;
	for(int m;t--;){
		scanf("%d",&m);
		for(int i=1;i<=m;i++)
			scanf("%d",&a[i]);
		nxt[0] = -1;
		for(int j=0,k=1;k<=m;)
			if(j==0 || a[k] == a[j]) nxt[k++] = j++;
			else j = nxt[j-1]+1;
		int ans = 0;
		for(int i=m;i;i=nxt[i])
			ans = (ans + pw[i]) % mod;
		printf("%04d\n",ans);
	}
}