计数好帮手生成函数

Mongo_w

于 2021-11-26 21:29:09 发布

阅读量220

点赞数

分类专栏：笔记文章标签： c++ 算法机器学习

本文链接：https://blog.csdn.net/weixin_45916935/article/details/121569098

版权

笔记专栏收录该内容

15 篇文章 0 订阅

订阅专栏

OGF

记数列 ${a_i\}$ 的生成函数为 $\sum\limits_{k=0}^{\infty} a_kx^k$ .

常用的生成函数有

$\frac{1}{1-x}=1+x+x^2+\ldots =\sum\limits_{k=0}^{\infty}x_k$ .
$\frac{1}{1-x^n}=1+x^n+x^{2n}+\ldots=\sum\limits_{k=0}^{\infty}x^{kn}$ .
$\frac{1}{1-ax}=1+ax+a^2x^2+\ldots=\sum\limits_{k=0}^{\infty}a^kx^k$ .
$\frac{1}{(1-x)^n}=\sum\limits_{k=0}^{\infty}\binom{n+k-1}{k-1}x^k$ .

前两个等价于等比数列的求和式，根据 $x < 1$ 时的收敛性写出的封闭形式.

关于 $4$ 式提供两种推导:

一

使用广义二项式定理： $(1+x)^n=\sum\limits_{k=0}^{\infty}\binom{n}{k}x^k$ .

于是 $(1-x)^{-n}=\sum\limits_{k=0}^{\infty}\binom{n+k-1}{k}x^k$ .

同理可得 $(1+x)^{-n}=\sum\limits_{k=0}^{\infty}(-1)^k\binom{n+k-1}{k}x_k$ .

二

这相当于求解以下问题：

给出 $m$ ，问用 $n$ 个非负整数求和得到 $m$ 的方案数 $a_m$ .

根据插板法可知 $a_m=\binom{n+m-1}{n-1}=\binom{n+m-1}{m}$ .

用生成函数即写成 $\frac{1}{(1-x)^m}$ .

这同样可以通过麦克劳林展开得到，这里就不过多阐述.

一些很 $\text{simple}$ 的例子

一

确定苹果、香蕉、橘子和梨的 $n$ 组合的个数，其中在每个组合中苹果的个数是偶数，香蕉的个数是奇数，橘子的个数在 $0$ 和 $4$ 之间，而且至少要有一个梨。

直接列出生成函数 $\frac{1}{1-x^2}\times \frac{x}{1-x^2}\times \frac{1-x^5}{1-x}\times \frac{x}{1-x}$ ，第 $n$ 项系数就是答案.

二

给出递推公式 $a_{n+1}=8a_{n}+10^{n}$ ，其中 $a_1=9$ ，求 ${a_n\}$ 的通项公式.

记 $G(x)=\sum\limits_{k=0}^{\infty}a_kx^k$ ，特别的记 $a_0=1$ .

由递推式推出：

$(1-8x)G(x)=\sum\limits_{k=1}^{\infty}10^{k-1}x^k+1=x\sum\limits_{k=0}^{\infty}10^kx^k+1=\frac{x}{1-10x}+1=\frac{1-9x}{1-10x}$ .

进而得到：

$G(x)=\frac{1-9x}{(1-10x)(1-8x)}=\frac{1}{2}(\frac{1}{1-10x}+\frac{1}{1-8x})$ .

所以 $a_n=\frac{1}{2}(10^n+8^n)$ .

高级一点的应用

一. Bzoj 3028 食物

明明这次又要出去旅游了，和上次不同的是，他这次要去宇宙探险！我们暂且不讨论他有多么NC，他又幻想了他应

该带一些什么东西。理所当然的，你当然要帮他计算携带N件物品的方案数。他这次又准备带一些受欢迎的食物，

如：蜜桃多啦，鸡块啦，承德汉堡等等当然，他又有一些稀奇古怪的限制：每种食物的限制如下：

承德汉堡：偶数个

可乐：0个或1个

鸡腿：0个，1个或2个

蜜桃多：奇数个

鸡块：4的倍数个

