【题解】P4707 重返现世

【题解】P4707 重返现世

P4707 重返现世

期望下Min-Max容斥DP

$$E(kthmax(T))=\sum _{T\subset S}(-1{)}^{|T|-k}{C}_{|T|-1}^{k-1}E(\min (S))$$
我们可以把物品出现的时间记为它的值

则答案为期望下全部物品第k大值

对于子集T，第一个出现的时间即为最小值

每次出现其中任意一个的概率是 $\frac{{\sum }_{i\in T}{p}_i}{m}$

最小值期望为 $\frac{m}{{\sum }_{i\in T}{p}_i}$

答案为
$$ans=\sum _{T\subset S}(-1{)}^{|T|-k}{C}_{|T|-1}^{k-1}\frac{1}{p_T}$$
但是枚举每个子集复杂度达到了指数级别

我们考虑DP

设$dp[i][j]$ 为在前i个物品中出现k为j的${\sum }_{T\subset S}(-1{)}^{|T|-k}{C}_{|T|-1}^{k-1}$

DP三要素：状态、转移、边界

转移分两种情况： 1.选了当前点 2.没选当前点

没选当前点则不产生贡献

枚举当前点的p，考虑从$dp[i-p][]$转移过来

$$dp[i][j]=\sum _{T\subset S}(-1{)}^{|T|+1-j}{C}_{|T|+1-\mathrm{１}}^{ｊ-1}$$
$$dp[i-p][j]=\sum _{T\subset S}(-1{)}^{|T|-j}{C}_{|T|-1}^{ｊ-1}$$

我们借助
$$C_{|T|+1-1}^{ｊ-1}=C_{|T|-1}^{ｊ-1}+C_{|T|-1}^{j-2}$$
有
$$\sum _{T\subset S}(-1{)}^{|T|+1-j}{C}_{|T|+1-\mathrm{１}}^{ｊ-1}=(-1)\sum _{T\subset S}(-1{)}^{|T|-j}{C}_{|T|-\mathrm{１}}^{ｊ-1}+\sum _{T\subset S}(-1{)}^{|T|-(j-1)}{C}_{|T|－1}^{j-2}$$

即就是
$$dp[i][j]=－dp[i-p][j]+dp[i-p][j-1]$$
时间复杂度$O(nmf)$

设初始状态时$dp[0][j]=(-1{)}^{-j}{C}_{-1}^{j-1}=(-1{)}^{-j}(-1{)}^{j-1}{C}_{j-1}^{j-1}=-1$

#include<bits/stdc++.h>
using namespace std;
inline int Read(){
 	int s = 0 , w = 1;
   char ch = getchar();
   while(ch > '9' || ch < '0'){
   	if(ch == '-') w = -1;
   	ch = getchar();
   }
   while(ch >= '0' && ch <= '9'){
   	s = (s << 1) + (s << 3) + ch - '0';
   	ch = getchar();
   }
   return s * w;
}
#define int long long
const int MAXN = 1e3 + 5 , MAXM = 1e4 + 5 , mod = 998244353;
int add(int x,int y){
   x += y;
   return x >= mod ? x - mod : x;
}
inline int sub(int x,int y){
   x -= y;
   return x < 0 ? x + mod : x;
}
int n,m,s,w,ans;
int inv[MAXM],f[12][MAXM];
#undef int 
int main(){
   #define int long long
   n = Read() , s = n + 1 - Read() ;
   m = Read();
   for(int i = (inv[1] = 1) + 1 ; i <= m ; i ++){
   	inv[i] = inv[mod % i] * (mod - mod / i) % mod;
   }
   for(int i = 1 ; i <= s ; i ++) f[i][0] = -1;
   for(int i = 1 ; i <= n ; i ++){
   	w = Read();
   	for(int j = m ; j >= w ; j --){
   		for(int k = s ; k ; k--){
   			f[k][j] = add(f[k][j],sub(f[k - 1][j - w] , f[k][j - w]));
   		}
   	}
   }
   for(int i = 1 ; i <= m ;  i++){
   	ans = (ans + f[s][i] * inv[i] % mod) % mod;
   }
   cout << ans * m % mod;
   return 0;
}