集训队作业2018: 青春猪头少年不会梦到兔女郎学姐(多限制容斥)(生成函数)(组合数学)

本文主要是介绍集训队作业2018: 青春猪头少年不会梦到兔女郎学姐(多限制容斥)(生成函数)(组合数学),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

题意

给定 n n n 种颜色的球,第 i i i 种颜色的球数量为 a i a_i ai 个,一种排列的贡献可以如下计算:先把这个序列首尾相连,然后把所有相邻且颜色相同的段拿出来,贡献为他们的长度之积,求所有排列的贡献和,原排列不同,首尾相连后相同的排列不算同一种。模 998244353 998244353 998244353


先考虑一个序列怎么做
我们对每一个 i i i 枚举将 a i a_i ai 分成 b i b_i bi 段,算出这种情况下的方案数已经每一种颜色的贡献
那么答案可以表示为(用 f ( a , b ) f(a,b) f(a,b) 表示把 a a a 分成 b b b 段的所有情况的系数之和)
∑ b w a y s ∗ ∏ f ( a i , b i ) \sum_{b}ways*\prod f(a_i,b_i) bwaysf(ai,bi)
发现 f f f 就是把 a a a 分成 b b b 块然后每一块选一个的方案数,巧妙地发现 f ( a , b ) = ( a + b − 1 b ∗ 2 − 1 ) = ( a + b − 1 a − b ) f(a,b)=\binom{a+b-1}{b*2-1}=\binom{a+b-1}{a-b} f(a,b)=(b21a+b1)=(aba+b1)
考虑求 w a y s ways ways,把所有颜色放在一起容斥,枚举强制合并的端点个数
w a y s = ∑ c ( ∏ ( b i − 1 c i ) ( − 1 ) c i ) ( ∑ b i − c i ) ! ∏ ( b i − c i ) ! ways=\sum_{c}(\prod \binom{b_i-1}{c_i}(-1)^{c_i})\frac{(\sum b_i-c_i)!}{\prod(b_i-c_i)!} ways=c((cibi1)(1)ci)(bici)!(bici)!
那么放到一起就是,换成枚举合并之后的段数
∑ b ∑ c ∏ f ( a i , b i ) ( b i − 1 c i − 1 ) ( − 1 ) b i − c i ( ∑ c i ) ! ∏ c i ! \sum_b\sum_c\prod f(a_i,b_i)\binom{b_i-1}{c_i-1}(-1)^{b_i-c_i}\frac{(\sum c_i)!}{\prod c_i!} bcf(ai,bi)(ci1bi1)(1)bicici!(ci)!
发现后面有一个 ∏ c i ! \prod c_i! ci!,考虑构造每一种颜色的 E G F EGF EGF
∑ c x c c ! ∑ b ≥ c f ( a i , b i ) ( b i − 1 c i − 1 ) ( − 1 ) b i − c i \sum_{c}\frac{x^c}{c!}\sum_{b\ge c}f(a_i,b_i)\binom{b_i-1}{c_i-1}(-1)^{b_i-c_i} cc!xcbcf(ai,bi)(ci1bi1)(1)bici
后面的一坨可以 N T T NTT NTT 求,然后分治 F F T FFT FFT 就可以全部乘起来

考虑环的情况怎么做
强制以 1 开头,那么 1 的生成函数的 x c x^c xc 会变成 x c − 1 x^{c-1} xc1
减去强制以 1 开头,以 1 结尾的个数,这个的 x c x^c xc 会变成 x c − 2 x^{c-2} xc2

但发现还是不对,如果环的循环是 T T T 的话,我们需要统计 T T T 遍,而实际上只统计了
b 1 ∑ a i T \frac{b_1}{\frac{\sum a_i}{T}} Taib1 遍,我们将第一个生成函数中含 b b b 的项除一个 b 1 b_1 b1,最后乘上 m m m 就可以了

