bzoj 3771: Triple 快速傅里叶变换+容斥原理
来源:互联网 发布:高中物理模拟软件 编辑:程序博客网 时间:2024/06/14 11:08
题意
给定n个物品,可以用一个/两个/三个不同的物品凑出不同的价值,求每种价值有多少种拼凑方案(顺序不同算一种)
分析
显然搞一波母函数然后直接上fft即可。
问题是怎么处理重复的。
设x为只选一个,y为选两个相同的,z为选三个相同的,那么答案就是x+(x*x-y)/2+(x*x*x-3*x*y+z*2)/6
解释一下,x就是只选一个,(x*x-y)/2表示选两个减去重复的再除以排列,(x*x*x-3*x*y+z*2)/6表示选三个然后减去两个相同的乘上其排列再加上两倍三个相同的。看不懂的话就自己画一个图看看吧。
代码
#include<iostream>#include<cstdio>#include<cstdlib>#include<cstring>#include<algorithm>#include<complex>#include<cmath>#define N 500005#define pi acos(-1)using namespace std;typedef complex<double> com;int n,m,rev[N],lg;com a[N],b[N],c[N],ans[N];void fft(com *a,int f,int n){ for (int i=0;i<n;i++) if (i<rev[i]) swap(a[i],a[rev[i]]); for (int i=1;i<n;i<<=1) { com wn(cos(pi/i),f*sin(pi/i)); for (int j=0;j<n;j+=(i<<1)) { com w(1,0); for (int k=0;k<i;k++) { com u=a[j+k],v=w*a[j+k+i]; a[j+k]=u+v;a[j+k+i]=u-v; w*=wn; } } } if (f==-1) for (int i=0;i<n;i++) a[i]/=n;}int main(){ scanf("%d",&n); int mx=0; for (int i=1;i<=n;i++) { int x; scanf("%d",&x); a[x]=b[x*2]=c[x*3]=1; mx=max(mx,x*3); } for (m=1;m<=mx*2;m<<=1,lg++); for (int i=0;i<m;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(lg-1)); fft(a,1,m);fft(b,1,m);fft(c,1,m); for (int i=0;i<m;i++) ans[i]=a[i]+0.5*(a[i]*a[i]-b[i])+1.0/6*(a[i]*a[i]*a[i]-3.0*a[i]*b[i]+2.0*c[i]); fft(ans,-1,m); for (int i=0;i<m;i++) if ((int)(ans[i].real()+0.1)>0) printf("%d %d\n",i,(int)(ans[i].real()+0.1)); return 0;}
0 0
- bzoj 3771: Triple 快速傅里叶变换+容斥原理
- BZOJ 3771 Triple FFT+容斥原理
- bzoj 3771: Triple (容斥原理+生成函数+FFT)
- [BZOJ3771]Triple(快速傅里叶变换FFT)
- [SPOJ TSUM]Triple Sums(FFT+生成函数+容斥原理)
- [SPOJ TSUM]Triple Sums(FFT+容斥原理)
- [BZOJ3771][生成函数][FFT][容斥原理]Triple
- [BZOJ3771]Triple(生成函数+FFT+容斥原理)
- 【XSY1728】【BZOJ3771】Triple 生成函数 FFT 容斥原理
- bzoj-3771 Triple
- bzoj 3771 Triple
- BZOJ 3771 Triple
- 容斥定理 njust1923 triple
- bzoj 2194 [快速傅里叶变换]
- bzoj 2179 [快速傅里叶变换]
- bzoj 3527 [快速傅里叶变换]
- BZOJ 3527 快速傅里叶变换
- bzoj 3622 容斥原理
- PHP:创建存储函数
- Docker 入门学习笔记二:常用命令
- 关于如何将java web项目上传至腾讯云服务器
- Java之jar打包
- 主表和从表的一对一设置
- bzoj 3771: Triple 快速傅里叶变换+容斥原理
- MongoDB 3.4 高可用集群搭建(二)replica set 副本集
- 排序网络(Sorting network )构造的C语言实现
- iis连接oracle数据库的历程!
- 几种页面置换算法
- Kotlin & Spring boot 使用@Valid校验无效解决方法
- c#调用c++写成的dll文件,返回char*,返回数组,用ref接收,byte[] (zz)
- 蓝桥杯第四届个人赛
- 用mycat做读写分离:基于 MySQL主从复制