生成函数Ⅰ

2022年7月5日和6日，day02

形式幂级数

多项式与形式幂级数

• 多项式：$A(x)={\sum }_{i=0}^n{a}_i{x}^i$
• 形式幂级数：$A(x)={\sum }_{i\ge 0}{a}_i{x}^i$
  其中 $a_i\in K$，$K$ 是一个域，通常，我们考虑 $K=\mathbb{R}$ 或 $K={\mathbb{Z}}_p$。
  注意这里的 $x$ 可以理解为一个独立于域 $K$ 的一个符号。

形式幂级数的运算

设 $A(x)={\sum }_{i\ge 0}{a}_i{x}^i$，$B(x)={\sum }_{i\ge 0}{b}_i{x}^i$。

• 加法：$A(x)+B(x)={\sum }_{i\ge 0}(a_i+b_i)x^i$
• 减法：$A(x)-B(x)={\sum }_{i\ge 0}(a_i-b_i)x^i$
• 乘法：$A(x)\times B(x)={\sum }_{k\ge 0}({\sum }_{i+j=k}{a}_i{b}_j)x^k$
  可以验证，形式幂级数在以上定义的 $+$，$-$，$\times$ 下形成一个环。
• 记号：记形式幂级数（或多项式）$A(x)$ 的 $x^n$ 项的系数为 $[x^n]A(x)$

常生成函数

定义：一个数列 { a n } \{a_n\} {an​} 对应的常生成函数为 $A(x)={\sum }_{n\ge 0}{a}_n{x}^n$。

例子：

• 有两种物体，其中取 $i$ 个第 1 种物体的方案数为 $a_i$ ，取 $j$ 个第 2 种物体的方案数为 $b_j$ ，求取 $k$ 个物体的方案数。

$$\begin{gathered} 令A(x)=\sum _{i\ge 0}{a}_i{x}^i，B(x)=\sum _{i\ge 0}{b}_i{x}^i。 \\ 则c_k=\sum _{i=0}^k{a}_i{b}_{k-i} \\ C(x)=\sum _{k\ge 0}\sum _{i=0}^k{a}_i{b}_{k-i}{x}^k=A(x)\times B(x) \end{gathered}$$

答案为 $[x^k]C(x)$。

• 在一个自助水果店，有苹果、香蕉、草莓三种水果，你可以取 $n$ 个水果，但是水果店要求，取的苹果数量必须是偶数，取的香蕉数量必须是 $3$ 的倍数，取的草莓数量不能超过 $5$ 个。求有多少种取 $n$ 个水果的方案。

$$\begin{gathered} 苹果：A(x)=\sum _{i\ge 0,2|i}{x}^i=1+x^2+x^4+x^6+\cdots \\ 香蕉：B(x)=\sum _{i\ge 0,3|i}{x}^i=1+x^3+x^6+x^9+\cdots \\ 草莓：C(x)=\sum _{i=0}^5{x}^i=1+x+x^2+x^3+x^4+x^5 \\ S(x)=A(x)B(x)C(x) \end{gathered}$$

答案为 $[x^n]S(x)$。

• 求不定方程 $x_1+x_2+\cdots +x_k=n$ 的解的数量，其中 $x_i$ 为整数，且 $l_i\le x_i\le r_i,(1\le i\le k)$。

$$\begin{gathered} 定义：F_i(x)=x^{l_i}+x^{l_i+1}+x^{l_i+2}+\cdots +x^{r_i} \\ F(x)=F_1(x)\times F_2(x)\times \cdots \times F_k(x) \end{gathered}$$

答案为 $[x^n]F(x)$。

• 求 $x_1+2x_2+3x_3=n$ 的正整数解的个数。

$$\begin{gathered} F_1(x)=\sum _{i\ge 1}{x}^i=x+x^2+x^3+\cdots \\ {F}_2(x)=\sum _{i\ge 1,2|i}{x}^i=x^2+x^4+x^6+\cdots \\ {F}_3(x)=\sum _{i\ge 1,3|i}{x}^i=x^3+x^6+x^9+\cdots \\ F(x)=F_1(x)F_2(x)F_3(x) \end{gathered}$$

答案为 $[x^n]F(x)$。

