Description

为了打开返回现世的大门,\(Yopilla\) 需要制作开启大门的钥匙。\(Yopilla\) 所在的迷失大陆有 \(n\) 种原料,只需要集齐任意 \(k\) 种,就可以开始制作。
\(Yopilla\) 来到了迷失大陆的核心地域。每个单位时间,这片地域就会随机生成一种原料。每种原料被生成的概率是不同的,第 \(i\) 种原料被生成的概率是 \(\frac{p_i}{m}\) 。如果 \(Yopilla\) 没有这种原料,那么就可以进行收集。

\(Yopilla\) 急于知道,他收集到任意 \(k\) 种原料的期望时间,答案对 998244353 取模。

Input

第一行三个数 \(n,k,m\) 。

第二行 \(n\) 个数 \(p_1, p_2, ..., p_n\) 。

Output

输出一行。

Sample Input

3 3 3
1 1 1

Sample Output

499122182

HINT

对于 \(10 \%\) 的数据,\(p_1 = p_2 = ... = p_m\)

对于另外 \(10 \%\) 的数据,\(k = n\) 。

对于 \(70 \%\) 的数据,\(n \leq 100\)。

对于 \(100 \%\) 的数据,\(1 \leq n \leq 1000\) ,\(1 \leq k \leq n,\) \(| n - k | \leq 10\), \(0 \leq p_i \leq m\) , $ \sum p = m$, \(1 \leq m \leq 10000\) 。


想法

首先,看到“收集”“期望”这种字眼,就套路地想到 \(MIN-MAX\) 容斥
而这道题相当于求第 \(K\) 小的期望收集到的时间,那就是扩展 \(MIN-MAX\) 容斥

鉴于 \(|n-k|\leq 10\) ,不妨令 \(K=n-K+1\) ,改求第 \(K\) 大。
上式子, \(K-MAX(S)=\sum\limits_{T \subseteq S} (^{|T|-1}_{K-1}) \cdot (-1)^{|T|-K} \cdot \frac{m}{\sum\limits_{i \in T} p_i}\)
(证明挺容易的,就是二项式反演,略过……)

发现 \(n\) 好大不能枚举子集,于是考虑 \(dp\) 。
接下来的做法就十分神仙了【划重点】
我们发现 \(m\) 很小,也就是上面式子中 \(\frac{m}{\sum\limits_{i \in T} p_i}\) 的值只有 \(10^5\)种
那考虑合并同类项,把它拎出来,原式写成:
\[ ans= \sum\limits_{j=1}^m \frac{m}{j} \cdot f_{n,j,k} \]

于是我们就用 \(dp\) 搞这个系数。
\(f_{i,j,k}\) 中: \(i\) 表示考虑了前 \(i\) 个数; \(j\) 表示在前 \(i\) 个数中选出了集合 \(T\) ,\(T\) 中所有元素和为 \(j\) ;\(k\) 即求第 \(k\) 大。

转移:
1.对于 \(T\) 中没有第 \(i\) 个数的:贡献是 \(f_{i-1,j,k}\)
2.对于 \(T\) 中有第 \(i\) 个数的:贡献是 \(\sum\limits_{i \in T} (^{|T|-1}_{k-1}) \cdot (-1)^{|T|-k}\),这式中的 \(T\) 满足元素和为 \(j\)
而这个贡献肯定要由 \(f_{i-1,j-p[i],x}\) 转移而来,\(x\) 未知
于是我们把它改写成
\[ \begin{equation*} \begin{aligned} &\sum\limits_{T} (^{|T|}_{k-1}) \cdot (-1)^{|T|-k+1} 其中T即为上式T减去i,满足元素和为j-p[i]\\ =&\sum [(^{|T|-1}_{k-1})+(^{|T|-1}_{k-2})] \cdot (-1)^{|T|-k+1} \\ =&\sum (^{|T|-1}_{k-1}) \cdot (-1)^{|T|-k} \cdot (-1) + \sum (^{|T|-1}_{(k-1)-1}) \cdot (-1)^{|T|-k+1} \\ =&-f_{i-1,j-p[i],k}+f_{i-1,j-p[i],k-1} \end{aligned} \end{equation*} \]

综上所述,\(f_{i,j,k}=f_{i-1,j,k}+f_{i-1,j-p[i],k-1}-f_{i-1,j-p[i],k}\)
边界条件 \(f_{0,0,0}=1\)
注意要用滚动数组。

把所有系数求出来后枚举 \(\sum p[i]\)的值 ,带进去求就行了。


代码

