[TJOI2019]唱、跳、rap和篮球

题目

洛谷

题意

$n$个同学排队，有分别喜欢唱、跳、rap、篮球的（下文用$c、t、r、l$代替）。将他们排成一列，使没有连续$4$个是$ctrl$。求合法的排列数。

思路

首先，对于这一类求方案数，我们按照常规，将它反过来，求至少有一个$ctrl$的方案数。

分解

鸡你太美

考虑将讨论鸡你太美的人提出来，计算讨论鸡你太美的人的方案数。
设讨论鸡你太美的有$m$堆人，则不讨论鸡你太美的人有$n-4m$个。将这样的一堆看做一个点，显然，就将排列看成了$n-3m$个点。
接下来，要从这总共的$n-3m$个点中选出不讨论鸡你太美$n-4m$个人。易得，这样的选取数为
$$C_{n-3m}^{n-4m}=C_{n-3m}^m$$

鸡你不美

接下来，计算不讨论鸡你太美(简称鸡你不美)的排列数。
这个时候，不能直接简单地进行组合。因为有可能两个鸡你不美的数列拼在一起，会产生鸡你太美，导致实际讨论鸡你太美的人大于$4m$。
怎么解决这个问题呢？考虑进行容斥，枚举实际的讨论组数$m$ ~ $\frac{n}{4}$。
设题中$a、b、c、d$中的最小值为$p$
发现$Ans=$
$$\sum _{i=0}^p(-1{)}^i\times C_{n-3i}^i\times (剩余n-4i个人的排列数)$$
为什么呢？
因为枚举至少$i$组的时候，会对至少$j$组($j\ge i)$的产生$C_j^i$的贡献
例如：
当枚举至少$1$组时，会对至少$2$组的产生$C_2^1=2$的贡献，对至少$3$组的产生$C_3^1=3$的贡献
当枚举至少$2$组时，会对至少$3$组的产生$C_3^2=3$的贡献，对至少$4$组的产生$C_4^2=6$的贡献
那么，剩余人的排列数如何算呢？
其实很简单。
因为剩余$c、t、r、l$分别有$a-m、b-m、c-m、d-m$个，总排列数为$(n-4m)!$

• $a+b+c+d=n$，即一个人只喜欢一种，排列数就为
  $$\frac{(n-4m)!}{(a-m)!(b-m)!(c-m)!(d-m)!}$$

• $a+b+c+d>n$，一个人可能喜欢多种，这个时候该怎么办呢？我们考虑在计算排列的同时进行枚举。得到式子：

$$Pai=\sum _{a^{\prime }\le a-m,b^{\prime }\le b-m,c^{\prime }\le c-m,d^{\prime }\le d-m}[a^{\prime }+b^{\prime }+c^{\prime }+d^{\prime }=n-4m]\frac{(n-4m)!}{a^{\prime }!b^{\prime }!c^{\prime }!d^{\prime }!}$$
$$=(n-4m)!\sum _{a^{\prime }\le a-m,b^{\prime }\le b-m,c^{\prime }\le c-m,d^{\prime }\le d-m}[a^{\prime }+b^{\prime }+c^{\prime }+d^{\prime }=n-4m]\frac{1}{a^{\prime }!}\frac{1}{b^{\prime }!}\frac{1}{c^{\prime }!}\frac{1}{d^{\prime }!}$$
很显然，是一个四元卷积，进行$NTT$求值即可。

Code：

咕咕咕