P16639 [GKS 2018 #A] Lucky Dip
题目描述
你正在参加“盛大 Kickstart 幸运抽奖”活动,奖品非常精彩且令人惊叹(也有些不太好的奖品)!
在这个幸运抽奖中,有一个装有 $N$ 个物品的袋子。袋子中的第 $i$ 个物品的价值为 $V_i$。你将把手伸进袋子,随机抽取一个物品;袋子中的所有物品被抽中的概率相等。组织者希望让参赛者感到自己有一定的选择权,因此在你抽出一个物品后,你可以选择保留它,或者将其放回袋子并重新抽取(即“重抽”)。注意,放回的物品与袋子中其他物品被抽中的概率相同。你最多只能重抽 $K$ 次。如果你使用了 $K$ 次重抽,则必须保留第 $(K+1)$ 次抽到的物品。
如果你以最优策略进行游戏,以最大化最终持有的物品价值,那么该物品的期望价值是多少?
输入格式
输入的第一行包含一个整数 $T$,表示测试用例的数量。接下来有 $T$ 个测试用例。
每个测试用例由两行组成。第一行包含两个整数 $N$ 和 $K$:袋子中的物品数量,以及最多可以重抽的次数。第二行包含 $N$ 个整数 $V_i$,表示第 $i$ 个物品的价值。
输出格式
对于每个测试用例,输出一行,格式为 `Case #x: y`,其中 $x$ 是测试用例编号(从 $1$ 开始),$y$ 是上述期望价值。如果答案与正确答案的绝对误差或相对误差在 $10^{-6}$ 以内,则认为正确。
说明/提示
在样例 #1 中,你不能重抽,所以期望价值就是袋子中物品的平均值,即 $(1 + 2 + 3 + 4) / 4 = 2.5$。
在样例 #2 中,最佳策略是:如果抽到价值为 $10$ 的物品就保留,否则重抽。第一次或第二次抽到该物品的概率为 $1 - (2/3)^2 = 5/9$,因此期望价值为 $(5/9 \times 10) + (4/9 \times 1) = 6$。
在样例 #3 中,由于所有物品价值相同,重抽多少次都没有影响,因此期望价值为 $80000$。
注意,样例 #3 和 #5 不会出现在小数据集中。
### 限制条件
$1 \le T \le 100$。
$1 \le V_i \le 10^9$。
$1 \le N \le 2 \times 10^4$。
**小数据集(测试集 1 – 可见)**
$0 \le K \le 1$。
**大数据集(测试集 2 – 隐藏)**
$0 \le K \le 5 \times 10^4$。
翻译由 DeepSeek V4 Pro 完成