定理：设 S = { a 1 , a 2 , … , a k } S=\{a_1, a_2, \dots ,a_k\} S={a1​,a2​,…,ak​}，且 $a_i$ 可以取的次数的集合为 $M_i$ ，记 $F_i(x)={\sum }_{u\in M_i}{x}^u$，则从 $S$ 中取 $n$ 个元素组合成集合的方案数 $g(n)$ 的常生成函数 $G(x)={\sum }_{i\ge 0}g(i)x^i$，满足:
$$G(x)=F_1(x)F_2(x)\cdots F_k(x)$$

形式幂级数的逆元

• 形式幂级数 $A(x)$ 的逆元：$A(x)B(x)=1$
• 逆元存在条件：$[x^0]A(x)\ne 0$
• 暴力计算方法：递推

假设

$$\begin{gathered} A(x)=a_0+a_{1}x+a_2{x}^2+\cdots +a_n{x}^n+\cdots \\ B(x)=b_0+b_{1}x+b_2{x}^2+\cdots +b_n{x}^n+\cdots \\ A(x)B(x)=a_0{b}_0+(a_0{b}_1+a_1{b}_0)x+\cdots +(a_0{b}_n+a_1{b}_{n-1}+\cdots +a_n{b}_0)x^n+\cdots =1 \end{gathered}$$

那么

$$\begin{array}{rl}{a}_0{b}_0=1& \Rightarrow b_0=a_0^{-1}\\ a_0{b}_1+a_1{b}_0=0& \Rightarrow b_1=-(a_1{b}_0)a_0^{-1}\\ & ⋮\\ (a_0{b}_n+a_1{b}_{n-1}+\cdots +a_n{b}_0)=0& \Rightarrow b_n=-(a_1{b}_{n-1}+\cdots a_n{b}_0)a_0^{-1}\\ & ⋮\end{array}$$

常见的逆：

• $A(x)=1+x+x^2+\cdots$ 和 $B(x)=1-x$，即 $\frac{1}{1-x}={\sum }_{i\ge 0}{x}^i$
• $A(x)=1+ax+a^2{x}^2+\cdots$ 和 $B(x)=1-ax$，即 $\frac{1}{1-ax}={\sum }_{i\ge 0}{a}^i{x}^i$
• $A(x)=\left(\genfrac{}{}{0ex}{}{k-1}{0}\right)+\left(\genfrac{}{}{0ex}{}{k}{1}\right)x+\left(\genfrac{}{}{0ex}{}{k+1}{2}\right)x^2+\left(\genfrac{}{}{0ex}{}{k+2}{3}\right)x^3+\cdots$ 和 $B(x)=(1-x{)}^k$，即 $\frac{1}{(1-x{)}^k}={\sum }_{i\ge 0}\left(\genfrac{}{}{0ex}{}{i+k-1}{i}\right)x^i$

$$\begin{gathered} A(x)=(1+x+x^2+x^3+\cdots {)}^k \\ 其中[x^n]A(x)\iff x_1+x_2+\cdots +x_k=n，(x_i\ge 0) \\ 方案数：(\genfrac{}{}{0ex}{}{n+k-1}{k-1})=(\genfrac{}{}{0ex}{}{n+k-1}{n}) \end{gathered}$$

例题1：https://ac.nowcoder.com/acm/contest/24710/A

• 肥宅快乐水最多1个： $1+x$
• 大盘鸡最多2个： $1+x+x^2$
• 啤酒鸡最多3个： $1+x+x^2+x^3$
• 鸡翅偶数个： $1+x^2+x^4+\cdots =\frac{1}{1-x^2}=\frac{1}{(1-x)(1+x)}$
• 鸡汤奇数个： $x+x^3+x^5+\cdots =\frac{x}{1-x^2}=\frac{x}{(1-x)(1+x)}$
• 鸡块个数为4的倍数： $1+x^4+x^8+\cdots =\frac{1}{1-x^4}=\frac{1}{(1-x)(1+x+x^2+x^3)}$
• 鸡腿最多1个： $1+x$
• 鸡蛋个数为3的倍数： $1+x^3+x^6+\cdots =\frac{1}{1-x^3}=\frac{1}{(1-x)(1+x+x^2)}$

$$\begin{array}{rl}& (1+x)(1+x+x^2)(1+x+x^2+x^3)\frac{1}{(1-x)(1+x)}\times \\ & \frac{x}{(1-x)(1+x)}\frac{1}{(1-x)(1+x+x^2+x^3)}(1+x)\frac{1}{(1-x)(1+x+x^2)}\\ =& \frac{x}{(1-x{)}^4}\\ =& \left(\genfrac{}{}{0ex}{}{3}{0}\right)x+\left(\genfrac{}{}{0ex}{}{4}{1}\right)x^2+\left(\genfrac{}{}{0ex}{}{5}{2}\right)x^3+\cdots +\left(\genfrac{}{}{0ex}{}{n+2}{n-1}\right)x^n+\cdots \end{array}$$