#include<cstdio>
#include<iostream>
#include<algorithm>#define P 998244353using namespace std;int read(){int x=0;char ch=getchar();while(!isdigit(ch)) ch=getchar();while(isdigit(ch)) x=x*10+ch-'0',ch=getchar();return x;
}const int N = 1005;int n,K,m,p[N];
int f[2][N*10][12];int Pow_mod(int x,int y){int ret=1;while(y){if(y&1) ret=1ll*ret*x%P;x=1ll*x*x%P;y>>=1;}return ret;
}int main()
{n=read(); K=read(); m=read();K=n-K+1;for(int i=1;i<=n;i++) p[i]=read();f[0][0][0]=1;for(int i=1;i<=n;i++){int pre=((i-1)&1),id=(i&1);for(int j=0;j<=m;j++)for(int k=0;k<=i && k<=K;k++){f[id][j][k]=f[pre][j][k];if(j>=p[i] && k) f[id][j][k]=((f[id][j][k]+f[pre][j-p[i]][k-1])%P+P-f[pre][j-p[i]][k])%P; //手残写成 p[j] ,调了好久}}int ans=0;for(int i=1;i<=m;i++)ans=((ans+1ll*m*Pow_mod(i,P-2)%P*f[n&1][i][K]%P)%P+P)%P;printf("%d\n",ans);return 0;
}

转载于:https://www.cnblogs.com/lindalee/p/11348830.html

[洛谷P4707] 重返现世相关推荐

  1. 洛谷 P4707 重返现世

    洛谷 P4707 重返现世 k-minimax容斥 有这一个式子:\(E(\max_k(S))=\sum_{T\subseteq S}(-1)^{|T|-k}C_{|T|-1}^{k-1}\min(T ...

  2. [洛谷P4707]重返现世

    Description 有n种物品,每次操作你有pimpi\over mmpi​的概率获得第i种物品 求你获得k种物品的期望操作数. n<=1000,n-k<=10,m<=10000 ...

  3. 洛谷P4707 重返现世(扩展MinMax容斥+dp)

    传送门 我永远讨厌\(dp.jpg\) 前置姿势 扩展\(Min-Max\)容斥 题解 看纳尔博客去→_→ 咱现在还没搞懂为啥初值要设为\(-1\)-- //minamoto #include< ...

  4. 【题解】P4707 重返现世

    [题解]P4707 重返现世 P4707 重返现世 期望下Min-Max容斥DP E(kthmax(T))=∑T⊂S(−1)∣T∣−kC∣T∣−1k−1E(min⁡(S))E(kthmax(T)) = ...

  5. [luogu P4707] 重返现世

    Description 洛谷题目链接:https://www.luogu.org/problemnew/show/P4707 Solution 前置广义\(\min-\max\)容斥,不懂的可以看看我 ...

  6. 【洛谷P4707】重返现世【扩展Min-Max容斥】【dp】

    传送门 题意:NNN种物品,每次第iii种产生概率为piM\frac{p_i}{M}Mpi​​,∑pi=M\sum p_i=M∑pi​=M.求生成KKK种不同物品的期望时间 模998244353998 ...

  7. 洛谷 P4707 【重返现世】

    题目分析 题目就是求第K种原料的出现期望时间. 考虑广义min-max容斥. \(\text{kthmax}(S)=\sum\limits_{T\subseteq S}(-1)^{|T|-k}\bin ...

  8. 【洛谷P4707】重返现世(Min-Max容斥)(背包)

    传送门 题解: 首先令k=n−k+1k=n-k+1k=n−k+1,转化为求第kkk大. 直接利用Kth−MinMaxKth-MinMaxKth−MinMax容斥的式子我们知道要求这个东西: E(kth ...

  9. Luogu P4707 重返现世 (拓展Min-Max容斥、DP)

    题目链接 https://www.luogu.org/problem/P4707 题解 最近被神仙题八连爆了-- 首先Min-Max容斥肯定都能想到,问题是这题要用一个扩展版的--Kth Min-Ma ...

最新文章

  1. Selenium IDE工具界面剖析
  2. 使用iBATIS3.0完成增删改查
  3. hibernate get方法有执行sql但是后台拿不到_「6」进大厂必须掌握的面试题-Hibernate...
  4. 初学者python笔记(文件的操作)
  5. 机器学习模型的理解(三)
  6. cmd 修改ie快捷方式_windows使用技巧之Win + R 与 CMD 的不同
  7. [算法]机器人运动范围
  8. SQL Server 2008 R2 学习总结
  9. Ubuntu升级Ruby
  10. 微信抢票应用开发总结
  11. Python用python-docx抓取公众号文章写入word
  12. 领导艺术:如何留住员工的心
  13. 图片不超过200kb怎么调整?一分钟学会图片压缩到指定大小
  14. STM32的USART串口通信
  15. Hadoop第三天--linux指令
  16. 【信息系统项目管理师】信息系统与信息化
  17. 【爱生活】人生箴言 一
  18. 今天生日,还要在家远程上班。你们公司什么情况?
  19. 计算机开机测试,电脑开机检测怎么开启
  20. 全世界最像人的「机器人」,本身就是人

热门文章

  1. Unity中的截图方法(包括全屏截图、区域截图、Camera截图和摄像头截图)
  2. php析构函数有什么用,php析构函数的作用
  3. Java进阶(五十五)-Java Lambda表达式入门
  4. Google Play下载超千万!这款AI人脸风格、换脸APP
  5. 图片预览-放大镜效果
  6. 海康威视研究院算法岗实习
  7. 403 Forbidden You don‘t have permission to access this resource. Apache Server at IP Port 80的解决方法
  8. AI学者推动人工智能技术在现实生活中的大规模应用
  9. (转)DVD转换RMVB格式的方法
  10. 【Excel】三秒钟将表格内数据累加