[TJOI2019]唱、跳、rap和篮球——NTT+生成函数+容斥

题目链接:

[TJOI2019]唱、跳、rap和篮球

直接求不好求,我们考虑容斥,求出至少有$i$个聚集区间的方案数$ans_{i}$,那么最终答案就是$\sum\limits_{i=0}^{n}(-1)^i\ ans_{i}$

那么现在只需要考虑至少有$i$个聚集区间的方案数,我们枚举这$i$个区间的起始点位置,一共有$C_{n-3i}^{i}$种方案(可以看作是刚开始先将每个区间后三个位置去掉,从剩下$n-3i$个位置中选出$i$个区间起点,然后再在每个起点后面加上三个位置)。

那么剩下的$n-4i$个位置就是随便放这四种学生,假设第$j$种学生放了$a_{j}$个、一共有$num_{j}$个,那么方案数就是$\frac{(n-4i)!}{\prod_{j=1}^{4}a_{j}!}$。

由此可以构造出这四种学生的生成函数,以第一种学生为例:$\sum\limits_{j=0}^{num_{1}-i}\frac{x^j}{j!}$

将四个生成函数分别用$NTT$乘在一起然后取$x^{n-4i}$前的系数乘上$(n-4i)!$即可得到$n-4i$个位置随便放的方案数。

#include<set>
#include<map>
#include<cmath>
#include<stack>
#include<queue>
#include<bitset>
#include<cstdio>
#include<vector>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int mod=998244353;
int f[3000];
int g[3000];
int inv[2000];
int fac[2000];
int mask;
int n,a,b,c,d;
int ans;
int mn,mx;
int quick(int x,int y)
{
	int res=1;
	while(y)
	{
		if(y&1)
		{
			res=1ll*res*x%mod;
		}
		x=1ll*x*x%mod;
		y>>=1;
	}
	return res;
}
void NTT(int *a,int len,int opt)
{
	for(int i=0,k=0;i<len;i++)
	{
		if(i>k)
		{
			swap(a[i],a[k]);
		}
		for(int j=len>>1;(k^=j)<j;j>>=1);
	}
	for(int i=2;i<=len;i<<=1)
	{
		int t=i>>1;
		int x=quick(3,(mod-1)/i);
		if(opt==-1)
		{
			x=quick(x,mod-2);
		}
		for(int j=0;j<len;j+=i)
		{
			int w=1;
			for(int k=j;k<j+t;k++)
			{
				int tmp=1ll*a[k+t]*w%mod;
				a[k+t]=(a[k]-tmp+mod)%mod;
				a[k]=(a[k]+tmp)%mod;
				w=1ll*w*x%mod;
			}
		}
	}
	if(opt==-1)
	{
		int x=quick(len,mod-2);
		for(int i=0;i<len;i++)
		{
			a[i]=1ll*a[i]*x%mod;
		}
	}
}
int C(int n,int m)
{
	return 1ll*fac[n]*inv[m]%mod*inv[n-m]%mod;
}
int solve(int x)
{
	memset(f,0,sizeof(f));
	memset(g,0,sizeof(g));
	for(int i=0;i<=a-x;i++)
	{
		f[i]=inv[i];
	}
	for(int i=0;i<=b-x;i++)
	{
		g[i]=inv[i];
	}
	NTT(f,mask,1);
	NTT(g,mask,1);
	for(int i=0;i<mask;i++)
	{
		f[i]=1ll*f[i]*g[i]%mod;
	}
	memset(g,0,sizeof(g));
	for(int i=0;i<=c-x;i++)
	{
		g[i]=inv[i];
	}
	NTT(g,mask,1);
	for(int i=0;i<mask;i++)
	{
		f[i]=1ll*f[i]*g[i]%mod;
	}
	memset(g,0,sizeof(g));
	for(int i=0;i<=d-x;i++)
	{
		g[i]=inv[i];
	}
	NTT(g,mask,1);
	for(int i=0;i<mask;i++)
	{
		f[i]=1ll*f[i]*g[i]%mod;
	}
	NTT(f,mask,-1);
	return 1ll*f[n-4*x]*fac[n-4*x]%mod*C(n-3*x,x)%mod;
}
int main()
{
	inv[1]=inv[0]=fac[0]=1;
	for(int i=1;i<=1000;i++)
	{
		fac[i]=1ll*fac[i-1]*i%mod;
	}
	for(int i=2;i<=1000;i++)
	{
		inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod;
	}
	for(int i=1;i<=1000;i++)
	{
		inv[i]=1ll*inv[i-1]*inv[i]%mod;
	}
	mask=1;
	scanf("%d%d%d%d%d",&n,&a,&b,&c,&d);
	mn=min(n/4,min(min(a,b),min(c,d)));
	mx=max(max(a,b),max(c,d));
	while(mask<=(mx<<2))
	{
		mask<<=1;
	}
	for(int i=0;i<=mn;i++)
	{
		if(i&1)
		{
			ans=(ans-solve(i)+mod)%mod;
		}
		else
		{
			ans=(ans+solve(i))%mod;
		}
	}
	printf("%d",ans);
}

原文地址:https://www.cnblogs.com/Khada-Jhin/p/10806596.html

时间: 2024-11-09 12:10:32

[TJOI2019]唱、跳、rap和篮球——NTT+生成函数+容斥的相关文章

[TJOI2019]唱、跳、rap和篮球——容斥原理+生成函数

