【BZOJ3028】食物（生成函数）

【BZOJ3028】食物（生成函数）

题面

一个人要带\(n\)个物品，共有\(8\)种物品，每种的限制分别如下：

偶数个;0/1个;0/1/2个;奇数个;

4的倍数个;0/1/2/3个;0/1个;3的倍数个。

求方案数。

\(n<=10^{500}\)

题解

对于八个物品的限制，分别构造生成函数，然后乘起来就是我们要的解。

化简后得到$$F(x)=\frac{x}{(1-x)^4}=x(1+x+x2+...)^4$$

现在考虑求第\(n\)项系数。考虑第\(n\)项的组合意义,即将\(n\)划分为\(4\)个自然数的方案数。

所以第\(n\)项为\(C_{n+2}^3\),时间复杂度\(O(1)\)

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MOD 10007
#define INV 1668
inline int read()
{
    RG int x=0,t=1;RG char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=-1,ch=getchar();
    while(ch<='9'&&ch>='0')x=(x*10+ch-48)%MOD,ch=getchar();
    return x*t;
}
int main()
{
	int n=read();
	printf("%lld\n",1ll*n*(n+1)%MOD*(n+2)%MOD*INV%MOD);
	return 0;
}