【解题报告】[JLOI2012] 时间流逝

简化题意：

给定一长为 $n$ 的有序序列 $a$，序列内元素两两不同。你现在有一个栈，初始为空，你会不断执行以下操作直到栈内元素之和 $\ge T$ ，其中 $T$ 为给定值。问操作终止时进行操作次数的期望。一次操作过程如下：

• 如果栈不为空，则有 $p$ 的概率弹出一个元素。
• 如果没弹，在所有 $\le$ 栈顶元素的 $a_i$ 里随机地取一个，将其入栈（不把入栈元素 $a_i$ 删掉，下一次也可能继续入栈）。

首先栈内元素一定是有序的，因为每次入栈元素都不大于栈顶元素。也就是说，当前栈内状态如果是通过当前栈顶元素入栈得到的，而不是弹栈，那么可以唯一确定栈顶元素入栈前的栈内状态。当然，由当前栈内状态可以推出多个后继状态，包括弹栈，也就是由当前状态回到上一个状态。

将唯一的前驱状态看成父节点，将状态关系用图形表示，会形成一棵树，叶节点即为当前节点的状态已经满足元素之和 $\ge T$ 的节点。

如输入

0.5 3 2
1 2

由图可直观得出一次操作即在树上移动一步。
考虑 DP。设 $d{p}_x$ 为到达当前的节点 $x$ 的期望操作次数，$so{n}_x$ 为 $x$ 的后继状态，$cn{t}_x$ 为 $x$ 的后继状态的个数。那么有：

$$d{p}_x=p\ast (d{p}_{fa}+1)+\sum _{y\in so{n}_x}\frac{(1-p)(d{p}_y+1)}{cn{t}_x}$$

化简可得：

$$d{p}_x=p\ast d{p}_{fa}+1+\frac{(1-p)\ast {\sum }_{y\in so{n}_x}d{p}_y}{cn{t}_x}$$

上面的状态转移存在后效性，于是尝试从叶节点入手，当 $x$ 为叶节点时，有状态转移：

$$d{p}_x=p\ast d{p}_{fa}+1$$

发现是 $y=kx+b$ 的结构，可以用归纳法证明，可以将普通的节点也用父节点的状态表示：

假设当前节点 $x$ 的子节点 $y$ 可以用 $x$ 的状态表示为 $d{p}_y=k_y\ast d{p}_x+b_y$，于是有：

$$d{p}_x=p\ast d{p}_{fa}+1+\frac{(1-p)\ast {\sum }_{y\in so{n}_x}d{p}_y}{cn{t}_x}⟺d{p}_x=\frac{p}{1-\frac{1-p}{cn{t}_x}\ast \sum k_y}\ast d{p}_{fa}+\frac{1+\frac{1-p}{cn{t}_x}\ast \sum b_y}{1-\frac{1-p}{cn{t}_x}\ast \sum k_y}$$

由于 $k_y,b_y$ 已知，可以得出 $k_x,b_x$，证毕。

实现方面，考虑搜索，从根节点开始，递归至叶节点返回。