2023牛客OI赛前集训营-提高组(第二场) 集合

文章讲述了如何使用动态规划解决一个计算给定整数范围内非空子集和的乘积对998244353取模的问题,利用扩展欧拉定理优化求解过程,时间复杂度为O(n^3)。

题目大意

给定正整数nnn,求有nnn个元素的集合{1,2,…,n}\{1,2,\dots,n\}{1,2,,n}的所有非空子集和的乘积模998244353998244353998244353后的结果。

1≤n≤2001\leq n\leq 2001n200


题解

依题意,非空子集和的取值范围为[1,n×(n+1)2][1,\dfrac{n\times (n+1)}{2}][1,2n×(n+1)]

考虑DPDPDP。设fi,jf_{i,j}fi,j表示有iii个元素的集合{1,2,…,i}\{1,2,\dots,i\}{1,2,,i}中非空子集和为jjj的子集数量。那么,可以用O(n3)O(n^3)O(n3)的时间复杂度求出fi,jf_{i,j}fi,j

由扩展欧拉定理,可得在求fi,jf_{i,j}fi,j时模数为ϕ(998244353)\phi(998244353)ϕ(998244353),也就是998244352998244352998244352

那么,答案为

ans=∑i=1n×(n+1)/2ifn,ians=\sum\limits_{i=1}^{n\times (n+1)/2}i^{f_{n,i}}ans=i=1n×(n+1)/2ifn,i

这个式子的模数为998244353998244353998244353

于是,问题就解决了。

时间复杂度为O(n3)O(n^3)O(n3)

code

#include<bits/stdc++.h>
using namespace std;
const long long mod=998244353;
int n,l;
long long ans=1,f[205][40005];
long long mi(long long t,long long v){
	if(!v) return 1;
	long long re=mi(t,v/2);
	re=re*re%mod;
	if(v&1) re=re*t%mod;
	return re;
}
int main()
{
	scanf("%d",&n);
	l=n*(n+1)/2;
	f[0][0]=1;
	for(int i=1;i<=n;i++){
		for(int j=0;j<=l;j++){
			f[i][j]=f[i-1][j];
			if(j>=i) f[i][j]=(f[i][j]+f[i-1][j-i])%(mod-1);
		}
	}
	for(int i=1;i<=l;i++){
		ans=ans*mi(i,f[n][i])%mod;
	}
	printf("%lld",ans);
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值