[TJOI2019]唱、跳、rap和篮球（指数型生成函数+NTT+卷积）

最新推荐文章于 2020-07-30 21:43:26 发布
最新推荐文章于 2020-07-30 21:43:26 发布
阅读量514 收藏 1
点赞数 2
分类专栏： # 生成函数 # FFT/NTT/FWT # 容斥 文章标签： 生成函数 容斥 NTT
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/Emm_Titan/article/details/103744694
版权

文章目录

在这里插入图片描述

题目

大中锋的学院要组织学生参观博物馆，要求学生们在博物馆中排成一队进行参观。他的同学可以分为四类：一部分最喜欢唱、一部分最喜欢跳、一部分最喜欢rap，还有一部分最喜欢篮球。如果队列中 k , k + 1 , k + 2 , k + 3 k,k + 1,k + 2,k + 3 k,k+1,k+2,k+3位置上的同学依次，最喜欢唱、最喜欢跳、最喜欢rap、最喜欢篮球，那么他们就会聚在一起讨论蔡徐坤。大中锋不希望这种事情发生，因为这会使得队伍显得很乱。大中锋想知道有多少种排队的方法，不会有学生聚在一起讨论蔡徐坤。两个学生队伍被认为是不同的，当且仅当两个队伍中至少有一个位置上的学生的喜好不同。由于合法的队伍可能会有很多种，种类数对998244353取模。

输入格式
输入数据只有一行。每行5个整数，第一个整数n，代表大中锋的学院要组织多少人去参观博物馆。接下来四个整数a、b、c、d，分别代表学生中最喜欢唱的人数、最喜欢跳的人数、最喜欢rap的人数和最喜欢篮球的人数。保证 a + b + c + d ≥ n a+b+c+d \ge n a+b+c+d≥n。

输出格式
每组数据输出一个整数，代表你可以安排出多少种不同的学生队伍，使得队伍中没有学生聚在一起讨论蔡徐坤。结果对 998244353 998244353 998244353取模。

输入输出样例
输入
4 4 3 2 1
输出
174

输入
996 208 221 132 442
输出
442572391

说明/提示
对于20%的数据，有 n = a = b = c = d ≤ 500 n=a=b=c=d\le500 n=a=b=c=d≤500
对于100%的数据，有 n ≤ 1000 n \le 1000 n≤1000 ， a , b , c , d ≤ 500 a, b, c, d \le 500 a,b,c,d≤500

题解

考虑枚举有 i i i堆人在讨论cxk小姐姐，那么被cxk迷倒的人就有 4 i 4i 4i个，且每一堆人是挨在一起的，
所以我们可以把这 4 i 4i 4i个人缩成 i i i个群体，每一个群体可以展开成为 4 4 4人
那么现在就只用考虑剩下的 n − 3 i n-3i n−3i人，放置的方案数就是 C n − 3 i i C_{n-3i}^i Cn−3ii​
简单用整体法证明一下：
在 n − 3 i n-3i n−3i人中，要选 i i i个拿来讨论， n − 4 i n-4i n−4i个不讨论，方案数是对应的 C n − 3 i i = C n − 3 i n − 4 i C_{n-3i}^i=C_{n-3i}^{n-4i} Cn−3ii​=Cn−3in−4i​
也可以理解为在 n − 3 i n-3i n−3i个空格里面找 i i i个起点开始讨论小姐姐