先附一组sd图 然后放上原题链接 注意,队伍不同指的是喜好不同,不是人不同 先想到\(DP\),然后你会发现并没有什么优秀的状态设计,然后我们考虑容斥 设\(lim\)表示选的癌坤组数的上限,\(f_i\)为先选出来\(i\)组剩下随便排的方案数,那么答案就是 \[\sum\limits_{i=0}^{lim}(-1)^i\times\ f_i\] 于是问题转化为了求\(f_i\).显然\(f_i\)可以表示为一个组合数再乘一个东西,具体来说组合数代表在\(n\)个同学中选\(i\)组癌坤的方案

[TJOI2019]唱,跳,rap,篮球(生成函数,组合数学,NTT)

算是补了个万年大坑了吧. 根据 wwj 的题解(最准确),设一个方案 \(S\)(不一定合法)的鸡你太美组数为 \(w(S)\). 答案就是 \(\sum\limits_{S}[w(S)=0]\). 用二项式定理:\(\sum\limits_{S}[w(S)=0]=\sum\limits_{S}(1-1)^{w(S)}=\sum\limits_{S}\sum\limits_{i\ge 0}(-1)^i\binom{w(S)}{i}=\sum\limits_{i\ge 0}(-1)^i\sum\l

[luogu5339] [TJOI2019]唱、跳、rap和篮球(容斥原理+组合数学)(不用NTT)

[luogu5339] [TJOI2019]唱.跳.rap和篮球(容斥原理+组合数学)(不用NTT) 题面 略 分析 首先考虑容斥,求出有i堆人讨论的方案. 可以用捆绑法,把每堆4个人捆绑成一组,其他人每个人一组.这样一共有\(n-3i\)组(这些组可以被看成相同的点). 我们从中选出n-4i个点,这些点展开成1个人,其他\(i\)个点展开成4个人.那么方案数就是\(C_{n-3i}^{n-4i}\) 由于\(i\)堆人的喜好已经确定,最终答案为\(\sum_{i=0}^n (-1)^i \ti

[TJOI2019]唱、跳、rap和篮球

题目 套路题啊 发现正向计数不太好记,考虑容斥 考虑求至少有\(i\)段连续四个位置是不合法的,容斥系数显然是\((-1)^i\) 我们先选出这样的\(i\)段长度为\(4\)的区间来 我们考虑分配一下空格,问题就等价于把\(n-4i\)个空格分到\(i+1\)组里,插板一下就能知道答案是\(\binom{n-3i}{i}\) 考虑剩下的\(n-4i\)个空格,我们现在需要把这些空格填满 先来考虑一下这四种分别填了\(a,b,c,d\)个方案数是多少 这\(n-4i\)个位置直接去排列,是\((

【题解】Luogu P5339 [TJOI2019]唱、跳、rap和篮球

原题传送门 这题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)\] 我们考虑如何计算\(f(i)\) 如果视讨论cxk的组为一个元素,则一共有\(n-3*i\)个元素 我们把问题转换成一个

Luogu P5339 [TJOI2019]唱、跳、rap和篮球

题目 设\(f_i\)表示从\((a-4i,b-4i,c-4i,d-4i)\)中选\(n-4i\)个排队的方案数. 那么我们可以容斥,答案为\(\sum\limits_{i=0}^{lim}(-1)^i{n-3i\choose i}f_i\). 考虑一下这个\(f\),它就是四个指数型生成函数卷起来\((\sum\limits_{i=0}^a\frac{x^i}{i!})(\sum\limits_{i=0}^b\frac{x^i}{i!})(\sum\limits_{i=0}^c\frac{x^

唱、跳、rap和篮球

唱.跳.rap和篮球 大中锋的学院要组织学生参观博物馆,要求学生们在博物馆中排成一队进行参观.他的同学可以分为四类:一部分最喜欢唱.一部分最喜欢跳.一部分最喜欢rap,还有一部分最喜欢篮球.如果队列中\(k,k + 1,k+1,k + 2,k+2,k + 3,k+3\)位置上的同学依次,最喜欢唱.最喜欢跳.最喜欢rap.最喜欢篮球,那么他们就会聚在一起讨论蔡徐坤.大中锋不希望这种事情发生,因为这会使得队伍显得很乱.大中锋想知道有多少种排队的方法,不会有学生聚在一起讨论蔡徐坤.两个学生队伍被认为是

HAOI2018 [HAOI2018]染色 【组合数 + 容斥 + NTT】

题目 为了报答小 C 的苹果, 小 G 打算送给热爱美术的小 C 一块画布, 这块画布可 以抽象为一个长度为 \(N\) 的序列, 每个位置都可以被染成 \(M\) 种颜色中的某一种. 然而小 C 只关心序列的 \(N\) 个位置中出现次数恰好为 \(S\) 的颜色种数, 如果恰 好出现了 \(S\) 次的颜色有 \(K\) 种, 则小 C 会产生 \(W_k\) 的愉悦度. 小 C 希望知道对于所有可能的染色方案, 他能获得的愉悦度的和对 1004535809 取模的结果是多少. 输入格式 从

HAOI 2018 染色(容斥+NTT)

题意 https://loj.ac/problem/2527 思路 设 \(f(k)\) 为强制选择 \(k\) 个颜色出现 \(s\) 种,其余任取的方案数. 则有 \[ f(k)={m\choose k}{n\choose sk}{(sk)!\over(s!)^k}(m-k)^{n-sk} \] 不难看出,这个方案可能包括了超过 \(k\) 种颜色,也有重复的方案,所以恰有 \(k\) 个颜色出现 \(s\) 种的方案 \(ans_k\) 满足 \[ ans_k=\sum_{i=k}^{\m