【校内模拟】Slay(容斥原理)_slaynoip模拟-程序员宅基地

技术标签: 校内模拟  容斥原理  

题解:

f ( p , k ) f(p,k) f(p,k)表示从 [ 0 , n − 1 ] [0,n-1] [0,n1]中选择 k k k个数,使得和为 p p p的倍数的方案数。

接下来用一种很诡异的姿势容斥。

直接算并不好算,考虑枚举最后有 i i i个数固定且相同,前面选择 k − i k-i ki个不同的数和为 s u m sum sum,这两部分分开计算,互不影响。

发现会出现 i + 1 i+1 i+1个数重复的情况,需要减掉,然后又多减掉了 i + 2 i+2 i+2个数重复的情况,就这样容斥下去。

考虑算固定 i i i个数的情况。

显然我们需要下面方程有解:

p ∣ i x + s u m p\mid ix+sum pix+sum

g c d ( i , p ) ∣ s u m gcd(i,p)\mid sum gcd(i,p)sum,设 s u m = t ⋅ g c d ( i , p ) sum=t\cdot gcd(i,p) sum=tgcd(i,p),则:

p ∣ i x + t ⋅ g c d ( i , p ) p g c d ( i , p ) ∣ i g c d ( i , p ) x + t p\mid ix+t\cdot gcd(i,p)\\\frac{p}{gcd(i,p)}\mid \frac{i}{gcd(i,p)}x+t pix+tgcd(i,p)gcd(i,p)pgcd(i,p)ix+t

显然由于 p ∣ n p\mid n pn x x x [ 0 , n − 1 ] [0,n-1] [0,n1]中的合法取值有 n p g c d ( i , p ) = n p ⋅ g c d ( i , p ) \frac{n}{\frac{p}{gcd(i,p)}}=\frac{n}p\cdot gcd(i,p) gcd(i,p)pn=pngcd(i,p),且与 t t t无关,也就是说子问题为 f ( g c d ( i , p ) , k − i ) f(gcd(i,p),k-i) f(gcd(i,p),ki)

容斥完了之后,需要考虑一个东西,在算没有任何数重复的合法方案的时候,我们实际上还是固定了一个数,则每一个方案被算了 k k k次,需要除掉。

那么,很容易地,我们得到容斥递归求解的式子:

f ( p , k ) = 1 k ∑ i = 1 k ( − 1 ) i + 1 ⋅ n ⋅ g c d ( i , p ) p ⋅ f ( g c d ( i , p ) , k − i ) f(p,k)=\frac{1}{k}\sum_{i=1}^{k}(-1)^{i+1}\cdot \frac{n\cdot gcd(i,p)}{p}\cdot f(gcd(i,p),k-i) f(p,k)=k1i=1k(1)i+1pngcd(i,p)f(gcd(i,p),ki)

总复杂度是一个调和级数 O ( k log ⁡ k ) O(k\log k) O(klogk)?常数较大。


代码:

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

using std::cerr;
using std::cout;

inline int gcd(int a,int b){
    
	if(!a||!b)return a|b;
	int shift=__builtin_ctz(a|b);
	for(b>>=__builtin_ctz(b);a;a-=b)if((a>>=__builtin_ctz(a))<b)std::swap(a,b);
	return b<<shift;
}

cs int mod=1e9+7;
inline int add(int a,int b){
    return (a+=b)>=mod?a-mod:a;}
inline int dec(int a,int b){
    return (a-=b)<0?a+mod:a;}