答案为

$$\left(\genfrac{}{}{0ex}{}{n+2}{n-1}\right)=\left(\genfrac{}{}{0ex}{}{n+2}{3}\right)=\frac{(n+2)\times (n+1)\times n}{3\times 2\times 1}=\frac{(n+2)(n+1)n}{6}$$

#include <bits/stdc++.h>
using namespace std;

const int MOD = 1e9 + 7;
const int N = 1e5 + 50;

long long n;

long long fpow (long long x, long long p) {
    long long ans = 1;
    for (; p; p >>= 1, x = x * x % MOD)
        if (p & 1) ans = ans * x % MOD;
    return ans;
}

long long solve () {
    n %= MOD;
    return (n + 2) * (n + 1) % MOD * n % MOD;
}

int main () {
    cin >> n;
    cout << solve () * fpow (6, MOD - 2) % MOD << endl;
    return 0;
}

例题2：https://ac.nowcoder.com/acm/contest/24710/B

每一朵花的生成函数：

$$\begin{gathered} F_i(x)=1+x+x^2+\cdots +x^{f_i}=\frac{1-x^{f_i+1}}{1-x} \\ F(x)=\prod _{i=1}^n{F}_i(x)=\prod _{i=1}^n\frac{1-x^{f_i+1}}{1-x}=\frac{{\prod }_{i=1}^n(1-x^{f_i+1})}{(1-x{)}^n} \end{gathered}$$

记

$$\begin{gathered} A(x)=\prod _{i=1}^n(1-x^{f_i+1}) \\ B(x)=\frac{1}{(1-x{)}^n}=\sum _{i\ge 0}\left(\genfrac{}{}{0ex}{}{i+n-1}{i}\right)x^i \end{gathered}$$

那么答案为

$$\sum _i\left(\genfrac{}{}{0ex}{}{s-i+n-1}{s-i}\right)\times [x^i]A(x)=\sum _i\left(\genfrac{}{}{0ex}{}{s-i+n-1}{n-1}\right)\times [x^i]A[x]$$

#include <bits/stdc++.h>
using namespace std;

const int N = 2e6 + 50;
const int MOD = 1e9 + 7;

struct node { long long x, y; } a[N];
long long f[25], n, s, cnt, inv[25], fnv[25];

void dfs (int step, long long v, long long c) {
    if (step > n) return void(a[++ cnt] = {v, c});
    dfs (step + 1, v, c);
    dfs (step + 1, -v, c + f[step] + 1);
}

void init () {
    fnv[0] = fnv[1] = inv[0] = inv[1] = 1;
    for (int i = 2; i < n; i ++) {
        inv[i] = (MOD - MOD / i) * inv[MOD % i] % MOD;
        fnv[i] = fnv[i - 1] * inv[i] % MOD;
    }
}

long long calc (long long x) {
    if (x < 0) return 0;
    x %= MOD;
    long long res = 1;
    for (int i = 1; i < n; i ++) 
        res = res * (x + i) % MOD;
    return res * fnv[n - 1] % MOD;
}

int main () {
    cin >> n >> s;
    init ();
    for (int i = 1; i <= n; i ++) cin >> f[i];
    dfs (1, 1, 0);
    long long ans = 0;
    for (int i = 1; i <= cnt; i ++) 
        //cout << a[i].x << ' ' << a[i].y << endl,
        ans = (ans + a[i].x * calc(s - a[i].y)) % MOD;
    cout << (ans + MOD) % MOD << endl;
    return 0;
}

例题3：https://ac.nowcoder.com/acm/contest/24710/C

每一个糖果的生成函数：

$$\begin{gathered} F_i(x)=1+x+x^2+\cdots +x^{m_i}=\frac{1-x^{m_i+1}}{1-x} \\ F(x)=\prod _{i=1}^n{F}_i(x)=\prod _{i=1}^n\frac{1-x^{m_i+1}}{1-x}=\frac{{\prod }_{i=1}^n(1-x^{m_i+1})}{(1-x{)}^n} \end{gathered}$$

记

