Codeforces Round #728 (Div. 1) B Tree Array 题解

计不如人，肝败吓疯。

Description

传送门

Solution

翻译作品。

算法一

考虑固定一个根 $r$，作为序列中的第一个位置。从而，一个合法的 $\text{bfs}$ 序列，就等价于从 $r$ 往外扩展，每新扩展出一个节点就将其压入的序列。

看到期望逆序对数，不难想到枚举 $x,y(x>y)$，并求出 $x$ 在 $\text{bfs}$ 序列中比 $y$ 先出现的概率。此时，我们已经三重循环地枚举了 $r,x,y$，需要 $O(1)$ 计算其贡献。

我们从 $x$ 走到根的某个儿子，将路径上的所有点依次压入栈 $S_1$ 中。同理，通过从 $y$ 出发走到根的某个儿子，我们也得到了栈 $S_2$。问题等价于：

• 做多次操作，每次操作形如：

  • 找到两个栈的栈顶 $p_1,p_2$。
  • 若 $p_1=p_2$，那么我们有 $w$ 的概率将它们同时弹出，也有 $1-w$ 的概率啥也不做。
  • 若 $p_1\ne p_2$，那么我们有 $w_1$ 的概率弹出 $p_1$，也有 $w_2$ 的概率弹出 $p_2$（注意不能将它们同时弹出），也有 $1-w_1-w_2$ 的概率啥也不做。

• 求 $p_1$ 先被弹空的概率。

首先，一个观察是，$w_1=w_2$。原因显然: 令当前树上一共有 $k$ 个可扩展的位置，那么 $w_1=\frac{1}{k}$，$w_2=\frac{1}{k}$。更进一步，无论在什么时候，弹 $S_1,S_2$ 的栈顶的概率均等。

然后，还有一个观察: 当 $p_1=p_2$ 时，此时 $p_1$ 和 $p_2$ 既是 $x$ 的祖先，也是 $y$ 的祖先；同时，两个栈要么同时弹，要么同时不弹。显然，这对最终谁先被弹完并无影响，于是我们可以忽略一切是 $\text{LCA}(x,y)=l$ 的祖先的点（包含它自己），仅考虑以其为根的子树，并初始化 $S_1$ 为从 $x$ 到 $l$ 的某个儿子经过的所有节点，$S_2$ 为从 $y$ 到 $l$ 某个儿子经过的所有节点。

根据 $w_1=w_2$，我们可以不考虑 $w$ 的影响，从而这个贡献就是：

• 给定两个长度为 $d(x,l)$ 与 $d(y,l)$ 的栈，这里 $d$ 表示两点间的距离。

• 每次等概率地选择一个栈并将其栈顶弹出，求第一个栈先被弹空的概率。

考虑 $\text{dp}$。令 $f_{i,j}$ 表示，当两个栈长分别为 $i,j$ 时第一个栈先被弹空的概率，很容易得到：

$$f_{i,j}=\frac{f_{i,j-1}+f_{i-1,j}}{2}$$

边界: $\forall len\in [1,n],f_{0,len}=1$。

预处理出 $f$，得到结果的表达式

$$ans=\sum _{r=1}^n\sum _{u=1}^n\sum _{v=1}^{u-1}{f}_{d(u,l),d(v,l)}$$

总复杂度 $O(n^3)$，本题被解决。