P9838 挑战 NPC IV ( luogu NOIP模拟赛T3 )

Tonvia

已于 2023-11-22 16:32:16 修改

阅读量126

点赞数 1

分类专栏： dp 考试总结题解文章标签：算法 1024程序员节 c++ 动态规划贪心算法排序算法科技

于 2023-11-14 11:08:57 首次发布

本文链接：https://blog.csdn.net/Tonvia/article/details/134391540

版权

考试总结同时被 3 个专栏收录

27 篇文章 1 订阅

订阅专栏

题解

21 篇文章 0 订阅

订阅专栏

13 篇文章 0 订阅

订阅专栏

绝世好题。逐步推导部分分逼近正解。

题解稍长，请耐心读完。

题目大意：给定序列 $a=\lbrace1,2,\dots n\rbrace$ ，求 $n!$ 种排列中 $\sum_{i=1}^n\sum_{j=i}^n\sum_{k=i}^{j}1+\log_2{\operatorname{lowbit}(a_i)}$ 的第 $k$ 小值。

照搬具体范围如下：

【数据范围】

本题各测试点时间限制不相同。具体地，每个点的时间限制为 $\max(q\times 0.5, 2)\ \rm{s}$ 。

测试点编号	$n$	$\leq$	$q = $
$\sim 3$	$\leq 10$	$n!$	$2$
$\sim 8$	$\leq 10^3$	$2$	$7$
$\sim 13$	$\in [10^5, 10^6]$	$min(10^{18}, n!)$	$7$
$\sim 17$	$\leq 10^6$	$min(10^{18}, n!)$	$7$
$\sim 25$	$\leq 10^{18}$	$min(10^{18}, n!)$	$10$

题解：

显然地，式子可以化简成 $\sum_{i=1}^ni\times (n-i+1)(1+\log_2{\operatorname{lowbit}(a_i)})$ 。
为了描述方便，将 $a_i$ 赋值为 $(1+\log_2{\operatorname{lowbit}(a_i)}$ ，原式化为 $\sum_{i=1}^ni\times (n-i+1)\times a_i$ 。

对于 $1\sim 3$ 测试点，直接暴力求出 $n!$ 种取值即可，不再赘述。
对于 $4\sim 8$ 测试点，发现 $k$ 极小，

考虑 $k = 1$ ，由小奥得，设序列 $b$ ， $b_i=i\times (n-i+1)$ ，答案即为 $a$ 中最小的 $a_i$ 乘上 $b$ 中最大的 $b_i$ 加上 $a$ 中次小的 $a_i$ 乘上 $b$ 中次大的 $b_i$ ，以此类推求和。
考虑 $k = 2$ ，设 $k = 1$ 时 $a$ 的答案序列为 $A$ ，发现由于 $b_1=b_n$ ，只需调换 $A_1,A_n$ 位置即可，答案大小不变，仍为最小值。

对于 $9\sim 13$ 测试点，受 $4\sim 8$ 测试点启发，可以发现 $A n s$ 为最小值的答案实际上有很多，
设序列 $A, B$ 分别为 $a, b$ 排序后的序列，容易发现 $A_i$ 的取值极小，考虑计算答案同为最小值的方案数，即计算与序列 $A$ 同构的序列总数，易得答案为 $A_i$ 中每个值个数的阶乘的乘积，举个例子，设 $A=\lbrace1,1,1,1,1,2,2,3,4\rbrace$ ，同构数即 $5!\times 2!\times 1!\times 1!=240$ 种。
可以发现当 $n\ge 39$ 时， $A$ 中光是 $1$ 的个数就达到了 $20$ ，同构数超过了 $10^{18}$ ，所以对于该段测试点，答案与 $k = 1$ 时相等， $O (n)$ 计算即可。

对于 $14\sim 17$ 测试点，手算一下发现同构数小于等于 $10^{18}$ 的 $n$ 上界为 $28$ ，因此我们只需计算 $n\le28$ 时第 $k$ 小值的大小，此时 $A$ 的取值只有 $1, 2, 3, 4, 5$ 五种（ $A$ 与上文含义相同），考虑 dp 求方案数，设 $f [a] [b] [c] [d] [e] [i]$ 表示 $A$ 中前 $a + b + c + d + e$ 位中 $1, 2, 3, 4, 5$ 的个数分别是 $a, b, c, d, e$ 且此时总价值为 $i$ 的方案数 （不考虑同构情况），可得转移方程为

$\begin{aligned} f[a][b][c][d][e][i] &=f[a-1][b][c][d][e][i-sum\times (n-sum+1)]\\ &+f[a][b-1][c][d][e][i-sum\times (n-sum+1)\times 2]\\ &+f[a][b][c-1][d][e][i-sum\times (n-sum+1)\times 3]\\ &+f[a][b][c][d-1][e][i-sum\times (n-sum+1)\times 4]\\ &+f[a][b][c][d][e-1][i-sum\times (n-sum+1)\times 5] \end{aligned}$ ，其中 $s u m = a + b + c + d + e$ 。

注意到 $f [0] [0] [0] [0] [0] [0] = 1$ 。

想到状态的设置，式子还是比较好推的，不建议细看上面的式子。

求解的时候从 $i = 0$ 向上累加 $f[\text{a 中 1 的个数}][\text{a 中 2 的个数}][\text{a 中 3 的个数}][\text{a 中 4 的个数}][\text{a 中 5 的个数}][i]$ ，记得考虑乘上相同 $a_j$ 的排列情况，累加的和超过 $k$ 时立刻得到答案为 $i$ 。