inline int mul(int a,int b){
    ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline int power(int a,int b,int res=1){
    
	for(;b;b>>=1,a=mul(a,a))(b&1)&&(res=mul(res,a));
	return res;
}
inline void Inc(int &a,int b){
    (a+=b)>=mod&&(a-=mod);}
inline void Dec(int &a,int b){
    (a-=b)<0&&(a+=mod);}

cs int K=5e3+3;

int n,k;
int Cn[K],inv[K];

inline void init(){
    
	inv[0]=inv[1]=1;
	Cn[0]=1,Cn[1]=n;
	for(int re i=2;i<=k;++i){
    
		inv[i]=mul(inv[mod%i],mod-mod/i);
		Cn[i]=mul(Cn[i-1],mul(dec(n+1,i),inv[i]));
	}
}

int dp[K][K];

inline int f(int p,int k){
    
	if(!k)return 1;
	if(p==1)return Cn[k];
	if(~dp[p][k])return dp[p][k];
	int ans=0,tmp,val,d=n/p;
	for(int re i=1;i<=k;++i){
    
		tmp=gcd(p,i);
		val=mul(f(tmp,k-i),tmp);
		(i&1)?Inc(ans,val):Dec(ans,val);
	}
	return dp[p][k]=mul(ans,mul(d,inv[k]));
}

signed main(){
    
#ifdef zxyoi
	freopen("slay.in","r",stdin);
#endif
	std::cin>>n>>k;
	init();
	memset(dp,-1,sizeof dp);
	int ans=0,tmp,val;
	for(int re i=1;i<=k;++i){
    
		tmp=gcd(n,i);
		val=mul(f(tmp,k-i),tmp);
		(i&1)?Inc(ans,val):Dec(ans,val);
	}
	cout<<mul(ans,inv[k])<<"\n";
	return 0;
}
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/zxyoi_dreamer/article/details/97819504

智能推荐

oracle 12c 集群安装后的检查_12c查看crs状态-程序员宅基地

文章浏览阅读1.6k次。安装配置gi、安装数据库软件、dbca建库见下:http://blog.csdn.net/kadwf123/article/details/784299611、检查集群节点及状态:[root@rac2 ~]# olsnodes -srac1 Activerac2 Activerac3 Activerac4 Active[root@rac2 ~]_12c查看crs状态

解决jupyter notebook无法找到虚拟环境的问题_jupyter没有pytorch环境-程序员宅基地

文章浏览阅读1.3w次,点赞45次,收藏99次。我个人用的是anaconda3的一个python集成环境,自带jupyter notebook,但在我打开jupyter notebook界面后,却找不到对应的虚拟环境,原来是jupyter notebook只是通用于下载anaconda时自带的环境,其他环境要想使用必须手动下载一些库:1.首先进入到自己创建的虚拟环境(pytorch是虚拟环境的名字)activate pytorch2.在该环境下下载这个库conda install ipykernelconda install nb__jupyter没有pytorch环境

国内安装scoop的保姆教程_scoop-cn-程序员宅基地

文章浏览阅读5.2k次,点赞19次,收藏28次。选择scoop纯属意外,也是无奈,因为电脑用户被锁了管理员权限,所有exe安装程序都无法安装,只可以用绿色软件,最后被我发现scoop,省去了到处下载XXX绿色版的烦恼,当然scoop里需要管理员权限的软件也跟我无缘了(譬如everything)。推荐添加dorado这个bucket镜像,里面很多中文软件,但是部分国外的软件下载地址在github,可能无法下载。以上两个是官方bucket的国内镜像,所有软件建议优先从这里下载。上面可以看到很多bucket以及软件数。如果官网登陆不了可以试一下以下方式。_scoop-cn

Element ui colorpicker在Vue中的使用_vue el-color-picker-程序员宅基地

文章浏览阅读4.5k次,点赞2次,收藏3次。首先要有一个color-picker组件 <el-color-picker v-model="headcolor"></el-color-picker>在data里面data() { return {headcolor: ’ #278add ’ //这里可以选择一个默认的颜色} }然后在你想要改变颜色的地方用v-bind绑定就好了,例如:这里的:sty..._vue el-color-picker

迅为iTOP-4412精英版之烧写内核移植后的镜像_exynos 4412 刷机-程序员宅基地

文章浏览阅读640次。基于芯片日益增长的问题,所以内核开发者们引入了新的方法,就是在内核中只保留函数,而数据则不包含,由用户(应用程序员)自己把数据按照规定的格式编写,并放在约定的地方,为了不占用过多的内存,还要求数据以根精简的方式编写。boot启动时,传参给内核,告诉内核设备树文件和kernel的位置,内核启动时根据地址去找到设备树文件,再利用专用的编译器去反编译dtb文件,将dtb还原成数据结构,以供驱动的函数去调用。firmware是三星的一个固件的设备信息,因为找不到固件,所以内核启动不成功。_exynos 4412 刷机

Linux系统配置jdk_linux配置jdk-程序员宅基地

文章浏览阅读2w次,点赞24次,收藏42次。Linux系统配置jdkLinux学习教程,Linux入门教程(超详细)_linux配置jdk

随便推点

matlab(4):特殊符号的输入_matlab微米怎么输入-程序员宅基地

文章浏览阅读3.3k次,点赞5次,收藏19次。xlabel('\delta');ylabel('AUC');具体符号的对照表参照下图:_matlab微米怎么输入

C语言程序设计-文件(打开与关闭、顺序、二进制读写)-程序员宅基地

文章浏览阅读119次。顺序读写指的是按照文件中数据的顺序进行读取或写入。对于文本文件,可以使用fgets、fputs、fscanf、fprintf等函数进行顺序读写。在C语言中,对文件的操作通常涉及文件的打开、读写以及关闭。文件的打开使用fopen函数,而关闭则使用fclose函数。在C语言中,可以使用fread和fwrite函数进行二进制读写。‍ Biaoge 于2024-03-09 23:51发布 阅读量:7 ️文章类型:【 C语言程序设计 】在C语言中,用于打开文件的函数是____,用于关闭文件的函数是____。

Touchdesigner自学笔记之三_touchdesigner怎么让一个模型跟着鼠标移动-程序员宅基地

文章浏览阅读3.4k次,点赞2次,收藏13次。跟随鼠标移动的粒子以grid(SOP)为partical(SOP)的资源模板,调整后连接【Geo组合+point spirit(MAT)】,在连接【feedback组合】适当调整。影响粒子动态的节点【metaball(SOP)+force(SOP)】添加mouse in(CHOP)鼠标位置到metaball的坐标,实现鼠标影响。..._touchdesigner怎么让一个模型跟着鼠标移动

【附源码】基于java的校园停车场管理系统的设计与实现61m0e9计算机毕设SSM_基于java技术的停车场管理系统实现与设计-程序员宅基地

文章浏览阅读178次。项目运行环境配置:Jdk1.8 + Tomcat7.0 + Mysql + HBuilderX(Webstorm也行)+ Eclispe(IntelliJ IDEA,Eclispe,MyEclispe,Sts都支持)。项目技术:Springboot + mybatis + Maven +mysql5.7或8.0+html+css+js等等组成,B/S模式 + Maven管理等等。环境需要1.运行环境:最好是java jdk 1.8,我们在这个平台上运行的。其他版本理论上也可以。_基于java技术的停车场管理系统实现与设计

Android系统播放器MediaPlayer源码分析_android多媒体播放源码分析 时序图-程序员宅基地

文章浏览阅读3.5k次。前言对于MediaPlayer播放器的源码分析内容相对来说比较多,会从Java-&amp;amp;gt;Jni-&amp;amp;gt;C/C++慢慢分析,后面会慢慢更新。另外,博客只作为自己学习记录的一种方式,对于其他的不过多的评论。MediaPlayerDemopublic class MainActivity extends AppCompatActivity implements SurfaceHolder.Cal..._android多媒体播放源码分析 时序图

java 数据结构与算法 ——快速排序法-程序员宅基地

文章浏览阅读2.4k次,点赞41次,收藏13次。java 数据结构与算法 ——快速排序法_快速排序法