[AGC007C]Pushing Balls

题目

传送门 to AtCoder

思路

这是一道人类智慧题；果然我不配做人！

最 $\mathrm{n}\mathrm{a}\mathrm{i}\mathrm{v}\mathrm{e}$ 的想法是 $\mathcal{O}(n^2)$ 的 $\mathtt{d}\mathtt{p}$ 求出每一个小段的期望经过次数。没找到优化点。

我们是不是应该考虑一下 距离是等差 这个别有用心的设计？——没错，就是 “别有用心” 而不是 “别具匠心”。

本来 $\mathtt{d}\mathtt{p}$ 就是递归子问题，但是它更彻底。设原距离数组是 $a_x=dx+b$，考虑进行一次操作后，剩下的长度为 $(2n-2)$ 的距离数组的每一项的 期望值（就是概率乘长度的和嘛，相当于在递归中的贡献系数）。

为了方便，可以忽略 “球” 和 “洞” 的身份，笼统说为 “相邻元素可同时删去”。那么，对于第 $x(1⩽x⩽2n-2)$ 项，若删去前面 $a_y(1⩽y⩽x)$ 的左右端点，则 $a_y$ 会再拓展 $2$ 个区间，使得 $a_x^{\prime }=a_{x+2}$（顺次补齐两个空洞）。概率是 $\frac{x}{2n}$ 。

若删去 $a_{x+1}$ 的左右端点则 $a_x^{\prime }=a_x+a_{x+1}+a_{x+2}$，概率是 $\frac{1}{2n}$ 。其他情况下 $a_x^{\prime }=a_x$，概率是 $\frac{2n-x-1}{2n}$ 。三者相加，得到新期望长度
$$\begin{array}{rl}{a}_x^{\prime }& =\frac{x{a}_{x+2}+3a_{x+1}+(2n-x-1)a_x}{2n}\\ & =\frac{(2n-1)a_x+2dx+3a_{x+1}}{2n}\end{array}$$

于是 $a_x^{\prime }-a_{x-1}^{\prime }=\frac{(2n-1)d+2d+3d}{2n}=\frac{n+2}{n}d$ 为常数。新的期望距离仍然是等差数列！

所以，我们递归到一个 $a_x=\frac{n+2}{n}dx+b^{\prime }$ 的情况，仍然可以用上面的方法计算。现在只需要快速求出第一次操作提供的贡献。

第一次操作时，每个距离都可以被等概率选中；期望应为 $\frac{\sum a_i}{2n}=\frac{a_1+a_{2n}}{2}$，等差数列的平均值嘛。每次 $\mathcal{O}(1)$，时间复杂度就 $\mathcal{O}(n)$ 了。

代码

极度舒适的代码 😎

#include <cstdio>
int main(){
	int n; double ans, head, d;
	scanf("%d %lf %lf",&n,&head,&d);
	for(ans=0; n; d=d*(n+2)/n,--n){
		ans += head+((n<<1)-1)*d/2;
		head += (2*head+5*d)/(n<<1);
	}
	printf("%.12f\n",ans);
	return 0;
}