#include<bits/stdc++.h>
#define cs const
using namespace std;
cs int N = 2e5 + 5;
int read(){int cnt = 0, f = 1; char ch = 0;while(!isdigit(ch)){ ch = getchar(); if(ch == '-') f = -1; }while(isdigit(ch)) cnt = cnt*10 + (ch-'0'), ch = getchar();return cnt * f;
}
#define poly vector<int>
#define pb push_back
cs int Mod = 998244353;
int add(int a, int b){ return a + b >= Mod ? a + b - Mod : a + b; }
int mul(int a, int b){ return 1ll * a * b % Mod; }
int dec(int a, int b){ return a - b < 0 ? a - b + Mod : a - b; }
void Add(int &a, int b){ a = add(a, b); }
void Dec(int &a, int b){ a = dec(a, b); }
void Mul(int &a, int b){ a = mul(a, b); }
int coe(int a){ return (a&1)?Mod-1:1; }
int ksm(int a, int b){ int ans=1; for(;b;b>>=1,a=mul(a,a)) if(b&1) ans=mul(ans,a); return ans; } 
int n, m, a[N];
int fac[N], ifac[N], inv[N];
poly f[N];
int C(int n, int m){ if(n<0||m<0||n<m) return 0; return mul(fac[n],mul(ifac[n-m],ifac[m])); }
void Fac_init(int n){fac[0]=fac[1]=ifac[0]=ifac[1]=inv[0]=inv[1]=1;for(int i=2; i<=n; i++) inv[i]=mul(Mod-Mod/i,inv[Mod%i]);for(int i=2; i<=n; i++) fac[i]=mul(fac[i-1],i), ifac[i]=mul(ifac[i-1],inv[i]);
}
cs int K = 18;
poly w[K+1];
poly rev; int up, bit;
void init(int deg){up=1,bit=0; while(up<deg) up<<=1,++bit; rev.resize(up);for(int i=0; i<up; i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(bit-1));
}
void NTT_init(){for(int i=1; i<=K; i++) w[i].resize(1<<i-1);int wn=ksm(3,(Mod-1)/(1<<K)); w[K][0] = 1; for(int i = 1; i < (1<<K-1); i++) w[K][i]=mul(w[K][i-1], wn);for(int i = K-1; i; i--) for(int j=0; j<(1<<i-1); j++) w[i][j]=w[i+1][j<<1];
}
void NTT(poly &a, int typ){for(int i=0; i<up; i++) if(i<rev[i]) swap(a[i],a[rev[i]]);for(int i=1,l=1; i<up; i<<=1,++l)for(int j=0; j<up; j+=(i<<1))for(int k=0; k<i; k++){int x=a[k+j], y=mul(a[k+j+i],w[l][k]);a[k+j]=add(x,y); a[k+j+i]=dec(x,y);}if(typ==-1){reverse(a.begin()+1,a.end());for(int iv=ksm(up,Mod-2),i=0; i<up; i++) a[i]=mul(a[i],iv);}
}
poly operator * (poly a, poly b){int deg = a.size()+b.size()-1; init(deg); a.resize(up); b.resize(up); NTT(a,1); NTT(b,1);for(int i=0; i<up; i++) a[i]=mul(a[i],b[i]); NTT(a,-1); a.resize(deg); return a;
}
poly Solve(int l, int r){if(l==r){ for(int i=1; i<f[l].size(); i++) Mul(f[l][i], ifac[i]); return f[l]; } int mid=(l+r)>>1; return Solve(l,mid) * Solve(mid+1,r);
}
int main(){n = read();for(int i = 1; i <= n; i++) m+=a[i]=read();	Fac_init(m); NTT_init();for(int i = 1; i <= n; i++){poly g(a[i]+1, 0), h(a[i]+1, 0);for(int j = 1; j <= a[i]; j++) g[j]=mul(fac[j-1], C(a[i]+j-1,a[i]-j));if(i==1) for(int j=1; j<=a[i]; j++) Mul(g[j], inv[j]);for(int j=0; j<=a[i]; j++) h[a[i]-j] = mul(ifac[j], coe(j));g=g*h; f[i].resize(a[i]+1);for(int j=1; j<=a[i]; j++) f[i][j]=mul(g[a[i]+j], ifac[j-1]);}poly F=Solve(2,n);poly G(a[1],0);for(int i=0; i<G.size(); i++) G[i]=mul(f[1][i+1], ifac[i]);G=G*F;int ans=0;for(int i=1; i<G.size(); i++) Add(ans, mul(G[i],fac[i]));if(a[1]>1){G=poly(a[1]-1,0);for(int i=0; i<G.size(); i++) G[i]=mul(f[1][i+2], ifac[i]);G=G*F;for(int i=1; i<G.size(); i++) Dec(ans, mul(G[i],fac[i]));} cout << mul(ans, m); return 0;
}