接着枚举好了 i i i及它们可能出现的地方 C n − 3 i i C_{n-3i}^i Cn−3ii​，我们就要去考虑剩下 n − 4 i n-4i n−4i人的排列，可以乱来
( n − 4 i ) ! (n-4i)! (n−4i)!
但是 a t t e n t i o n attention attention，如果看了我的上一篇指数型生成函数专练，就会与这里有一点相通
题目说的方案数不同的要求是至少有一个位置上的学生的喜好不同，而不是学生本人不同
所以我们要除掉喜好篮球，跳舞，唱歌，rap本身内部的乱拍，因为从爱好上来看是看不出来排列不同的
设 a , b , c , d a,b,c,d a,b,c,d分别代表学生中最喜欢唱的人数、最喜欢跳的人数、最喜欢rap的人数和最喜欢篮球的总人数，我们考虑的是剩下的不讨论cxk姐姐的学生乱排，所以要剪掉外层所枚举的去参与讨论的人数
( n − 4 i ) ! ( a − i ) ! ( n − i ) ! ( c − i ) ! ( d − i ) ! \frac{(n-4i)!}{(a-i)!(n-i)!(c-i)!(d-i)!} (a−i)!(n−i)!(c−i)!(d−i)!(n−4i)!​

最后写出每种喜好的生成函数，以喜欢篮球的人为例：
∑ i = 0 c x i i ! \sum_{i=0}^{c}\frac{x^i}{i!} i=0∑c​i!xi​
把四种爱好卷起来，卷出最后乘积的第 n − 4 i n-4i n−4i项就是我们需要除掉的东西，乘上 ( n − 4 i ) ! (n-4i)! (n−4i)!就是真正的乱排数

因为带取模，所以是用 N T T NTT NTT跑，除的话就要变成乘逆元，所以我们可以在卷的时候就直接卷逆元

但是我们又发现虽然假设的是 i i i堆人在讨论，但是统计答案的时候却把 ≥ i \ge i ≥i堆人讨论的情况都统计了
而且当统计 i = 1 i=1 i=1时，会算两遍至少两堆人讨论的方案，三遍至少三堆人讨论的方案…在统计 i = 2 i=2 i=2时，会算三遍至少三堆人讨论的方案…
当统计 i i i的方案时，会多算 C j i C_j^i Cji​次至少 j j j堆人讨论的方案，所以我们可以用容斥来解决

总结一下答案应该是
∑ i = 0 m i n ( − 1 ) i ∗ C n − 3 i i ∗ ( n − 4 i ) ! ( a − i ) ! ( n − i ) ! ( c − i ) ! ( d − i ) ! \sum_{i=0}^{min}(-1)^i*C_{n-3i}^i*\frac{(n-4i)!}{(a-i)!(n-i)!(c-i)!(d-i)!} i=0∑min​(−1)i∗Cn−3ii​∗(a−i)!(n−i)!(c−i)!(d−i)!(n−4i)!​
在这里插入图片描述

code1（NTT）

#include <cstdio>
#include <iostream>
using namespace std;
#define mod 998244353
#define LL long long
#define MAXN 10005
int n, anum, bnum, cnum, dnum;
LL pi, ni, result;
LL a[MAXN], b[MAXN], c[MAXN], d[MAXN], rev[MAXN], fac[MAXN], Invfac[MAXN];

LL qkpow ( LL x, LL y ) {
	LL ans = 1;
	while ( y ) {
		if ( y & 1 )
			ans = ans * x % mod;
		x = x * x % mod;
		y >>= 1;
	}
	return ans;
}

LL C ( int n, int m ) {
	return fac[n] * Invfac[m] % mod * Invfac[n - m] % mod;
}

void NTT ( LL *c, LL limit, LL f ) {
	for ( LL i = 0;i < limit;i ++ )
		if ( i < rev[i] )
			swap ( c[i], c[rev[i]] );
	for ( LL i = 1;i < limit;i <<= 1 ) {
		LL omega = qkpow ( f == 1 ? pi : ni, ( mod - 1 ) / ( i << 1 ) );
		for ( LL j = 0;j < limit;j += ( i << 1 ) ) {
			LL w = 1;
			for ( LL k = 0;k < i;k ++, w = w * omega % mod ) {
				LL x = c[k + j], y = w * c[i + j + k] % mod;
				c[k + j] = ( x + y ) % mod;
				c[k + j + i] = ( x - y + mod ) % mod;
			}
		}
	}
	LL inv = qkpow ( limit, mod - 2 );
	if ( f == -1 )
		for ( LL i = 0;i < limit;i ++ )
			c[i] = c[i] * inv % mod;
}

