【BZOJ 3028】食物

传送门

---

problem

有若干种物品，每种物品选取的限制如下：

• $A$：偶数个
• $B$：$0$ 个或 $1$ 个
• $C$：小于等于 $2$ 个
• $D$：奇数个
• $E$：$4$ 的倍数个
• $F$：$0$个，$1$ 个，$2$ 个或 $3$ 个
• $G$：不超过一个
• $H$：$3$ 的倍数个

求从中选择n个物品的方案数模 $10007$ 的结果。

数据范围：$n\le 10^{500}$。

---

solution

这道题重在思维，代码非常简单。

我们先把这 $8$ 种东西的生成函数搞出来，即：

• $A(x)={\sum }_{i=0}^{\infty }{x}^{2i}=\frac{1}{1-x^2}$
• $B(x)=1+x$
• $C(x)=1+x+x^2$
• $D(x)={\sum }_{i=0}^{\infty }{x}^{2i+1}=\frac{x}{1-x^2}$
• $E(x)={\sum }_{i=0}^{\infty }{x}^{4i}=\frac{1}{1-x^4}$
• $F(x)=1+x+x^2+x^3$
• $G(x)=1+x$
• $H(x)={\sum }_{i=0}^{\infty }{x}^{3i}=\frac{1}{1-x^3}$

然后把它们乘起来，化个简，发现乘积是 $\frac{x}{(1-x{)}^4}$。

现在的目标就是，把这个式子展开，然后求指数为 $n$ 时的系数。

泰勒展开

我们令 $f(x)=\frac{x}{(1-x{)}^4}$，那把 $f(x)$ 在 $0$ 处展开，得：

$$f(x)=\sum _{i=0}^{\infty }\frac{f^{(i)}(0)}{i!}{x}^i$$

那么第 $n$ 项的系数就是 $\frac{f^{(n)}(0)}{n!}$，话句话说，我们需要求 $\frac{x}{(1-x{)}^4}$ 展开后在 $x=0$ 处的取值。

莱布尼茨公式

$$(uv{)}^{(n)}=\sum _{i=0}^n(\genfrac{}{}{0ex}{}{n}{i})u^{(i)}{v}^{(n-i)}$$

我们令 $u=x,v=(1-x{)}^{-4}=(x-1{)}^{-4}$。

发现只有当 $i=1$ 时，此时 $u^{\prime }=1$（其它时候 $u^{(i)}=0$），所以我们只需要求 $v^{(n-1)}$。

发现 $v$ 是一个复合函数（即令 $h(x)=x^{-4},g(x)=x-1$ 的话，$v=h(g(x))$），那么有：

$$v^{\prime }=h^{\prime }(g(x))g^{\prime }(x)=h^{\prime }(g(x))$$

那再求一次导，得到：

$$v^{\prime \prime }=h^{\prime \prime }(g(x))g^{\prime }(x)=h^{\prime \prime }(g(x))$$

那照这个规律，第 $(n-1)$ 次求导之后，就可以得到：

$$v^{(n-1)}=h^{(n-1)}(g(x))$$

先把 $h^{(n-1)}(x)$ 搞出来，即：

$$\begin{array}{rl}{h}^{(n-1)}(x)& =(x^{-4}{)}^{(n-1)}\\ & =(\prod _{i=0}^{n-2}(-4-i))x^{-4-(n-1)}\\ & =(-1{)}^{n-1}\frac{(n+2)!}{6}{x}^{-n-3}\end{array}$$

那么 $h^{(n-1)}(g(x))=h^{(n-1)}(x-1)=(-1{)}^{n-1}\frac{(n+2)!}{6}(x-1{)}^{-n-3}$。

由于我们需要的是这个式子在 $x=0$ 处的取值，那么：

$$h^{(n-1)}(g(0))=(-1{)}^{n-1}\frac{(n+2)!}{6}(-1{)}^{-n-3}=\frac{(n+2)!}{6}$$

然后乘个 $n$（因为 ${(}_1^n)=n$），再除个 $n!$（看泰勒展开部分），得到：

$$ans=\frac{n(n+1)(n+2)}{6}$$

那把 $n$ 读进来直接算就可以了。

---

code

#include<cstdio>
#include<cctype> 
#include<cstring>
#include<algorithm>
#define P 10007
using namespace std;
const int inv6=1668;
int n=0;
int main(){
	char c=getchar();
	while(isdigit(c))  n=(n*10+(c^'0'))%P,c=getchar();
	printf("%d\n",1ll*n*(n+1)*(n+2)*inv6%P);
	return 0;
}