$$\begin{gathered} A(x)=\prod _{i=1}^n(1-x^{m_i+1}) \\ B(x)=\frac{1}{(1-x{)}^n}=\sum _{i\ge 0}\left(\genfrac{}{}{0ex}{}{i+n-1}{i}\right)x^i \end{gathered}$$

那么答案为

$$\begin{array}{rl}& \sum _{t=a}^b\sum _i\left(\genfrac{}{}{0ex}{}{t-i+n-1}{t-i}\right)\times [x^i]A(x)\\ =& \sum _{t=a}^b\sum _i\left(\genfrac{}{}{0ex}{}{t-i+n-1}{n-1}\right)\times [x^i]A(x)\\ =& \sum _i[x^i]A(x)\times \sum _{t=a}^b\left(\genfrac{}{}{0ex}{}{t-i+n-1}{n-1}\right)\\ =& \sum _i[x^i]A(x)\times \left(\sum _{t=a}^b\right(\genfrac{}{}{0ex}{}{t-i+n-1}{n-1})+(\genfrac{}{}{0ex}{}{a-i+n-1}{n})-(\genfrac{}{}{0ex}{}{a-i+n-1}{n}\left)\right)\\ =& \sum _i[x^i]A(x)\times \left(\right(\genfrac{}{}{0ex}{}{b-i+n}{n})-(\genfrac{}{}{0ex}{}{a-i+n-1}{n}\left)\right)\end{array}$$

#include <bits/stdc++.h>
using namespace std;

const int N = 1e7 + 50;
const int MOD = 2004;

long long n, a, b;
long long m[11];

long long C (long long x, long long y) {
    if (y < 0 || x < y) return 0;
    long long r = 1, p = MOD, ans = 1;
    for (int i = 1; i <= y; i ++) p *= i, r *= i;
    for (int i = 1; i <= y; i ++) ans = ans * (x - i + 1) % p;
    return ans / r % MOD;
}

int main () {
    cin >> n >> a >> b;
    for (int i = 0; i < n; i ++) cin >> m[i];
    long long ans = 0;
    for (int i = 0; i < 1 << n; i ++) {
        long long v = 1, c = 0;
        for (int j = 0; j < n; j ++) 
            if (i >> j & 1) v *= -1, c += m[j] + 1;
        ans = (ans + v * C(b - c + n, n) - v * C(a - c + n - 1, n)) % MOD;
    }
    cout << (ans + MOD) % MOD << endl;
    return 0;
}

递推关系

例子：斐波那契数列满足 $a_0=0$，$a_1=1$，$a_n=a_{n-1}+a_{n-2}(n\ge 2)$，求其常生成函数。

$$\begin{array}{rl}A(x)& =a_0+a_{1}x+a_2{x}^2+\cdots +a_n{x}^n+\cdots \\ & =0+x+(a_1+a_0)x^2+(a_2+a_1)x^3+\cdots +(a_{n-1}+a_{n-2})x^n+\cdots \\ & =x+(a_0{x}^2+a_1{x}^3+\cdots +a_{n-2}{x}^n+\cdots )+(a_1{x}^2+a_2{x}^3+\cdots +a_{n-1}{x}^n+\cdots )\\ & =x+x^{2}A(x)+xA(x)\end{array}$$

那么

$$\begin{array}{rl}& (1-x-x^2)A(x)=x\\ \Rightarrow & A(x)=\frac{x}{1-x-x^2}\\ \Rightarrow & A(x)=\frac{1}{\sqrt{5}}(\frac{1}{1-\frac{1+\sqrt{5}}{2}x}-\frac{1}{1-\frac{1-\sqrt{5}}{2}x})\\ \Rightarrow & A(x)=\sum _{i\ge 0}\frac{1}{\sqrt{5}}\left[\right(\frac{1+\sqrt{5}}{2}{)}^i-\left(\frac{1-\sqrt{5}}{2}{)}^i\right]x^i\\ \Rightarrow & a_n=\frac{1}{\sqrt{5}}\left[\right(\frac{1+\sqrt{5}}{2}{)}^n-\left(\frac{1-\sqrt{5}}{2}{)}^n\right]\end{array}$$
扩展：假设数列 { a n } \{a_n\} {an​} 存在递推关系 $a_n=c_1{a}_{n-1}+c_2{a}_{n-2}+\cdots +c_k{a}_{n-k}$，求它对应的常生成函数 $A(x)={\sum }_{n\ge 0}{a}_n{x}^n$。

