「营业日志 2020.11.26」一道纳什均衡数数题

和 xzz 讨论的一道集训队自选题。

简要题意 一颗二叉树，划分成 $A,B,L$ 三部分，其中 $L$ 为叶节点，其余点均有两个孩子。每个叶节点有 $c,d$ 值，一组策略为给每个点选择一个重子，称双方的收益为从根沿重子走到的对应 $c,d$ 值。称一个策略纳什均衡当且仅当任取 $A$ 中策略改变均不使得 $c$ 更大，任取 $B$ 中策略改变均不使 $d$ 更大。对所有 $\forall u\in L,1\le c_u,d_u\le k$ 的情况，对纳什均衡的策略数求和。

首先一个简单的想法是考虑一个朴素的 DP，$F_u(i,j)$ 表示 $u$ 所在的子树达到纳什均衡，其中 $A$ 的最优策略达到了 $i$，$B$ 的最优策略达到了 $j$。我们来先看一下假设 $u\in A$ 的情况。其 $u\in B$ 的转移是完全对称的：

设两子为 $l,r$。若以 $l$ 为重子，注意此时 $l$ 的子树维持纳什均衡，但 $r$ 的子树只需保证修改 $A$ 可达的最大值 $\le i$。我们记 $X_r(i)$ 表示最大值 $\le i$ 的情况。那么就有

$$F_u(i,j)\leftarrow F_l(i,j)\cdot X_r(i)$$

对称地，我们可得此时有

$$F_u(i,j)=F_l(i,j)\cdot X_r(i)+F_r(i,j)\cdot X_l(i)$$

而此时的 $X,Y$ 呢？对于本层的 $X$，两个子树均能取到，故有

$$X_u(i)=2X_l(i)\cdot X_r(i)$$

而对于 $Y$（$B$ 的最优化）来说，这被 $A$ 完全摆布，另一子树的细节就没用了，所以是

$$Y_u(i)=Y_l(i)\cdot 2^{|A_r|+|B_r|}{k}^{2|L_r|}+Y_r(i)\cdot 2^{|A_l|+|B_l|}{k}^{2|L_l|}$$

不难看出 $X,Y$ 是多项式，但 $F$ 是个二元多项式。然而经过归纳，我们发现 $F$ 的两元是实质上独立的！我们考虑归纳，若

$$F_u=U_u\cdot V_u$$

有良定义的关系

$$\begin{array}{rl}{X}_u(i)& =2^{|A_u|}{k}^{|L_u|}{U}_u(i)\cdot i\\ Y_u(i)& =2^{|B_u|}{k}^{|L_u|}{V}_u(i)\cdot i\end{array}$$

我们进行归纳。在叶节点处，令 $U_u(i)=V_u(i)=1$ 即可。还是考虑 $A$ 位置。

$$\begin{array}{rl}{F}_u(i,j)& =F_l(i,j)\cdot X_r(i)+F_r(i,j)\cdot X_l(i)\\ & =U_l(i,j)V_l(i,j)U_r(i,j)i{2}^{|A_r|}{k}^{|L_r|}+U_r(i,j)V_r(i,j)U_l(i,j)i{2}^{|A_l|}{k}^{|L_l|}\\ U_u(i,j)V_u(i,j)& =U_l(i,j)U_r(i,j)i\left(V_l(i,j)2^{|A_r|}{k}^{|L_r|}+V_r(i,j)2^{|A_l|}{k}^{|L_l|}\right)\\ U_u(i,j)& =U_l(i,j)U_r(i,j)i\\ V_v(i,j)& =V_l(i,j)2^{|A_r|}{k}^{|L_r|}+V_r(i,j)2^{|A_l|}{k}^{|L_l|}\\ X_u(i)& =2X_l(i)\cdot X_r(i)\\ & =2\left(2^{|A_l|}{k}^{|L_l|}{U}_l(i)\cdot i\right)\left(2^{|A_r|}{k}^{|L_r|}{U}_r(i)\cdot i\right)\\ & =2^{|A_u|}{k}^{|L_u|}{U}_l(i)U_r(i)i^2\\ & =2^{|A_u|}{k}^{|L_u|}{U}_u(i)\cdot i\\ Y_u(i)& =Y_l(i)\cdot 2^{|A_r|+|B_r|}{k}^{2|L_r|}+Y_r(i)\cdot 2^{|A_l|+|B_l|}{k}^{2|L_l|}\\ & =\left(V_l(i)\cdot 2^{|A_r|}{k}^{|L_r|}+V_r(i)\cdot 2^{|A_l|}{k}^{|L_l|}\right)\cdot 2^{|B_l|+|B_r|}{k}^{|L_u|}\\ & =Y_l(i)\cdot 2^{|A_r|}{k}^{|L_r|}+Y_r(i)\cdot 2^{|A_l|}{k}^{|L_l|}\end{array}$$

故归纳成立，因此我们只需维护 $U,V$ 即可。注意到这一关系是两个孩子的乘法或线性组合，我们可以对表达式树通过分治在 $\Theta (n{\log }^{2}n)$ 内完成计算。然后藉由伯努利数即可在 $\Theta (n\log n)$ 内求出答案。