void init () {
	Invfac[0] = fac[0] = 1;
	for ( int i = 1;i <= n;i ++ )
		fac[i] = fac[i - 1] * i % mod;
	Invfac[n] = qkpow ( fac[n], mod - 2 );
	for ( int i = n - 1;i;i -- )
		Invfac[i] = Invfac[i + 1] * ( i + 1 ) % mod;
}

LL solve ( int n, int A, int B, int C, int D ) {
	LL len = 1, l = 0;
	while ( len < ( ( A + B + C + D ) << 1 ) ) {
		len <<= 1;
		l ++;
	}
	for ( LL i = 0;i < len;i ++ )
		rev[i] = ( rev[i >> 1] >> 1 ) | ( ( i & 1 ) << ( l - 1 ) );
	for ( int i = 0;i < len;i ++ )
		a[i] = ( i <= A ? Invfac[i] : 0 );
	for ( int i = 0;i < len;i ++ )
		b[i] = ( i <= B ? Invfac[i] : 0 );
	for ( int i = 0;i < len;i ++ )
		c[i] = ( i <= C ? Invfac[i] : 0 );
	for ( int i = 0;i < len;i ++ )
		d[i] = ( i <= D ? Invfac[i] : 0 );
	NTT ( a, len, 1 );NTT ( b, len, 1 );NTT ( c, len, 1 );NTT ( d, len, 1 );
	for ( int i = 0;i < len;i ++ )
		a[i] = a[i] * b[i] % mod * c[i] % mod * d[i] % mod;
	NTT ( a, len, -1 );
	return a[n] * fac[n] % mod;
}

int main() {
	pi = 3;
	ni = mod / pi + 1;
	scanf ( "%d %d %d %d %d", &n, &anum, &bnum, &cnum, &dnum );
	init();
	int k = min ( min ( min ( min ( anum, bnum ), cnum ), dnum ), n / 4 );
	anum -= k;bnum -= k;cnum -= k;dnum -= k;
	result = 0;
	for ( k;k >= 0;k -- ) {
		LL tmp = C ( n - 3 * k, k ) % mod * solve ( n - 4 * k, anum, bnum, cnum, dnum ) % mod;
		anum ++;bnum ++;cnum ++;dnum ++;
		result = ( result + ( ( k & 1 ) ? mod - tmp : tmp ) ) % mod;
	}
	printf ( "%lld\n", result );
	return 0;
}

code2（EGF+卷积）

#include <cstdio>
#include <iostream>
using namespace std;
#define mod 998244353
#define LL long long
#define MAXN 1005
int n, a, b, c, d;
LL result;
LL foldAB[MAXN], foldCD[MAXN], fac[MAXN], Invfac[MAXN];

LL qkpow ( LL x, LL y ) {
	LL ans = 1;
	while ( y ) {
		if ( y & 1 )
			ans = ans * x % mod;
		x = x * x % mod;
		y >>= 1;
	}
	return ans;
}

LL C ( int n, int m ) {
	return fac[n] * Invfac[m] % mod * Invfac[n - m] % mod;
}

void Fac () {
	Invfac[0] = fac[0] = 1;
	for ( int i = 1;i <= n;i ++ )
		fac[i] = fac[i - 1] * i % mod;
	Invfac[n] = qkpow ( fac[n], mod - 2 );
	for ( int i = n - 1;i;i -- )
		Invfac[i] = Invfac[i + 1] * ( i + 1 ) % mod;
}

