【bzoj3771】Triple FFT
来源:互联网 发布:潘小涛 知乎 编辑:程序博客网 时间:2024/04/29 08:03
a表示一个的方案数
b表示取两个相同的
c表示取三个相同的
最终,取一个的是a
取两个的是(a*a-b)/2
取三个的是(a*a*a-3*a*b+2*z)/6
b表示取两个相同的
c表示取三个相同的
最终,取一个的是a
取两个的是(a*a-b)/2
取三个的是(a*a*a-3*a*b+2*z)/6
a*a*a用FFT算就可以了
乘法是序列的卷积
#include<cstdio>#include<cstring>#include<cstdlib>#include<cmath>#include<algorithm>#include<iostream>#define maxn 200100 #define pi acos(-1)using namespace std;struct yts{double r,i;yts operator+(yts x) {yts ans;ans.r=r+x.r;ans.i=i+x.i;return ans;}yts operator-(yts x) {yts ans;ans.r=r-x.r;ans.i=i-x.i;return ans;}yts operator*(yts x) {yts ans;ans.r=r*x.r-i*x.i;ans.i=r*x.i+i*x.r;return ans;}}a[maxn],b[maxn],c[maxn],d[maxn],temp[maxn];long long ans[3][maxn];int n,m,mx,digit;int seq[maxn];void FFT(yts x[],int n,int type){if (n==1) return;for (int i=0;i<n;i+=2) temp[i>>1]=x[i],temp[i+n>>1]=x[i+1];memcpy(x,temp,sizeof(yts)*n);yts *l=x,*r=l+(n>>1);FFT(l,n>>1,type);FFT(r,n>>1,type);yts root,w;root.r=cos(2*type*pi/n),root.i=sin(2*type*pi/n);w.r=1;w.i=0;for (int i=0;i<(n>>1);i++,w=w*root) temp[i]=l[i]+w*r[i],temp[(n>>1)+i]=l[i]-w*r[i];memcpy(x,temp,sizeof(yts)*n);}int main(){scanf("%d",&n);for (int i=1;i<=n;i++){int x;scanf("%d",&x);seq[i]=x;a[x].r++;ans[0][x]++;mx=max(mx,x);}mx*=3;for (digit=1;digit<mx;digit<<=1);FFT(a,digit,1);for (int i=0;i<=digit;i++) b[i]=a[i]*a[i],c[i]=a[i]*a[i]*a[i];FFT(b,digit,-1);FFT(c,digit,-1);for (int i=1;i<=n;i++) d[2*seq[i]].r++;FFT(d,digit,1);for (int i=0;i<=digit;i++) d[i]=d[i]*a[i];FFT(d,digit,-1);for (int i=0;i<=digit;i++){ans[1][i]=(long long)(b[i].r/digit+0.5);ans[2][i]=(long long)(c[i].r/digit+0.5)-3*(long long)(d[i].r/digit+0.5);}for (int i=1;i<=n;i++) ans[1][2*seq[i]]--,ans[2][3*seq[i]]+=2;for (int i=0;i<=digit;i++) ans[1][i]/=2,ans[2][i]/=6;for (int i=0;i<=digit;i++) if (ans[0][i]+ans[1][i]+ans[2][i]>0) printf("%d %lld\n",i,ans[0][i]+ans[1][i]+ans[2][i]);return 0;}
0 0
- 【bzoj3771】Triple FFT
- [BZOJ3771]Triple(FFT+生成函数)
- [生成函数+容斥+FFT] BZOJ3771: Triple
- [BZOJ3771]Triple(快速傅里叶变换FFT)
- BZOJ3771: Triple
- BZOJ3771 Triple
- bzoj3771 Triple
- BZOJ3771: Triple
- bzoj3771 Triple
- BZOJ3771: Triple
- [BZOJ3771][生成函数][FFT][容斥原理]Triple
- [BZOJ3771]Triple(生成函数+FFT+容斥原理)
- 【XSY1728】【BZOJ3771】Triple 生成函数 FFT 容斥原理
- 【FFT】 SPOJ Triple Sums
- 【SPOJ】Triple Sums【FFT】
- 【SPOJ】Triple Sums【FFT】
- [SPOJ TSUM] Triple Sums [FFT]
- Triple Sums(FFT 建模)
- C语言中数组与指针
- PHP学习笔记-流程控制语句
- TCP/IP协议(1)——基础知识(协议与接口)
- Bzoj:[Poi2011]Lightning Conductor:决策单调性优化DP详解
- HTML页面绝对定位+相对定位
- 【bzoj3771】Triple FFT
- c++中指针和引用的区别
- 提取字符串最长子串
- Gradle构建工具基础
- 【笔试/面试】排列组合与概率计算(二)
- 开更
- BZOJ_P2118 墨墨的等式(最短路)
- android 6.0权限动态管理
- iOS 之滤镜 详解