acm-(组合计数、好题、期望、推式子)IMO Harder Problem The 2019 ICPC Vietnam National Contest

kattis传送门
vj传送门
本题非常考验数学推理能力，是一道非常不错的题目。先简化下题意：

设仅由$\text{H,T}$两个字符组成的串$\text{A}$中有$k$个$\text{H}$字符，然后我们可以进行一种操作，这种操作每次都会把$\text{A}[k]$反转(即原来是$\text{H}$反转后变成$\text{T}$，原来是$\text{T}$反转后则变成$\text{H}$)。现在如果我们一直重复这样的操作那么最终串$\text{A}$中的字符会全变成$\text{T}$(可以被证明，可以手算几个样例很容看出来)，假设需要操作$L(\text{A})$次恰好将$\text{A}$的字符变成全为$\text{T}$。
现在的问题如下：假设字符串$\text{S}$由$\text{H,T,?}$三种字符组成，其中$\text{?}$有$M$个，$\text{?}$号可以变成$\text{H}$或$\text{T}$字符，这样$\text{S}$可以变成$2^M$种仅含$\text{H}$和$\text{T}$的字符串，问你这些串对应的$L$的平均值是多少。

先考虑对于仅含$\text{H,T}$字符的$\text{S}$串如何计算$L(\text{S})$。

先给出结论：$L(\text{S})=2(a[1]+a[2]+a[3]+...+a[k|])-k^2$,其中$a[i]$表示$\text{S}$第串中$i$个$\text{H}$字符的位置(从1开始编号),$k$表示$\text{S}$串中的$\text{H}$字符数量

