能力强化!!!--NOI爆零赛

数据范围 n,m,k<=100

首先我们得知道有一个东西叫MIN-MAX容斥

然后我们可以知道只有一个能力时

f[i]表示对于一个能力强化i次后,不强化到满级的方案数

那么dp[i][j]表示对于i个能力,总共强化j次,没有任何一个能力强化到满级的方案数

那么dp[i][j]每次用f来转移一下,相当于做个背包,乘上组合数就是新的方案数

然后我们把dp[i][0]-dp[i][i*(k-1)]全部加起来,没有一个强化到满级的方案数,就等于强化次数大于i次的方案数

所以我们累加起来就是只有一个满级的方案数,再除以总的方案数,就是当前子集期望了

我们要求的是全部满级的期望步数,那么就是MIN-MAX容斥一下就可以了

复杂度大概是O(n^4)

#include <cstdio>
#include <algorithm>
using namespace std ; 
#define LL long long  

int mod = 998244353 ; 
int fp ( int x , int k ) 
{
	int yl = 1 ; 
	while ( k ) 
	{
		if ( k % 2 ) yl = 1ll * yl * x % mod ; 
		x = 1ll * x * x % mod ; 
		k = k / 2 ; 
	}
	return yl ; 
} 

int a[105] , f[105][205] , N , M , K ; 
int p[105] , g[10005] ;  
int MD ( int x ) { return x >= mod ? x - mod : x ; }

int jc[100005] , vjc[100005] ;  
void init ( ) 
{
	jc[0] = vjc[0] = 1 ; 
	for ( int i = 1 ; i <= 100000 ; i++ ) jc[i] = 1ll * jc[i-1] * i % mod ; 
	vjc[100000] = fp ( jc[100000] , mod - 2 ) ;
	for ( int i = 100000 ; i >= 2 ; i-- ) vjc[i-1] = 1ll * vjc[i] * i % mod ;  
}

int group ( int big , int small ) { return 1ll * jc[big] * vjc[big - small] % mod * vjc[small] % mod ; } 
int ans = 0 ; 
 
int main ( ) 
{
	scanf ( "%d%d%d" , &N , &M , &K ) ; 
	for ( int i = 1 ; i <= N ; i++ ) scanf ( "%d" , &a[i] ) ;
	f[0][0] = 1 ; 
	for ( int i = 0 ; i < K ; i++ )
	    for ( int k = 0 ; k < K ; k++ ) 
	    	for ( int j = 1 ; j <= N ; j++ ) 
	        	f[i+1][k+a[j]] = MD( f[i+1][k+a[j]] + f[i][k] )  ; 
	for ( int i = 0 ; i < K ; i++ ) 
	    for ( int k = 0 ; k < K ; k++ ) 
	        p[i] = MD ( p[i] + f[i][k] ) ; 
	init ( ) ;  
	int pl = -1 ;
	for ( int i = 0 ; i < K ; i++ ) g[i] = p[i] ;  
	for ( int i = 1 ; i <= M ; i++ ) 
	{
		pl *= -1 ; int yl = 0 ;
		int temp=fp(i*N,mod-2);
		int another=1;
		int maxa=i*(K-1);
		for ( int s = 0 ; s <= maxa ; s++ )
		    yl =MD ( yl + 1ll * g[s] * another % mod) ,another=1ll*another*temp%mod; 
		ans = MD ( ( ans + 1ll * yl * group ( M , i ) % mod * M % mod * fp ( i , mod - 2 ) * pl ) % mod + mod ) ;
		for ( int s = ( i + 1 ) * ( K - 1 ) ; s >= 0 ; s-- )
		{
	    	yl = 0 ; 
			for ( int j = 0 ; j <= s && j < K ; j++ ) 
			    yl = ( yl + 1ll * g[s - j] * p[j] % mod * group ( s , j ) ) % mod ; 
			g[s] = yl ; 
		} 
	}
	printf ( "%d\n" , ans ) ; 
}
/*
1 2 1 
1

1 3 2
1

2 2 3 
1 2 

3 5 5 
1 2 4 
*/ 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值