[CTSC2006] 歌唱王国（概率生成函数 + KMP / 哈希）

题面

洛谷P4548 [CTSC2006]歌唱王国

分析

先了解一下概率生成函数，对于一个随机变量$a$，它的概率生成函数是$$F(x)=\sum _{i\ge 0}(\text{Pr}(a=i)\cdot x^i)$$对它求导得$$F^{\prime }(x)=\sum _{i\ge 0}(\text{Pr}(a=i)\cdot i\cdot x^{i-1})$$根据期望的定义可得$\mathbb{E}(a)=F^{\prime }(1)$。

---

那么对于这道题，设$f_i$表示，长度为$i$的串结束（找到匹配）的概率，$g_i$表示长度为$i$的串未结束（还没找到匹配）的概率。$f_0=0$，$g_0=1$。
对随机变量$a_1:=$结束时的长度 和随机变量$a_2:=$未结束时的长度 分别定义概率生成函数：$$F(x)=\sum _{i\ge 0}(f_i\cdot x^i)$$$$G(x)=\sum _{i\ge 0}(g_i\cdot x^i)$$可知题目要求的是$\mathbb{E}(a_1)=F^{\prime }(1)$。

---

为什么要定义$g$，当然是因为只定义$f$不好算，定义了$g$过后，我们能写$g$的转移式：$$g_i=f_{i+1}+g_{i+1}$$相当于往长度为$i$的串后面加一个字符，看能否结束，根据加法原理将两个概率相加。

于是有$$\begin{array}{rl}F(x)+G(x)& =\sum _{i\ge 0}\left[(f_i+g_i)\cdot x^i\right]\\ & =\sum _{i\ge 1}(g_{i-1}\cdot x^i)+1\\ & =\sum _{i\ge 0}(g_i\cdot x^{i+1})+1\\ & =G(x)\cdot x+1\end{array}$$两边同时求导得$$F^{\prime }(x)+G^{\prime }(x)=G^{\prime }(x)\cdot x+G(x)$$于是$$\mathbb{E}(x)=F^{\prime }(1)=G(1)$$

---

接下来就要考虑$G(1)$怎么算了。直觉告诉我们，两个变量得两个方程，所以还需要找到一个本质不同的$g$与$f$的关系，这个时候肯定得跟原串（牛人酋长的名字）$A$扯上关系。既然$g_i$是没有结束的概率，那么我们强制让它结束掉，不就和$f$扯上关系了么。设字符集大小为$n$，$A$的长度为$L$。考虑把$A$强制接在$g_i$对应的某个串后面：$$g_i\cdot {\left(\frac{1}{n}\right)}^L=\sum _{A[1,\cdots ,d]=A[L-d+1,\cdots ,L]}{f}_{i+d}\cdot {\left(\frac{1}{n}\right)}^{L-d}$$

如图对应了一种$f_{i+d}\cdot {\left(\frac{1}{n}\right)}^{L-d}$的情况，即阴影部分相等。对于$g_i$，强制规定上方的橙色部分，对于$f_{i+d}$，强制规定白色方框部分。由于下方的橙色部分可以移动，所以对所有border的情况求和。

令$r_i=[A[1,\cdots ,d]=A[L-d+1,\cdots ,L]]$，根据上式就有$$G(x)\cdot {\left(\frac{1}{n}\right)}^L\cdot x^L=\sum _{1\le i\le L}\left(r_i\cdot F(x)\cdot x^{L-i}\right)$$这个式子我只能暴力证明它是对的：怎么来的就只能感性认知了

---

于是得到$$G(1)\cdot {\left(\frac{1}{n}\right)}^L=F(1)\cdot \sum _{1\le i\le L}(r_i\cdot n^{i-L})$$注意$F(1)=\sum _{i\ge 0}{f}_i=1$，因为无论怎样必定有一个时刻会结束，所以$$\mathbb{E}(a_1)=G(1)=\sum _{1\le i\le L}(r_i\cdot n^i)$$哈希或者KMP处理border（$r_i$），结束。

代码

#include <algorithm>
#include <cstdio>
#include <cstring>

const int MAXN = 100000;
const int MOD = 10000;

int Pow(int x, int y) {
	int ret = 1;
	while (y) {
		if (y & 1)
			ret = ret * x % MOD;
		y >>= 1;
		x = x * x % MOD;
	}
	return ret;
}

int N, L;
int S[MAXN + 5], Fail[MAXN + 5];

int main() {
	int T, N;
	scanf("%d%d", &N, &T);
	N %= MOD;
	while (T--) {
		scanf("%d", &L);
		memset(Fail, 0, sizeof Fail);
		for (int i = 1; i <= L; i++)
			scanf("%d", &S[i]);
		for (int i = 2; i <= L; i++) {
			int k = Fail[i - 1];
			while (k && S[k + 1] != S[i])
				k = Fail[k];
			if (S[k + 1] == S[i])
				Fail[i] = k + 1;
		}
		int Ans = 0;
		for (int i = L; i; i = Fail[i])
			Ans = (Ans + Pow(N, i)) % MOD;
		printf("%04d\n", Ans);
	}
    return 0;
}