【香蕉OI】Fibonacci's Nightmare（概率期望）

题意

（仅描述该题的80分部分分）

一个数列 { a n } \{a_n\} {an​}由如下方式生成：

1. $a_0=1$
2. 对于每一个$a_i(i\ge 1)$，再$[0,i-1]$中等概率且独立随机选两个整数$j,k$，$a_i=a_j+a_k$。

给出n，求$E(a_n^2)$，即$a_n$的平方的期望。

思路

大力推式子：

$$E(a_i^2)=\frac{{\sum }_{j=0}^{i-1}{\sum }_{k=0}^{i-1}E[(a_j+a_k{)}^2]}{n^2}=\frac{{\sum }_{j=0}^{i-1}{\sum }_{k=0}^{i-1}E[a_j^2+a_k^2+2\ast a_j{a}_k]}{n^2}$$

因为$a_j$和$a_k$是互相独立的，所以$E[a_j^2+a_k^2+2\ast a_j{a}_k]$可以拆开：

$$E(a_i^2)=\frac{{\sum }_{j=0}^{i-1}{\sum }_{k=0}^{i-1}[E(a_j^2)+E(a_k^2)+2\ast E(a_j{a}_k)]}{n^2}$$

$$=\frac{2\ast {\sum }_{j=0}^{i-1}E(a_j^2)}{n}+\frac{2\ast E({\sum }_{j=0}^{i-1}{\sum }_{k=0}^{i-1}{a}_j{a}_k)}{n^2}$$

第一项是一个前缀和。现在来推第二项，尝试递推（为了方便j,k的范围变成了[0,i]）：

$$E(\sum _{j=0}^i\sum _{k=0}^i{a}_j{a}_k)=E[(\sum _{j=0}^i{a}_j{)}^2]$$

$$=E[(\sum _{j=0}^{i-1}{a}_j+a_i{)}^2]$$

$$=E[(\sum _{j=0}^{i-1}{a}_j{)}^2+2\ast a_i\ast \sum _{j=0}^{i-1}{a}_j+a_i^2]$$

$$=E[(\sum _{j=0}^{i-1}{a}_j{)}^2]+2\ast E(a_i\ast \sum _{j=0}^{i-1}{a}_j)+E(a_i^2)$$

又因为

$$E(a_i)=\frac{2\ast {\sum }_{j=0}^{i-1}{a}_j}{n}$$

代入得：

$$E(\sum _{j=0}^i\sum _{k=0}^i{a}_j{a}_k)=E[(\sum _{j=0}^{i-1}{a}_j{)}^2]+E(a_i^2)+2\ast E[(\sum _{j=0}^{i-1}{a}_j{)}^2\ast \frac{2}{n}]$$

$$=E[(\sum _{j=0}^{i-1}{a}_j{)}^2]+E(a_i^2)+\frac{4\ast E[({\sum }_{j=0}^{i-1}{a}_j{)}^2]}{n}$$

$$=(1+\frac{4}{n})E[(\sum _{j=0}^{i-1}{a}_j{)}^2]+E(a_i^2)$$

可以递推。

于是此题做完了。O(n)。

期望好题。我对期望一无所知。

注意

1.

关注期望的定义$$E(x)=\sum p_i\ast x_i$$

把期望展开再推。

2.

期望的平方$\ne$平方的期望。

展开就明白了。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e5 + 10, mod = 998244353;
int n, k;
LL a2[N], s1[N], s2[N], inv[N];

inline LL fpow(LL x, LL y, LL p){
	LL r = 1;
	while (y){
		if (y&1) r = r*x%p;
		x = x*x%p;
		y >>= 1;
	}
	return r;
}

int main()
{
	cin >> n >> k;
	for (int i = 1; i <= n; ++ i)
		inv[i] = fpow(i, mod-2, mod);
	a2[0] = s1[0] = s2[0] = 1;
	for (int i = 1; i <= n; ++ i){
		a2[i] = (2 * s2[i-1] * inv[i] % mod + 2 * s1[i-1] * inv[i] % mod * inv[i] % mod) % mod;
		s1[i] = ((1 + 4 * inv[i] % mod) * s1[i-1] % mod + a2[i]) % mod;
		s2[i] = (s2[i-1] + a2[i]) % mod;
	}
	printf("%lld\n", a2[n]);
	return 0;
}