BZOJ 3028 食物

母函数

论文例题。不会的就去看论文吧：《母函数的性质及应用》

思想就是对于每一种食物，对于它的一个可行取法 $i$，$x^i$为1，意义就是可以有这种取法。其他不可取到的项为0。

然后把这些函数式乘起来，回忆多项式乘法，此时 $x^n$ 的系数的意义就是由许多能组成 $x^n$ 的 $x^i$ 的系数乘起来得到，即不同方案数。

#include<cstdio>
#define MOD 10007
using namespace std;
int main()
{
    int r = 0; char c = getchar();
    for(; c < '0' || c > '9'; c = getchar());
    for(; c >='0'&& c <='9'; r=(r*10+c-'0')%MOD, c = getchar());
    printf("%lld\n",(long long)r*(r+1)*(r+2)/6%MOD);
}