int main() {
	scanf ( "%d %d %d %d %d", &n, &a, &b, &c, &d );
	Fac();
	int k = min ( min ( min ( min ( a, b ), c ), d), n / 4 );
	a -= k;b -= k;c -= k;d -= k;
	for ( int i = 0;i <= a;i ++ )
		for ( int j = 0;j <= b;j ++ )
			foldAB[i + j] = ( foldAB[i + j] + Invfac[i] * Invfac[j] % mod ) % mod;
	for ( int i = 0;i <= c;i ++ )
		for ( int j = 0;j <= d;j ++ )
			foldCD[i + j] = ( foldCD[i + j] + Invfac[i] * Invfac[j] % mod ) % mod;
	result = 0;
	for ( k;k >= 0;k -- ) {
		LL ans = 0;
		int tp = n - 4 * k;
		for ( int i = 0;i <= tp;i ++ )
			ans = ( ans + foldAB[i] * foldCD[tp - i] % mod ) % mod;
		ans = ans * fac[tp] % mod * C ( n - k * 3, k ) % mod;
		result = ( result + ( ( k & 1 ) ? mod - ans : ans ) ) % mod;
		a ++;b ++;c ++;d ++;
		for ( int i = 0;i <= a;i ++ )
			foldAB[i + b] = ( foldAB[i + b] + Invfac[i] * Invfac[b] % mod ) % mod;
		for ( int i = 0;i <= b;i ++ )
			foldAB[i + a] = ( foldAB[i + a] + Invfac[i] * Invfac[a] % mod ) % mod;
		foldAB[a + b] = ( foldAB[a + b] - Invfac[a] * Invfac[b] % mod + mod ) % mod;
		for ( int i = 0;i <= c;i ++ )
			foldCD[i + d] = ( foldCD[i + d] + Invfac[i] * Invfac[d] % mod ) % mod;
		for ( int i = 0;i <= d;i ++ )
			foldCD[i + c] = ( foldCD[i + c] + Invfac[i] * Invfac[c] % mod ) % mod;
		foldCD[c + d] = ( foldCD[c + d] - Invfac[c] * Invfac[d] % mod ) % mod;
	}
	printf ( "%lld\n", result );
	return 0;
}

代码或者思路有任何问题欢迎评论
在这里插入图片描述

