【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+x^2+...)^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
",1ll*n*(n+1)%MOD*(n+2)%MOD*INV%MOD);
return 0;
}