2021 牛客多校第一场 I.Increasing Subsequence

笑死比赛时读错题了.

题目

题意

给一个大小为n的排列P, 两个人在上面选数, 要求如下:

• 某个人选的数的下标要比之前这个人选的大
• 每次选的数值要比之前两个人选的数都大

每次选都是等概率从可以选的数中选一个, 第一个人随便选(选择每个数的概率都是$\frac{1}{n}$). 最后不能选了结束. 问结束时期望两人一共选多少个数.

$n\le 5000$

题解

!注意下标是比自己之前选的大, 不和对方比较.

补一下基础知识: 期望dp

期望dp一般这样设:

$dp(i)$ 为当前在 $i$, 还需要期望 $dp(i)$ 的代价到终点(不是从起点开始到dp[i]期望步数)

方程一般长这样:

$dp(i)=v_i+\sum p_{j}dp(j)$

其中 $v_i$ 是这一步的代价, $j$ 是后继状态, $p_j$ 是这个从状态 $i$ 走到状态 $j$ 的概率. 答案是起点的dp值, 比如 $dp(0)$.

在这个题中, 走一步对应的代价是1, 而等概率选取, 则某个 $i$ 在方程中对应的 $p_j$ 都是相等的, 且是 $\frac{1}{ou{t}_i}$, $ou{t}_i$ 表示 $i$ 的后继个数.

这个题是两个人在走, 所以状态要两维, 设 $dp(i,j)$ 表示第一个人在$i$, 第二个人在$j$, 期望还需要走多少步结束. 还需要考虑是哪一个人走, 由于是排列, 并且满足每次选的数比所有人上次的大, 所以比对手大, 这样直接判断 $a_i,a_j$ 的大小关系就可以知道是谁在走了.

方程分两部分转移

• $a_i>a_j$, 第二个人走: $dp(i,j)=1+\frac{1}{c_1}\sum dp(i,k_1)$, 其中, $k_1$ 满足 $k_1>j,a_{k_1}>a_i$, $c_1$ 是满足该条件的 $k$ 的个数
• $a_i<a_j$, 第一个人走: $dp(i,j)=1+\frac{1}{c_2}\sum dp(k_2,j)$, 其中, $k_2$ 满足 $k_2>i,a_{k_2}>a_j$, $c_2$ 是满足该条件的 $k$ 的个数

朴素转移复杂度 $O(n^3)$, 需要优化. 显而易见, $k_1$ 要满足比 $j$ 大, 且 $k_1,j$ 都是第二维, $k_2$ 和 $i$ 同理. 那么可以逆序枚举, 用前缀和(后缀和)的思想来加速转移, 同时可知, $1/c_1\sum dp(i,k_1)$ 和 $j$ 无关(关系已经通过转移顺序搞掉了).

设 $f(i)=\sum dp(i,k_1),c_1(i)$ 为第一个方程中的$c_1$. 对称地, 再设 $g(j)=\sum dp(k_2,j),c_2(j)$. 以第一个方程为例, 现在还需要考虑的是 $dp(i,k_1)$ 的另一个条件 $a_{k_1}>a_i$. 只有这样的状态, 才能够对 $f(i)$ 产生贡献. 然后会发现, 满足这个条件的状态, 不就是第二个方程吗? 所以做完第二个方程, 即 $a_i<a_j$ 的 $dp(i,j)$ 后, 把它加到 $f(i)$ 中, 同时 $c_1(i)+1$. 另一边同理.

这样就把复杂度降到了 $O(n^2)$.

答案需要枚举一下第一个人选的位置, 第二个人在0, 即 $dp(i,0)$, 第一个人等概率选, 所以最后是 $\frac{1}{n}{\sum }_{i=1}^{n}dp(i,0)$

代码