[TJOI2019]唱、跳、rap和篮球
C202044zxy的博客
12-27 232
一、题目 点此看题 二、解法 先枚举iii，表示有iii段人在讨论蔡徐坤，那么在nnn中选iii段的方案数是Cn−3iiC_{n-3i}^iCn−3ii​，可以理解为将444个点看成一个点，就是在n−3in-3in−3i个点中选出iii个点作为讨论蔡徐坤的点，可以用组合数。 考虑对iii容斥，那么答案就是： ∑i=0up(−1)i×Cn−3ii×[剩下数的可重集的排列]\sum_{i=0}^{up...
Luogu P5339「TJOI2019」唱、跳、rap 和篮球 题解
C2021hyf的博客
08-26 271
题意就不用讲了吧…… 鸡你太美！！！ 题意： 有 444 种喜好不同的人，分别最爱唱、跳、 rapraprap、篮球，他们个数分别为 A,B,C,DA,B,C,DA,B,C,D ，现从他们中挑选出 nnn 个人并进行排列，规定不能出现喜爱唱、跳、 rap、篮球的人在序列中依次出现，问合法方案数。 下文将喜爱唱、跳、 rapraprap、篮球的人依次出现的区间称为聚集区间，长度为 444。 思路（容斥原理 + 生成函数 + NTT\mathcal{NTT}NTT） 首先，我们可以发现如果顺着求方案数并不好求。
[TJOI2019]唱、跳、rap和篮球——NTT+生成函数+容斥
weixin_34367257的博客
05-03 134
题目链接： [TJOI2019]唱、跳、rap和篮球 直接求不好求，我们考虑容斥，求出至少有$i$个聚集区间的方案数$ans_{i}$，那么最终答案就是$\sum\limits_{i=0}^{n}(-1)^i\ ans_{i}$ 那么现在只需要考虑至少有$i$个聚集区间的方案数，我们枚举这$i$个区间的起始点位置，一共有$C_{n-3i}^{i}$种方案(可以看作是刚开始先将每个区间后...
「TJOI2019」唱、跳、rap 和篮球
TIANFENG_123456的博客
07-20 270
题目链接 题目分析 据说这是一道生成函数题 看到限制条件，我们首先想到的就是对有多少组讨论cxk的人进行容斥。然后就是求剩下的人随便放有多少种方法了。考虑现在每种剩\(a,b,c,d\)人，还需要排\(n\)人，那么方案数就是 \[ \sum_{i=1}^a\sum_{j=1}^b\sum_{k=1}^c\sum_{l=1}^d [i+j+k+l=n]{n\choose i}{n - i...
[BZOJ2259]异化多肽（生成函数+NTT+多项式求逆）
zyf2000
04-26 1114
题目描述传送门题目大意：有M种氨基酸，每个的相对分子质量为Ci，问组合出的肽链水解之后有多少个相对分子质量和为N（排列）。题解选的氨基酸的数量不限。 一个比较显然的思路是对于每一种氨基酸搞一个生成函数出来，也就是x^(Ci的倍数)项的系数为1，然后把M个卷起来就行了 但是这样复杂度是O(MNlogN)O(MNlogN)的，并且没法加速换一种思路… 将所有的氨基酸搞成一个生成函数，x^k的系数为
洛谷P4705 玩游戏（生成函数+分治ntt）
苟为蒟蒻又何妨
08-14 230
传送门 sbsbsb题。。。 考虑到Anst=1nm∑i=1n∑j=1m(ai+bj)t=1nm∑i=1n∑j=1m∑k=0tCtkaikbjt−k=t!nm∗[xn]((∑i≥0∑j=1najij!xi)(∑i≥0∑j=1mbjij!xi))\begin{aligned}Ans_t=&amp;\frac1{nm}\sum_{i=1}^n\sum_{j=1}^m(a_i+b_j)^t\\=...
指数型生成函数(EGF)略解
qq_38253946的博客
07-30 1609
因为我菜，所以我只能把重点的写一手。 学习了这篇文章 首先，生成函数是来简化运算、方便表达的。 ex=∑i=0∞xii!e^x=\sum_{i=0}^{\infty}\frac{x^i}{i!}ex=i=0∑∞​i!xi​ 然后给出一堆式子： G^(x)=1−x+x22!−x33!+⋯=e−xG^(x)=1+x22!+x44!+x66!⋯=ex+e−x2G^(x)=x+x33!+x55!+x77!⋯=ex−e−x2G^(x)=x−x22+x33⋯=ln⁡(1+x)G^(x)=x−x33!+x55!⋯=sin
[TJOI2019]唱、跳、rap和篮球 题解
GKxx
07-30 471
传送门 老年退役选手来写题解了 题意 有aaa个人喜欢唱，bbb个人喜欢跳，ccc个人喜欢rap，ddd个人喜欢篮球，现在要从中选出nnn个人排成一队，使得不存在位置kkk满足第kkk，k+1k+1k+1，k+2k+2k+2，k+3k+3k+3个人依次喜欢唱、跳、rap、篮球，求方案数。 两种方案不同，当且仅当有一个位置上的同学的喜好不同。 题解 为了方便，我们将连续的唱跳rap篮球四个人称为“i...
方波与单边指数信号卷积及卷积过程演示.zip_HDM_MATLAB方波与指数函数的卷积_方波_方波卷积指数
07-14
方波的生成，单边指数函数的生成，以及二者的卷积
洛谷5339 BZOJ5510 TJOI2019 唱、跳、rap和篮球 容斥 dp 组合数
forever_shi的博客
05-04 486
题目链接 题意： 给你一个nnn，表示你要选出一个nnn个人组成的队列。有四种不同的人，人数分别是a,b,c,da,b,c,da,b,c,d个，保证a+b+c+d&gt;=na+b+c+d&gt;=na+b+c+d>=n。我们要求队伍里不能有连续的四个位置依次是第一种人、第二种人、第三种人和第四种人。求方案数。对998244353998244353998244353取模。n&...
【题解】Luogu P5339 [TJOI2019]唱、跳、rap和篮球
weixin_30583563的博客
05-25 149
原题传送门 这题zsy写的是\(O(n^2)\)，还有NTT\(O(n^2\log n)\)的做法。我的是暴力，\(O(\frac{a b n}{4})\)，足够通过 考虑设\(f(i)\)表示序列中至少有\(i\)组人讨论cxk的方案数 这样就珂以进行容斥，易知答案ans为： \[ans=\sum_{i=0}^{Min(n/4,a,b,c,d)} (-1)^i f(i)\] 我们考虑如...
TJOI 2019 简要题解
FSYo 的博客
10-10 341
[TJOI2019]唱、跳、rap和篮球 暴力枚举有多少个不满足条件的，有 iii 个不满足的情况相当于在 n−3∗in-3*in−3∗i 个里面选 iii 个 其它的随便填，大力容斥，NTTNTTNTT即可 ans=∑i=0n/4(−1)i(n−3∗ii)(n−i∗4)!∑x1+x2+x3+x4=n−4∗i1x1!1x2!1x3!1x4!ans=\sum_{i=0}^{n/4}(-1)^i\bi...
指数函数
dengtan9179的博客
04-03 804
指数函数 定义 \(e^x\) 为使得 \((e^x)'=e^x\) 的数，由泰勒展开可以得到： \[ e^x=\sum _{i=0}^\infty \frac {x^i}{i!} \] 这个形式的特点在于指数的幂和下面的阶乘。它是 \(f_i=i!\) 的普通生成函数，\(f_i=1\) 的指数生成函数。它简单的形式和特殊的性质有很多应用。 下面是最近发现的几个。 自然数幂求和 观察到...
指数型生成函数 及 多项式求ln
weixin_30515513的博客
06-15 302
指数型生成函数 我们知道普通型生成函数解决的是组合问题，而指数型生成函数解决的是排列问题 对于数列\(\{a_n\}\)，我们定义其指数型生成函数为 \[G(x) = a_0 + a_1x + a_2\frac{x^2}{2!} + a_3\frac{x^3}{3!} + a_4\frac{x^4}{4!} + \dots = \sum\limits_{i = 0}^{\infty} a_i\f...
卷积（一）
selous的专栏
12-14 2900
写在开头：第一次接触到卷积还是在概率论里面，写在不管是卷积神经网络，还是图像处理里面的卷积运算，都已经涉及到这个概念。所以就想弄懂卷积是干什么的，看了网易的公开课，还不错，记录一下。——2016.12.15先不给出卷积公式，我们来看看卷积的定义。拉普拉斯变化Laplace transform：Laplace\ transform： F(s)=∫+∞0f(t)e−stdtF(s)=\int_0^{
【LOJ6363】「地底蔷薇」【点双】【指数型生成函数】【扩展拉格朗日反演】【多项式幂函数】
Lstdo的博客
01-31 361
传送门 题意：给定nnn和集合SSS,求出nnn个点的「所有极大点双连通分量的大小都在SSS 内」的不同简单无向连通图的个数 模 998244353998244353998244353。 n,∑i∈Si≤105n,\sum_{i\in S}i \leq10^5n,∑i∈S​i≤105 道理我都懂，可为啥我百度搜地灵殿ex终符一半都是cookie☆同人OI题 思路挺清奇的 首先注意题目中的点双定义是...
ntt算法的输入和输出
最新发布
05-13
NTT（Number Theoretic Transform）算法是一种快速傅里叶变换（FFT）的变种，它的输入和输出都是整数序列。 输入： - 一个长度为 N 的整数序列 a，其中 N 通常是 2 的幂次方。 输出： - 一个长度为 N 的整数序列 A，它是 a 的离散傅里叶变换（DFT）。 NTT 可以在模意义下进行计算，因此输入和输出通常都是在某个模数下的整数序列。
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值