包子：0个，1个，2个或3个

土豆片炒肉：不超过一个。

面包：3的倍数个

注意，这里我们懒得考虑明明对于带的食物该怎么搭配着吃，也认为每种食物都是以‘个’为单位（反正是幻想嘛

），只要总数加起来是N就算一种方案。因此，对于给出的N,你需要计算出方案数，并对10007取模。

直接列出生成函数:

$\frac{1}{1-x^2}\times\frac{1-x^2}{1-x}\times\frac{1-x^3}{1-x}\times\frac{x}{1-x^2}\times\frac{1}{1-x^4}\times \frac{1-x^4}{1-x}\times\frac{1-x^2}{1-x}\times\frac{1}{1-x^3}$ .

整理一下得到：

$\frac{x}{(1-x)^4}$ .

即 $G(x)=x\sum\limits_{k=0}^{\infty}\binom{k+3}{3}x^k$ .

所以 $n$ 项的系数就是 $\binom{n+2}{3}$ .

二. [CEOI2004] Sweets

John 得到了 $n$ 罐糖果。不同的糖果罐，糖果的种类不同（即同一个糖果罐里的糖果种类是相同的，不同的糖果罐里的糖果的种类是不同的）。第 $i$ 个糖果罐里有 $m_i$ 个糖果。John 决定吃掉一些糖果，他想吃掉至少 $a$ 个糖果，但不超过 $b$ 个。问题是 John 无法确定吃多少个糖果和每种糖果各吃几个。有多少种方法可以做这件事呢？

单个列出来是 $1+x+\ldots+x^{m_i}=\frac{1-x^{m_i+1}}{1-x}$ .

所以总的来说是 $\frac{(1-x^{m_1+1})\cdots(1-x^{m_n+1})}{(1-x)^n}=\frac{1}{(1-x)^n}F(x)$ .

upd 11.25 这题难在处理模数…

关于 $a$ 和 $b$ 的限制转化成 $x^m$ 的前缀模数一起处理.

考虑 $F (x)$ 中的 $ax^b$ 对答案的贡献：
$a\sum\limits_{i=0}^{lmt-b}\binom{i+n-1}{n-1}=a\binom{lmt + n - b}{n}$ .

于是剩下怎么处理这个模数.

展开写成 $\frac{(n+lmt-b)(n+lmt-b-1)\cdots}{n!}$ .

上下分开处理.

对于上面，注意到总共只有 $n$ 个，记为 $x$ .

我们最终写成 $x=2004\times n!\times k_2+r\ , \ \frac{x}{n!}\equiv ans(\mod2004)$ ， $k_2$ 为整数.

则：

$\frac{x}{n!}=ans+2004\times k_1$ .

$2004\times n!\times k_2+r=ans\times n!+2004\times k_1\times n!$ .

移项：

$r=n!\times (ans+2004\times k_1-2004\times k_2)$ .

$\frac{r}{n!}=k'+ans$ .

即 $\frac{r}{n!}\equiv ans(\mod 2004)$ .

EGF

显然我们 $\text{OGF}$ 主要是拿来解决组合问题的，那如果有序的话又怎么做呢.

我们引入指数生成函数(EGF)，定义数列 ${h_n\}$ 的指数生成函数为 $\hat{G}(x)=$ $\sum\limits_{k=0}^\infty h_k\frac{x^k}{k!}$ .

常用的形式有：