这篇关于集训队作业2018: 青春猪头少年不会梦到兔女郎学姐(多限制容斥)(生成函数)(组合数学)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/658692

相关文章

IDEA自动生成注释模板的配置教程

《IDEA自动生成注释模板的配置教程》本文介绍了如何在IntelliJIDEA中配置类和方法的注释模板,包括自动生成项目名称、包名、日期和时间等内容,以及如何定制参数和返回值的注释格式,需要的朋友可以... 目录项目场景配置方法类注释模板定义类开头的注释步骤类注释效果方法注释模板定义方法开头的注释步骤方法注

Python如何自动生成环境依赖包requirements

《Python如何自动生成环境依赖包requirements》:本文主要介绍Python如何自动生成环境依赖包requirements问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑... 目录生成当前 python 环境 安装的所有依赖包1、命令2、常见问题只生成当前 项目 的所有依赖包1、

shell编程之函数与数组的使用详解

《shell编程之函数与数组的使用详解》:本文主要介绍shell编程之函数与数组的使用,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录shell函数函数的用法俩个数求和系统资源监控并报警函数函数变量的作用范围函数的参数递归函数shell数组获取数组的长度读取某下的

MySQL高级查询之JOIN、子查询、窗口函数实际案例

《MySQL高级查询之JOIN、子查询、窗口函数实际案例》:本文主要介绍MySQL高级查询之JOIN、子查询、窗口函数实际案例的相关资料,JOIN用于多表关联查询,子查询用于数据筛选和过滤,窗口函... 目录前言1. JOIN(连接查询)1.1 内连接(INNER JOIN)1.2 左连接(LEFT JOI

MySQL中动态生成SQL语句去掉所有字段的空格的操作方法

《MySQL中动态生成SQL语句去掉所有字段的空格的操作方法》在数据库管理过程中,我们常常会遇到需要对表中字段进行清洗和整理的情况,本文将详细介绍如何在MySQL中动态生成SQL语句来去掉所有字段的空... 目录在mysql中动态生成SQL语句去掉所有字段的空格准备工作原理分析动态生成SQL语句在MySQL

MySQL中FIND_IN_SET函数与INSTR函数用法解析

《MySQL中FIND_IN_SET函数与INSTR函数用法解析》:本文主要介绍MySQL中FIND_IN_SET函数与INSTR函数用法解析,本文通过实例代码给大家介绍的非常详细,感兴趣的朋友一... 目录一、功能定义与语法1、FIND_IN_SET函数2、INSTR函数二、本质区别对比三、实际场景案例分

C++ Sort函数使用场景分析

《C++Sort函数使用场景分析》sort函数是algorithm库下的一个函数,sort函数是不稳定的,即大小相同的元素在排序后相对顺序可能发生改变,如果某些场景需要保持相同元素间的相对顺序,可使... 目录C++ Sort函数详解一、sort函数调用的两种方式二、sort函数使用场景三、sort函数排序

C语言函数递归实际应用举例详解

《C语言函数递归实际应用举例详解》程序调用自身的编程技巧称为递归,递归做为一种算法在程序设计语言中广泛应用,:本文主要介绍C语言函数递归实际应用举例的相关资料,文中通过代码介绍的非常详细,需要的朋... 目录前言一、递归的概念与思想二、递归的限制条件 三、递归的实际应用举例(一)求 n 的阶乘(二)顺序打印

Java利用docx4j+Freemarker生成word文档

《Java利用docx4j+Freemarker生成word文档》这篇文章主要为大家详细介绍了Java如何利用docx4j+Freemarker生成word文档,文中的示例代码讲解详细,感兴趣的小伙伴... 目录技术方案maven依赖创建模板文件实现代码技术方案Java 1.8 + docx4j + Fr

C/C++错误信息处理的常见方法及函数

《C/C++错误信息处理的常见方法及函数》C/C++是两种广泛使用的编程语言,特别是在系统编程、嵌入式开发以及高性能计算领域,:本文主要介绍C/C++错误信息处理的常见方法及函数,文中通过代码介绍... 目录前言1. errno 和 perror()示例:2. strerror()示例:3. perror(