Catalan 数

递推公式：
$$\begin{gathered} c_0=1，c_1=1 \\ {c}_n=c_0{c}_{n-1}+c_1{c}_{n-2}+\cdots +c_{n-1}{c}_0=\sum _{i=0}^{n-1}{c}_i{c}_{n-i-1} \end{gathered}$$
那么
$$\begin{array}{rl}C(x)& =\sum _{n\ge 0}{c}_n{x}^n\\ & =1+x+\sum _{n\ge 2}\sum _{i=0}^{n-1}{c}_i{c}_{n-i-1}{x}^n\\ & =1+x+\sum _{n\ge 1}\sum _{i=0}^n{c}_i{c}_{n-i}{x}^n+1\\ & =1+x+x(\sum _{n\ge 0}\sum _{i=0}^n{c}_i{c}_{n-i}{x}^n-c_0{c}_0)\\ & =1+x+x(C^2(x)-1)\end{array}$$
解出 $C(x)$ 的表达式
$$\begin{array}{rl}& C(x)=\frac{1\pm \sqrt{1-4x}}{2x}\\ \Rightarrow & C(x)=\frac{1-\sqrt{1-4x}}{2x}\end{array}$$
由于
$$(b+ax{)}^k=\sum _{n\ge 0}(\genfrac{}{}{0ex}{}{k}{n})b^{k-n}{a}^n{x}^n，k\in \mathbb{R}$$
则
$$\begin{array}{rl}C(x)& =\frac{1}{2x}(1-\sum _{n\ge 0}(\genfrac{}{}{0ex}{}{\frac{1}{2}}{n})(-4{)}^n{x}^n)\\ & =-\frac{1}{2}\sum _{n\ge 1}\left(\genfrac{}{}{0ex}{}{\frac{1}{2}}{n}\right)(-4{)}^n{x}^{n-1}\\ & =\sum _{n\ge 0}-\frac{1}{2}\left(\genfrac{}{}{0ex}{}{\frac{1}{2}}{n+1}\right)(-4{)}^{n+1}{x}^n\end{array}$$
通项 $c_n$ 为
$$\begin{array}{rl}{c}_n& =-\frac{1}{2}\left(\genfrac{}{}{0ex}{}{\frac{1}{2}}{n+1}\right)(-4{)}^{n+1}\\ & =-\frac{1}{2}\frac{\frac{1}{2}(\frac{1}{2}-1)(\frac{1}{2}-2)\cdots (\frac{1}{2}-n)}{(n+1)!}(-4{)}^{n+1}\\ & =\frac{1}{2}\frac{\frac{1}{2}(1-\frac{1}{2})(2-\frac{1}{2})\cdots (n-\frac{1}{2})}{(n+1)!}{4}^{n+1}\\ & =\frac{1\times 3\times 5\times \cdots \times (2n-1)}{(n+1)!}{2}^n\\ & =\frac{1\times 2\times 3\times \cdots \times (2n-1)\times (2n)}{(n+1)!\times n!}\\ & =\frac{1}{n+1}\left(\genfrac{}{}{0ex}{}{2n}{n}\right)\end{array}$$

指数生成函数

一个数列 { a n } \{a_n\} {an​} 对应的指数生成函数为 $A(x)={\sum }_{n\ge 0}{a}_n\frac{x^n}{n!}$。

例如，数列 $a_n=1$ ，其指数生成函数为：
$$A(x)=1+x+\frac{x^2}{2!}+\frac{x^3}{3!}+\cdots +\frac{x^n}{n!}+\cdots =e^x=exp(x)$$
例子：

• 有两种物体，其中取 $i$ 个第 1 种物体的方案数为 $a_i$ ，取 $j$ 个第 2 种物体的方案数为 $b_j$ ，求取 $k$ 个物体并排成一列的方案数。

$$\begin{gathered} c_k=\sum _{i=0}^k\left(\genfrac{}{}{0ex}{}{k}{i}\right)a_i{b}_{k-i}=\sum _{i=0}^k\frac{k!}{i!\times (k-i)!}{a}_i{b}_{k-i} \\ \Rightarrow \frac{c_k}{k!}=\sum _{i=0}^k\frac{a_i}{i!}\times \frac{b_{k-i}}{(k-i)!}。 \end{gathered}$$
记 $A(x)$ 和 $B(x)$ 分别为
$$A(x)=a_0+a_1\frac{x}{1!}+a_2\frac{x}{2!}+\cdots +a_n\frac{x}{n!}+\cdots$$
$$B(x)=b_0+b_1\frac{x}{1!}+b_2\frac{x}{2!}+\cdots +b_n\frac{x}{n!}+\cdots$$
$$C(x)=\sum _{k\ge 0}{c}_k\frac{x^k}{k!}=\sum _{k\ge 0}\sum _{i=0}^k\frac{a_i}{i!}\frac{b_{k-i}}{(k-i)!}{x}^k=A(x)\times B(x)$$

