第八章 生成函数1

第八章 生成函数（数学）

8.1 笔记

• 形式幂级数：$A(x)=\sum _{i\ge 0}{a}_i{x}^i$

  多项式 $A^{\prime }(x)=\sum _{i\ge 0}^n{a}_i{x}^i$

  • 运算：设 $A(x)=\sum _{i\ge 0}{a}_i{x}^i,B(x)=\sum _{i\ge 0}{b}_i{x}^i$

    • 加减法：$A(x)\pm B(x)=\sum _{i\ge 0}(a_i\pm b_i)x^i$
    • 乘法：$A(x)B(x)=\sum _{k\ge 0}\left(\sum _{i=0}^k{a}_i{b}_{k-i}\right)x^k$

    形式幂级数在定义的 $+,-,\times$ 下形成环

  • 记 $A(x)$ 的 $x^n$ 项的系数为 $[x^n]A(x)$

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

  • 不定方程的解的数量：求不定方程 $x_1+x_2+...+x_k=n$ 的解的数量，其中 $x_i\in Z,l_i\le x_i\le r_i(1\le i\le k)$

    解：设 $F_i(x)=\sum _{j=l_i}^{r_i}{x}_j$，则 $F(x)=\prod _{i=1}^k{F}_i(x)$，故 $ans=[x^n]F(x)$

    $例题$：求 $x_1+2x_2+3x_3=100$ 的正整数解的个数

    解：设 $F_1=\sum _{i\ge 1}{x}^i$，$F_2=\sum _{i\ge 1,2\mid i}{x}^i$，$F_3=\sum _{i\ge 1,3\mid i}{x}^i$

    于是答案为 $[x^{100}]F(x)=[x^{100}]F_1(x)F_2(x)F_3(x)$

• 形式幂级数的逆元：$A(x)B(x)=1$

  • 存在条件：$[x^0]A(x)\ne 0$

  • 暴力计算：递推

  • 常见逆元：

    1. $1+x+x^2+...=\frac{1}{1-x}$

    2. $1+ax+a^2{x}^2+...=\frac{1}{1-ax}$

    3. $C_{k-1}^0+C_k^{1}x+C_{k+1}^2{x}^2+C_{k+2}^3{x}^3+...=\sum _{i\ge 0}{C}_{i+k-1}^i{x}^i=\sum _{i\ge 0}{C}_{i+k-1}^{k-1}{x}^i=\frac{1}{(1-x{)}^k}$

       于是 $[x^i]\frac{1}{(1-x{)}^k}=C_{k+i-1}^i$

    4. $1+f(x)+f^2(x)+f^3(x)+...=\frac{1}{1-f(x)}$

• 卢卡斯定理（Lucas）：对于素数 $p$​，有
  $$C_n^m\%p=C_{n/p}^{m/p}\cdot C_{n\%p}^{n\%p}\%p$$

  • 当 $n,m$ 较大时适用
  • 对于 $n\%p,m\%p$ 可直接求解

ll inv(ll i){
    if(i == 1) return 1;
    return (mod - mod / i) * inv(mod % i) % mod;
}
ll C(ll n, ll m){
    m = min(m, n - m);
    if(m == 0) return 1;
    return C(n - 1, m - 1) * n % mod * inv(m) % mod;
}
ll Lucas(ll n, ll m){
    if (m == 0) return 1;
    return (C(n % mod, m % mod) * Lucas(n / mod, m / mod) + mod) % mod;
}

8.2 习题

第八章习题链接

A. 背包

