首页 > 代码库 > BZOJ 3028 食物 ——生成函数

BZOJ 3028 食物 ——生成函数

把所有东西的生成函数搞出来。

发现结果是x*(1-x)^(-4)

然后把(1-x)^(-4)求逆,得到(1+x+x^2+...)^4

然后考虑次数为n的项前的系数,就相当于选任意四个非负整数构成n的方案数。

大概就是C(n+3,3)

前面还有一项是x,所以n--即可。

然后就A掉了。

#include <cstdio>
#include <cstring>
#define ll long long
const int inv=1668;
const int md=10007;
int n;char s[505];
int main()
{
    scanf("%s",s+1);
    for (int i=1;i<=strlen(s+1);++i)
        n=(10*n+s[i]-‘0‘)%md;
    printf("%lld\n",((ll)n*(n+1)*(n+2)*inv)%(ll)md);
}

  

BZOJ 3028 食物 ——生成函数