答案为 $[x^k]C(x)\times k!$。

• 用数字 1，2，3，4 作 6 位数字，每个数字在 6 位数中出现的次数不得大于2，求可以作出多少个不同的 6 位数。

每一个数字的生成函数为
$$\begin{gathered} F(x)=1+x+\frac{x^2}{2!} \\ G(x)=F^4(x) \end{gathered}$$

答案为 $[x^6]G(x)\times 6!$。

定理：设 S = { a 1 , a 2 , … , a k } S=\{a_1, a_2, \dots ,a_k\} S={a1​,a2​,…,ak​}，且 $a_i$ 可以取的次数的集合为 $M_i$ ，记 $F_i(x)={\sum }_{u\in M_i}\frac{x^u}{u!}$，则从 $S$ 中取 $n$ 个元素排成一列的方案数 $g(n)$ 的常生成函数 $G(x)={\sum }_{i\ge 0}g(i)\frac{x^i}{i!}$，满足:
$$G(x)=F_1(x)F_2(x)\cdots F_k(x)$$

公式（定义）：
$$\begin{gathered} exp(x)=1+x+\frac{x^2}{2!}+\cdots =\sum _{n\ge 0}\frac{x^n}{n!} \\ exp(ax)=1+ax+a^2\frac{x^2}{2!}+\cdots =\sum _{n\ge 0}{a}^n\frac{x^n}{n!} \\ exp(f(x))=1+f(x)+\frac{f^2(x)}{2!}+\cdots =\sum _{n\ge 0}\frac{f^n(x)}{n!} \end{gathered}$$

例题4：https://ac.nowcoder.com/acm/contest/24710/D

定义 $F(x)，G(x)$：
$$\begin{gathered} F(x)=1+\frac{x^2}{2!}+\frac{x^4}{4!}+\cdots =\frac{exp(x)+exp(-x)}{2} \\ G(x)=1+x+\frac{x^2}{2!}+\cdots =\sum _{n\ge 0}\frac{x^n}{n!}=exp(x) \end{gathered}$$
则方案数的生成函数 $H(x)$ 为：
$$\begin{array}{rl}H(x)& =F^2(x)G^2(x)\\ & =(\frac{exp(x)+exp(-x)}{2}{)}^{2}exp(x{)}^2\\ & =\frac{exp(x{)}^2+2+exp(-x{)}^2}{4}exp(2x)\\ & =\frac{exp(4x)+2exp(2x)+1}{4}\\ & =\sum _{n\ge 0}\frac{1}{4}(4^n+2\cdot 2^n)\frac{x^n}{n!}+\frac{1}{4}\end{array}$$
答案为
$$4^{n-1}+2^{n-1}\mathrm{m}\mathrm{o}\mathrm{d}10007$$

#include <bits/stdc++.h>
#define endl '\n'
using namespace std;

const int MOD = 10007;

int T, n;

int fpow (int x, int p) {
    int ans = 1;
    for (; p ; p >>= 1, x = (long long)x * x % MOD)
        if (p & 1) ans = (long long)ans * x % MOD;
    return ans;
}

int main () {
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin >> T;
    while (T --) {
        cin >> n;
        cout << (fpow (4, n - 1) + fpow(2, n - 1)) % MOD << endl; 
    }
    return 0;
}

例题5：https://ac.nowcoder.com/acm/contest/24710/E