$e^x=1+x+\frac{x^2}{2!}+\cdots=\sum\limits_{k=0}^\infty \frac{x^k}{k!}$ .
$e^{ax}=1+ax+\frac{(ax)^2}{2!}+\cdots=\sum\limits_{k=0}^\infty a^k\frac{x^k}{k!}$ .
$\frac{e^x+e^{-x}}{2}=1+\frac{x^2}{2!}+\frac{x^4}{4!}+\cdots=\sum\limits_{k=0}^\infty \frac{x^{2k}}{(2k)!}$ .
$\frac{e^x-e^{-x}}{2}=x+\frac{x^3}{3!}+\frac{x^5}{5!}+\cdots=\sum\limits_{k=0}^\infty \frac{x^{2k+1}}{(2k+1)!}$ .
$\ln(x+1) = x-\frac{x^2}{2}+\frac{x^3}{3!}-\cdots=\sum\limits_{k=1}^\infty (-1)^{k+1}\frac{x^k}{k}$ .
$\sin x=x-\frac{x^3}{3!}+\frac{x^5}{5!}-\cdots=\sum\limits_{k=0}^\infty (-1)^{2k+1}\frac{x^{2k+1}}{(2k+1)!}$ .
$\cos x= 1-\frac{x^2}{2!}+\frac{x^4}{4!}-\cdots=\sum\limits_{k=0}^\infty (-1)^{2k}\frac{x^{2k}}{(2k)!}$ .
$\arcsin x = x + \frac{1}{2}\times \frac{x^3}{3}+\frac{1\times 3}{2\times 4}\times \frac{x^5}{5}+\frac{1\times 3\times 5}{2\times 4\times 6}\times \frac{x^7}{7}+\cdots$ .

这玩意相乘的话可以保证局部组合，整体排列，具体给出以下例子.

很 $\text{simple}$ 的例子

一

求 $h_k=P_n^k$ 的指数生成函数.

直接列出来：

$\sum\limits_{k=0}^nh_k\frac{x^k}{k!}=\sum\limits_{k=0}^n\frac{n!}{(n-k!)k!}x^k=\sum\limits_{k=0}^n\binom{n}{n-k}x^k=(1+x)^n$ .

二

求多重集合 $\{n_1·a_1\ , \ n_2·a_2\ , \ \ldots\ , n_k·a_k\}$ 的 $n$ 排列数 $S_n$ ，其中 $n=n_1+n_2+\ldots+n_k$ .

由组合数学我们可知 $S_n=\frac{n!}{n_1!n_2!\cdots n_k!}$ .

设 $\hat{f}_{n_1}(x)=\sum\limits_{k=0}^{n_1}\frac{x^k}{k!}$ .

记 $\hat{G}(x)=\prod\limits_{i=1}^k \hat{f}_{n_i}(x)$ .

于是我们知道 $x^m$ 项的系数是：

$\sum\frac{m!}{m_1!m_2!\ldots m_k!}$ .

其中 $0\le m_1\le n_1\ , \ 0\le m_2\le n_2\ , \ \ldots\ , \ 0\le m_k\le n_k$ 且 $m_1+m_2+\cdots+m_k=m$ .

注意到 $n$ 只有由唯一一组 $(n_1\ , \ n_2\ , \ \ldots \ , \ n_k)$ 可以得到.

所以 $x^n$ 的系数就是我们要求的值.

提高一点的应用

Poj3734 Blocks

用红黄蓝绿四种颜色染 $n$ 个格子，要求红色和绿色必须是偶数个，求方案数，对 $10007$ 取模.

直接写出指数生成函数 $\hat{G}(x)=(\frac{e^x+e^{-x}}{2})^2e^{2x}=\frac{e^{4x}+2e^{2x}+1}{4}=\frac{1}{4}+\frac{1}{4}\sum\limits_{k=0}^{\infty}(4^{k}+2\times 2^k)\frac{x^k}{k!}$ .

答案就是 $\frac{4^n+2^{n+1}}{4}$ .

[集训队作业2013]城市规划

Mongo_w

关注

0
点赞
踩
1

收藏

觉得还不错? 一键收藏
0
评论
计数好帮手生成函数

计数好帮手生成函数OGF记数列 {ai}\{a_i\}{ai} 的生成函数为 ∑k=0∞akxk\sum\limits_{k=0}^{\infty} a_kx^kk=0∑∞akxk .常用的生成函数有11−x=1+x+x2+…=∑k=0∞xk\frac{1}{1-x}=1+x+x^2+\ldots =\sum\limits_{k=0}^{\infty}x_k1−x1=1+x+x2+…=k=0∑∞xk .11−xn=1+xn+x2n+…=∑k=0∞xkn\frac{1}{1-x^n}
复制链接

扫一扫