证明：假设$l\ge k$且$l$是最小的使得$\text{S}[l]=\text{H}$成立的位置。不难理解$l$是一定存在的，我们现在将这两个位置标记出来，如下面所示(下标代表在$\text{S}$串中的位置)：
$${\text{T}}_k{\text{T}}_{k+1}{\text{T}}_{k+2}....{\text{T}}_{l-2}{\text{T}}_{l-1}{\text{H}}_l$$
注意可能有$l=k$的情况出现，此时有${\text{H}}_{k=l}$。
我们现在模拟一下操作的流程
$第0次操作：{\text{T}}_k{\text{T}}_{k+1}{\text{T}}_{k+2}....{\text{T}}_{l-2}{\text{T}}_{l-1}{\text{H}}_l$
$第1次操作：{\text{H}}_k{\text{T}}_{k+1}{\text{T}}_{k+2}....{\text{T}}_{l-2}{\text{T}}_{l-1}{\text{H}}_l$
$第2次操作：{\text{H}}_k{\text{H}}_{k+1}{\text{T}}_{k+2}....{\text{T}}_{l-2}{\text{T}}_{l-1}{\text{H}}_l$
$\cdots$
$第l-k次操作：{\text{H}}_k{\text{H}}_{k+1}{\text{H}}_{k+2}....{\text{H}}_{l-2}{\text{H}}_{l-1}{\text{H}}_l$
$第l-k+1次操作：{\text{H}}_k{\text{H}}_{k+1}{\text{H}}_{k+2}....{\text{H}}_{l-2}{\text{H}}_{l-1}{\text{T}}_l$
$第l-k+2次操作：{\text{H}}_k{\text{H}}_{k+1}{\text{H}}_{k+2}....{\text{H}}_{l-2}{\text{T}}_{l-1}{\text{T}}_l$
$第l-k+3次操作：{\text{H}}_k{\text{H}}_{k+1}{\text{H}}_{k+2}....{\text{T}}_{l-2}{\text{T}}_{l-1}{\text{T}}_l$
$\cdots$
$第2(l-k)次操作：{\text{H}}_k{\text{T}}_{k+1}{\text{T}}_{k+2}....{\text{T}}_{l-2}{\text{T}}_{l-1}{\text{T}}_l$
$第2(l-k)+1次操作：{\text{T}}_k{\text{T}}_{k+1}{\text{T}}_{k+2}....{\text{T}}_{l-2}{\text{T}}_{l-1}{\text{T}}_l$
因此我们用了$2(l-k)+1$次操作让字符串$\text{S}$的$k到l$段完全被$\text{T}$字符所覆盖，并且我们成功让一个$\text{H}$字符从串$\text{S}$中消失，如果我们继续操作，会发现当前的$\text{H}$字符的数量变成了$k^{\prime }=k-1$，然后我们考虑$l^{\prime }\ge k^{\prime }$满足$l^{\prime }$是最小的使得$\text{S}[l^{\prime }]=\text{H}$成立的位置，同样地，如果我们想让$l^{\prime }$处的$\text{H}$字符消失的话就需要花费$2(l^{\prime }-k^{\prime })+1$次操作，以此类推…
我们发现每一轮都会使得$k$减一，也就是$\text{H}$字符减少一个，那么势必最终$\text{S}$串中的$\text{H}$字符将完全消失，也就达到了我们的目的，这也顺便证明了这种操作一定是有限的。
在第一轮中我们会操作$2(l-k)+1$次，第二轮中我们会操作$2(l^{\prime }-(k-1)+1$，第三轮中我们会操作$2(l^{\prime \prime }-(k-2))+1$，最终$k$会减成0，将这些操作次数加起来就能得到$2(l+l^{\prime }+l^{\prime \prime }+...)-2(k+(k-1)+(k-2)+...+1)+k=2(l+l^{\prime }+l^{\prime \prime }+...)-k^2$，然后注意到这些$l,l^{\prime },l^{\prime \prime }...$其实就是$\text{S}$串初始时候各个$\text{H}$字符的位置，这是因为我们每一轮操作都没有增加$\text{H}$字符，并且也没有将$\text{S}$串中除$l$位置以外的$\text{H}$消除，故我们每次都是再消除$\text{S}$串初始时候的那些$\text{H}$字符，故$l^{\prime }+l^{\prime \prime }+l^{\prime \prime \prime }+...=a[1]+a[2]+...+a[k]$，于是我们证明了结论是正确的。

现在我们将本题的答案分为两部分来解决：

1. $sum=2(a[1]+a[2]+...+a[k])$
   我们想要求出$sum$的期望，不妨改写一下这个式子，显然这个式子等价于$sum=2(1\cdot [\text{S}[1]=\text{H}]+2\cdot [\text{S}[2]=\text{H}]+...+n[\text{S}[n]=\text{H}])$，于是考虑第$i$个位置上的$[\text{S}[i]=\text{H}]$的期望是多少，这里分三种情况讨论：

   1. $\text{S}[i]=\text{H}$：显然期望是1
   2. $\text{S}[i]=\text{T}$：显然期望是0
   3. $\text{S}[i]=\text{?}$：显然期望是$\frac{1}{2}$

   这样不难得到$E(sum)={\sum }_{i=1}^{|\text{S}|}i(\{\begin{array}{ll}0& if\text{S}[i]=\text{T}\\ 1& if\text{S}[i]=\text{?}\\ 2& if\text{S}[i]=\text{H}\end{array})$。

2. $k^2$
   我们想求出$k^2$的期望，不妨考虑将所有可能的串的$k^2$加起来($\sum k^2$)然后除以串的总数($2^{?的数量}$)即可，于是我们要计算$\frac{\sum k^2}{2^{?的数量}}$，那么如何计算$\sum k^2$，考虑组合计数，我们枚举串中有多少个$\text{?}$变成$\text{H}$，假设个数为$i$，那么计算这样的串有多少种，求和即可。
   假设$\text{S}$串中有$n_{?}$个$\text{?}$，$n_H$个$\text{H}$，于是能够得到${\sum }_{i=0}^{n_{?}}{C}_{n_{?}}^i(n_H+i{)}^2$，这个式子做展开能够得到$n_H^2{\sum }_{i=0}^{n_{?}}{C}_{n_{?}}^i+2n_H{\sum }_{i=0}^{n_{?}}i{C}_{n_{?}}^i+{\sum }_{i=0}^{n_{?}}{i}^2{C}_{n_{?}}^i=n_H^2{2}^{n_{?}}+2n_H{n}_{?}{2}^{n_{?}-1}+n_{?}(n_{?}+1)2^{n_{?}-2}$，于是我们知道$\frac{\sum k^2}{2^{n_{?}}}=n_H^2+n_H{n}_{?}+\frac{n_{?}(n_{?}+1)}{4}$，这个式子可以直接求。

综上所述，我们通过将式子分成两部分用不同的方法把期望求出然后再将两部分相减即是最终的答案。

char s[maxn];
int main(){
	register int n=rd(s+1),n1=0,n2=0;
	register ll ans=0;
	FOR(i,1,n+1){
		if(s[i]=='H'){
			ans+=i+i;
			n1++;
		}else if(s[i]=='?'){
			ans+=i;
			n2++;
		}
	}
	ans-=1ll*n1*n1+1ll*n1*n2;
	wrn(ans-1ll*n2*(n2+1)/4.0);
} 

本题还可以扩展一下，假设串长为$n$，且全是$\text{?}$，那么期望是多少？显然可以用前面的结论直接求出答案，不过这里还有一种巧妙的方法来解决这个问题，我们考虑每次操作会出现一个数字改变，而且这种改变往往具有交替性，随便举个例子，比如$11011$，它的操作变更情况如下所示：
$11011$
$11001$
$11101$
$11111$
$11110$
$11100$
$11000$
$10000$
$00000$
我们观察上述序列，先是一次操作使得1变成0，然后是连续的2次操作使0变成1，最后是连续的5次操作使1变成0。我们将操作次数写成一个集合得到 { 5 , 2 , 1 } \{5,2,1\} {5,2,1}，显然对于任何一个确定的串(即不含$?$)我们都能够给出一个确定的集合，这相当于定义了一个串$S$到集合的映射关系，我们设这个集合是$\varphi (S)$，它代表了串$S$到集合$\varphi (S)$的一个映射，容易证明这是一个单射。
然后考虑集合$1,2,3,...,n$的任意子集$S^{\prime }$，定义函数${\varphi }^{-1}(S^{\prime })$为从$S^{\prime }$到某个长度为$n$的01串的映射，满足$S^{\prime }$恰好是串${\varphi }^{-1}(S^{\prime })$的操作次数构成的集合，这个串${\varphi }^{-1}(S^{\prime })$是按如下方式产生的：
假设 S ′ = { a 1 , a 2 , . . . , a k } ， ( a 1 > a 2 > . . . > a k ) S'=\{a_1,a_2,...,a_k\}，(a_1>a_2>...>a_k) S′={a1​,a2​,...,ak​}，(a1​>a2​>...>ak​)，先令串$s_0$为$00...0(n个0)$，我们从$i=1$开始到$k$对$s_0$进行$k$轮操作来得到${\varphi }^{-1}(S^{\prime })$串。
第1轮，我们花费$a_1$次操作将1~$a_1$位置的数置为1。
第2轮，我们花费$a_2$次操作将$a_1-a_2$到$a_1-1$位置的数置为0。
第3轮，我们花费$a_3$次操作将$a_1-a_2+1$到$a_1-a_2+a_3$位置的数置为1。
…
最终我们能够得到${\varphi }^{-1}(S^{\prime })$串，并且容易验证${\varphi }^{-1}(S^{\prime })$的操作集合恰好也是$S^{\prime }$，而且恰好${\varphi }^{-1}(S^{\prime })$也是单射。

于是通过$\varphi (S)$与${\varphi }^{-1}(S^{\prime })$我们建立了操作次数集合与01串之间相互映射的关系，更确切的来说，我们建立了集合 { 1 , 2 , 3 , . . . , n } \{1,2,3,...,n\} {1,2,3,...,n}的幂集(即集合 { 1 , 2 , 3 , . . . , n } \{1,2,3,...,n\} {1,2,3,...,n}的子集构成的集合)的元素与所有长度为$n$的01串组成的集合的元素的映射关系，由于两个都是单射，因此它们的映射是一个一一映射，即满足两个集合的势相等。

并且根据$S^{\prime }$集合的实际意义我们知道它的所有元素的和就是串$L(S)$的值，那么根据一一映射我们可以进一步得知长度为$n$的01串的$L$的期望就是集合 { 1 , 2 , 3 , . . . , n } \{1,2,3,...,n\} {1,2,3,...,n}的子集元素和的期望。

集合 { 1 , 2 , 3 , . . . , n } \{1,2,3,...,n\} {1,2,3,...,n}的子集元素和的期望也就很好算了，考虑每个元素只有取与不取两种状态，这意味着元素$i$对总期望的贡献为$\frac{i}{2}$，于是总期望为$\frac{1}{2}(1+2+3+...+n)=\frac{n(n+1)}{4}$。

于是我们说长度为$n$的01串的$L$值的期望为$\frac{n(n+1)}{4}$，是不是非常巧妙呢！

注意这个证明并不完整，有关为什么$S$与$S^{\prime }$之间是一一对应的关系的证明还需要严谨地推敲，不过并不是很难想，如果实在想不明白这里可以参考下面英文版的详细完整版的证明：