每一次操作的贡献为：
$$result+=a_1{a}_2\cdots a_x\cdots a_n-a_1{a}_2\cdots (a_x-1)\cdots a_n$$
由于中间状态都会消去，总的贡献为：
$$result=a_1{a}_2\cdots a_n-(a_1-x_1)(a_2-x_2)\cdots (a_n-x_n)$$
$$\sum _{i=0}^n{x}_i=k，x_i\ge 0$$
问题变为求 $(a_1-x_1)(a_2-x_2)\cdots (a_n-x_n)$ 的期望：
$$\begin{array}{rl}& \mathbb{E}[(a_1-x_1)(a_2-x_2)\cdots (a_n-x_n)]\\ =& \frac{\sum _{x_1+x_2+\cdots +x_n=k}(a_1-x_1)(a_2-x_2)\cdots (a_n-x_n)\times \left(\genfrac{}{}{0ex}{}{k}{x_1,x_2,\cdots ,x_n}\right)}{n^k}\\ =& \frac{\sum _{x_1+x_2+\cdots +x_n=k}\frac{(a_1-x_1)}{x_1!}\frac{(a_2-x_2)}{x_2!}\cdots \frac{(a_n-x_n)}{x_n!}\times k!}{n^k}\\ =& \frac{k!}{n^k}\sum _{x_1+x_2+\cdots +x_n=k}\frac{(a_1-x_1)}{x_1!}\frac{(a_2-x_2)}{x_2!}\cdots \frac{(a_n-x_n)}{x_n!}\end{array}$$
构造每一个 $a_i$ 对应的生成函数
$$\begin{array}{rl}{F}_i(x)& =\sum _{j\ge 0}\frac{(a_i-j)}{j!}{x}^j\\ & =\sum _{j\ge 0}\frac{a_i}{j!}{x}^j-\sum _{j\ge 0}\frac{j}{j!}{x}^j\\ & =\sum _{j\ge 0}\frac{a_i}{j!}{x}^j-x\sum _{j\ge 0}\frac{1}{j!}{x}^j\\ & =a_{i}exp(x)-xexp(x)\end{array}$$
则 $F(x)$ 为
$$\begin{array}{rl}F(x)& =\prod _{i=1}^n{F}_i(x)\\ & =\sum _{k\ge 0}\sum _{x_1+x_2+\cdots +x_n=k}\frac{(a_1-x_1)}{x_1!}\frac{(a_2-x_2)}{x_2!}\cdots \frac{(a_n-x_n)}{x_n!}{x}^k\\ & =exp(nx)\times \prod _{i=1}^n(a_i-x)(记A(x)=\prod _{i=1}^n(a_i-x))\\ & =\sum _{i\ge 0}{n}^i\frac{x^i}{i!}\times A(x)\\ & =\sum _{k\ge 0}\sum _{i=0}^k[x^i]A(x)\times \frac{n^{k-i}}{(k-i)!}{x}^k\end{array}$$
那么答案 $[x^k]F(x)$ 为：
$$\sum _{i=0}^n[x^i]A(x)\times \frac{n^{k-i}}{(k-i)!}{x}^k$$

#include <bits/stdc++.h>
using namespace std;

const int MOD = 1e9 + 7;
const int N = 5050;

int n, k, a[N], f[N], g[N];
int b1[N], b2[N], c[N];

int fpow (int x, int p) {
    int res = 1;
    for (; p; p >>= 1, x = (long long) x * x % MOD)
        if (p & 1) res = (long long)res * x % MOD;
    return res;
}

void mul(int* b1, int* b2) {
    memset (c, 0, sizeof (c));
    for (int i = 0; i < 2; i ++) 
        for (int j = 0; j <= n; j ++) 
            c[i + j] = (c[i + j] + (long long)b1[j] * b2[i] % MOD) % MOD;
    for (int i = 0; i <= n; i ++) 
        b1[i] = c[i];
}

int main () {
    cin >> n >> k;
    for (int i = 1; i <= n; i ++) cin >> a[i];
    g[0] = f[0] = 1, g[1] = fpow (n, MOD - 2);
    for (int i = 1; i <= n; i ++) 
        f[i] = (long long)(k - i + 1) * f[i - 1] % MOD;
    for (int i = 2; i <= n; i ++) 
        g[i] = (long long)g[i - 1] * g[1] % MOD;
    b1[0] = a[1], b1[1] = -1;
    for (int i = 2; i <= n; i ++) {
        b2[0] = a[i], b2[1] = -1;
        mul (b1, b2);
    }
    int ans = 0, sum = 1;
    for (int i = 0; i <= n; i ++) 
        ans = (ans + (long long)b1[i] * f[i] % MOD * g[i] % MOD) % MOD;
    for (int i = 1; i <= n; i ++)
        sum = (long long) sum * a[i] % MOD;
    cout << ((long long)sum - ans + MOD) % MOD << endl;
    return 0;
}