BZOJ3028: 食物

好久没有更新博客了话说…
博主并没有退役….

我们写出每种食物的生成函数
承德汉堡：$1+x^2+x^4....=\frac{1}{1-x^2}$
可乐：$1+x$
鸡腿：$1+x+x^2=\frac{1-x^3}{1-x}$
蜜桃多：$x+x^3+x^5....=\frac{x}{1-x^2}$
鸡块：$1+x^4+x^8....=\frac{1}{1-x^4}$
包子：$1+x+x^2+x^3=\frac{1-x^4}{1-x}$
土豆片炒肉：$1+x$
面包：$1+x^3+x^6....=\frac{1}{1-x^3}$

乘一起，约分后得$\dfrac{x}{(1-x)^4}$
展开后第n项的系数就是$C_{n+2}^3$，于是套个Lucas就可以了

code：

#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<vector>
#include<string>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
#define inf 1e15
using namespace std;

const int Mod = 10007;
const int maxn = 1100;

int n;
char str[maxn];
int c[Mod+1][4];

int main()
{   
    c[0][0]=1ll;
    for(int i=1;i<=Mod;i++)
    {
        c[i][0]=1ll;
        for(int j=1;j<=3;j++) c[i][j]=(c[i-1][j-1]+c[i-1][j])%Mod;
    }

    scanf("%s",str); int len=strlen(str);
    for(int i=0;i<len;i++) n=(n*10+str[i]-'0')%Mod;
    n+=2; n%=Mod;
    printf("%d\n",c[n][3]);

    return 0;
}