• 思路：八种物品，每种有一个生成函数，
  $$\begin{array}{rl}{F}_1(x)& =1+x=\frac{1-x^2}{1-x}\\ F_2(x)& =1+x+x^2=\frac{1-x^3}{1-x}\\ F_3(x)& =1+x+x^2+x^3=\frac{1-x^4}{1-x}\\ F_4(x)& =1+x^2+x^4+x^6+...=\frac{1}{1-x^2}\\ F_5(x)& =x+x^3+x^5+x^7+...=\frac{x}{1-x^2}\\ F_6(x)& =1+x^4+x^8+x^{12}+...=\frac{1}{1-x^4}\\ F_7(x)& =1+x=\frac{1-x^2}{1-x}\\ F_8(x)& =1+x^3+x^6+x^9+...=\frac{1}{1-x^3}\\ \therefore F(x)& =\prod _{i=1}^8{F}_i(x)=\frac{x}{(1-x{)}^4}=x\sum _{i\ge 0}{C}_{i+3}^i{x}^i=\sum _{i\ge 0}{C}_{i+3}^3{x}^{i+1}=\sum _{i\ge 1}{C}_{i+2}^3{x}^i\end{array}$$

  于是 $[x^n]F(x)=C_{n+2}^3$

#include<bits/stdc++.h>
using namespace std;
#define rep(i, a, b) for(int i = (a); i <= (b); i++)
#define per(i, a, b) for(int i = (a); i >= (b); i--)
#define ll long long
#define db double
#define VI vector<int>
#define PII pair<int, int>
const db Pi = 3.141592653589793;
const int INF = 0x7fffffff;
const int N = 1e5 + 5;
const db eps = 1e-10;
const int mod = 1e9 + 7;
ll n;
ll inv(ll i){
    if(i == 1) return 1;
    return (mod - mod / i) * inv(mod % i) % mod;
}
int main(){
    cin >> n;
    n %= mod;
    cout << (n * (n + 1) % mod * (n + 2) % mod * inv(6) % mod) << endl;
}
/*
5
*/

B. Devu and Flowers

• 来源：CF451 E

• 题目：$n$ 种花，每种分别有 $f_1,f_2,...,f_n$ 朵，求取 $s$ 朵花的方案数。（$1\le n\le 20,0\le f_i\le 10^{12},0\le s\le 10^{14}$）

• 思路：

  • $F_i(x)=1+x+x^2+...+x^{f_i(x)}=\frac{1-x^{f_i(x)+1}}{1-x}$

  • $F(x)=\prod _{i=1}^n{F}_i(x)=\frac{\prod _{i=1}^n(1-x^{f_i(x)+1})}{(1-x{)}^n}$

  • 记 $A(x)=\prod _{i=1}^n\left(1-x^{f_i(x)+1}\right)$，故
    $$[x^s]F(x)=\sum _{i=1}^n[x^i]A(x)\cdot [x^{s-i}]\frac{1}{(1-x{)}^n}=\sum _{i=1}^n[x^i]A(x)\cdot C_{s-i+n-1}^{n-1}$$

  • 由于 $n$ 较小，故可用二进制来求出 $[x^i]A(x)$

#include<bits/stdc++.h>
using namespace std;
#define rep(i, a, b) for(int i = (a); i <= (b); i++)
#define per(i, a, b) for(int i = (a); i >= (b); i--)
#define ll long long
#define db double
#define VI vector<int>
#define PII pair<int, int>
const db Pi = 3.141592653589793;
const int INF = 0x7fffffff;
const int N = 2e1 + 5;
const db eps = 1e-10;
const int mod = 1e9 + 7;
ll n, s, f[N], ans = 0;
ll inv(ll i){
    if(i == 1) return 1;
    return (mod - mod / i) * inv(mod % i) % mod;
}
ll C(ll n, ll m){
    m = min(m, n - m);
    if(m == 0) return 1;
    return C(n - 1, m - 1) * n % mod * inv(m) % mod;
}
ll Lucas(ll n, ll m){
    if (m == 0) return 1;
    return (C(n % mod, m % mod) * Lucas(n / mod, m / mod) + mod) % mod;
}
int main(){
    cin >> n >> s;
    rep(i, 1, n) cin >> f[i];
    rep(t, 0, (1 << n) - 1){
        ll sign = 1, cnt = 0;
        rep(i, 1, n){
            if(t & (1 << (i - 1))){
                cnt += f[i] + 1;
                sign = -sign;
            }
        }
        if(cnt > s) continue;
        ans += sign * Lucas(s + n - 1 - cnt, n - 1) % mod;
        ans %= mod;
    }
    cout << (ans + mod) % mod << endl;
}
/*
2 3
1 3  

2 4
2 2

3 5
1 3 2
*/