【生成函数】五边形数定理与整数划分问题详解

Part 0. 前置知识

• 简单的公式推导
• 生成函数

大量数学公式警告

Part 1. 什么是五边形数

五边形数

看一张图吧：

XP 系统带的画图真的挺好用！

我们不难得出五边形数的数列：

P = { 1 , 5 , 12 , 22 , … } P = \{1, 5, 12, 22,\ldots\} P={1,5,12,22,…}

再找一下规律，我们就可以得到五边形数的通项公式：

$$P_n=\frac{n(3n-1)}{2}$$

似乎现在并没有什么用。。。

广义五边形数

也就是当 $n$ 可以为负数的时候的五边形数序列，它的通项公式为：

$$P_n=\frac{n(3n\pm 1)}{2}$$

Part 2. 五边形数定理

定义欧拉函数（注意不要把它看成数论里面的欧拉函数）为：

$$\varphi (n)=\prod _{i=1}^{\infty }(1-x^i)$$

则五边形数定理就是：

$$\varphi (n)=\prod _{i=-\infty }^{\infty }(-1{)}^i{x}^{\frac{i(3i-1)}{2}}=1+\prod _{i=0}^{\infty }(-1{)}^i{x}^{\frac{i(3i\pm 1)}{2}}$$

证明等一会给出。

Part 3. Ferrers 图

Ferrers 图是一个由 $k$ 行的点阵构成的方格图，要求上一行的格子数目不超过它下面一行的格子的数目，其中总和为 $n$ 的数量。

举个例子：（$n=20$ 时 的一个图，我用 @ 来表示一个格子）

@@@@@@
@@@@@
@@@@
@@@
@@

我们记这时的第 $k$ 行的格子数量为 $m$， $s$ 为第一行最右边的 @ 右上角的对角线格子数量。

定义一种变换：

• 当 $m>s$ 时， 把最右边的对角线上的格子扔到新的一行。
• 当 $m\le s$ 时，把最后一行的所有格子分给从第一行开始的每行上，每行分到一个格子。

举个例子：（用 $ 表示被移走的格子）

@@@@@@@@        @@@@@@@@$
@@@@@@@         @@@@@@@$
@@@@@    <=>    @@@@@
@@@             @@@
$$            

不难看出这个操作是显然可逆的。

这个操作还有一些性质：

我们对一个 Ferrers 图进行连续的两次变换后，它会变回它本身。

如果我们仅进行一次变换，那么行数的奇偶性可以改变。

Part 4. 五边形数定理的证明

我们将 $\varphi (n)$ 的定义式展开之后可以发现，第 $n$ 项的系数应该是将 $n$ 划分成偶数个互不相同的正整数的答案减掉将 $n$ 划分成奇数个互不相同的正整数方案，每一项的系数应该都是 $0$。但实际操作后发现有些项前面是有系数的。

那么我们考虑仔细研究一下 Ferrers 图。

---

当 $m=s+1$ 时，且底层的最右边的元素与对角线相遇，如下：

@@@@@@    @@@@@
@@@@@  => @@@@
@@@@      @@@
          @@@

发现此时无法操作回去了。这是一个不合法的状态。

令 $t=1-m$，可以得到 $n=(1-t)+(2-t)+(3-t)+\cdots +(-2t)=\frac{t(3t-1)}{2}$，这一项对答案的贡献为 $(-1{)}^t$。

---

当 $m=s$，且底层的最右边的元素与对角线相遇，如下：

@@@@@    @@@@@@
@@@@  => @@@@@
@@@      @

这个显然是不合法的，令 $t=m$，可以得到 $n=t+(t+1)+(t+2)+\cdots +2t=\frac{t(3t+1)}{2}$，对第 $n$ 项的贡献为 $(-1{)}^t$。

---

这样一来，当 $n=\frac{k(3k\pm 1)}{2}$ 时，即 $n$ 是一个广义五边形数时，$n$ 的奇偶拆分就在抵消之后留下一项。所以五边形定理成立。

Part 5. 整数划分问题

问题概要

将一个整数 $n$ 划分成任意个可以相等的整数，求方案数。

分析

我们不难写出一个 $O(n^2)$ 的 DP 来解决整数划分问题。但当 $n$ 超过 $10^4$ 之后就跑不动了。

考虑高效解法 （利用五边形数定理）

定义函数 $P(i)$ 为将 $n$ 拆分成若干个可以相等的整数的方案数。

考虑 $P(i)$ 的生成函数 $F(x)$：

$$\begin{array}{rl}F(x)& =\sum _{i=1}^{\infty }P(i)x^i\\ & =\prod _{i=1}^{\infty }(x^0+x^i+x^{2i}+\cdots )\\ & =\prod _{i=1}^{\infty }\frac{1}{1-x^i}\end{array}$$

再看一下上面提到的欧拉函数：

$$\varphi (x)=\prod _{i=1}^{\infty }(1-x^i)$$

显然可以得到：$F(x)\cdot \varphi (x)=1$。

然后根据五边形数定理将 $\varphi (x)$ 代入：

$$\left[\sum _{i=1}^{\infty }P(i)x^i\right]\cdot \left(\sum _{i=0}^{\infty }1+\prod _{i=0}^{\infty }(-1{)}^i{x}^{\frac{i(3i\pm 1)}{2}}\right)=1$$

直接暴力展开可以得到：

$$P(n)-P(n-1)-P(n-2)+P(n-5)+P(n-7)-\cdots =0$$

不难发现减去的每一项都是广义五边形数。于是直接暴力计算即可。

由于五边形数 $p_n=\frac{n(3n\pm 1)}{2}$ 的增长速度为 $O(n^2)$。所以最多枚举不超过 $O(\sqrt{n})$ 项五边形数就可以退出循环，故时间复杂度为 $O(n\sqrt{n})$。

Part 6. 例题

HDU 4651

题目大意

求将正整数 $n$ 划分成多个正整数的和的方案数。

分析

就是一个模板了。。。好好看一看上面的就可以了。

HDU 4658

题目大意

将一个正整数 $n$ 划分为多个正整数的和的方案数，但每个数的使用次数不能够大于等于 $k$ 次。

分析

记这种方式下的划分数为 $P^{\prime }(n)$。先给出生成函数 $G(x)=\prod _{i=1}^{\infty }(1+x^i+x^{2i}+\cdots +x^{(k-1)i})$。

尝试化简：

$$\begin{array}{rl}G(x)& =\prod _{i=1}^{\infty }\frac{1+x^i+x^i+x^i+\cdots }{1+x^{ki}+x^{2ki}+x^{3ki}+\cdots }\\ & =\prod _{i=1}^{\infty }\frac{\frac{1}{1-x^i}}{\frac{1}{1-x^{ki}}}\\ & =\prod _{i=1}^{\infty }\frac{1-x^{ki}}{1-x^i}\\ & =\frac{\varphi (x^k)}{\varphi (x)}\\ & =\varphi (x^k)\cdot F(x)\end{array}$$

然后暴力展开 $\varphi (x^k)\cdot F(x)$ 得到：

$$\sum _{i=1}^{\infty }{P}^{\prime }(i)x^i=(1-x^k-x^{2k}+x^{5k}+\cdots )[1+P(1)x+P(2)x^2+\cdots ]$$

然后可以得到 $P^{\prime }(i)=P(i)-P(i-k)-P(i-2k)+P(i-5k)+\cdots$

于是就可以在 $O(\sqrt{n})$ 的时间内回答询问了，总时间复杂度为 $O(n\sqrt{n}+T\sqrt{n})$。

参考代码

HDU 4651

#include <cstdio>
#include <algorithm>
using namespace std;

typedef long long ll;
const int Maxn = 1e5;
const ll Mod = 1000000007;

ll f[Maxn * 2 + 5];
ll p[Maxn + 5];
void Init() {
	for(int i = -Maxn; i <= Maxn; i++)
		f[i + Maxn] = 1LL * i * (i * 3 - 1) / 2;
	//计算五边形数
	p[0] = 1;
	for(int i = 1; i <= Maxn; i++)
		for(int j = 1; j <= i; j++) {
			if(f[j + Maxn] <= i) {
				if(j & 1) p[i] = (p[i] + p[i - f[j + Maxn]]) % Mod;
				else p[i] = (p[i] - p[i - f[j + Maxn]] + Mod) % Mod;
			} else break;//根据 j 的奇偶性来判断当前应该加还是应该减
			if(f[Maxn - j] <= i) {
				if(j & 1) p[i] = (p[i] + p[i - f[Maxn - j]]) % Mod;
				else p[i] = (p[i] - p[i - f[Maxn - j]] + Mod) % Mod;
			} else break;
		}
}

int main() {
#ifdef LOACL
	freopen("in.txt", "r", stdin);
	freopen("out.txt", "w", stdout);
#endif
	Init();
	int _;
	scanf("%d", &_);
	while(_--) {
		int n;
		scanf("%d", &n);
		printf("%lld\n", p[n]);
	}
	return 0;
}

HDU 4658

#include <cstdio>
#include <algorithm>
using namespace std;

typedef long long ll;
const int Maxn = 1e5;
const ll Mod = 1e9 + 7;

ll f[Maxn * 2 + 5];
ll p[Maxn + 5];
ll ans[Maxn + 5];
void Init() {
	for(int i = -Maxn; i <= Maxn; i++)
		f[i + Maxn] = 1LL * i * (3 * i - 1) / 2;
	p[0] = 1;
	for(int i = 1; i <= Maxn; i++)
		for(int j = 1; j <= i; j++) {
			if(f[j + Maxn] <= i) {
				if(j & 1) p[i] = (p[i] + p[i - f[j + Maxn]]) % Mod;
				else p[i] = (p[i] - p[i - f[j + Maxn]] + Mod) % Mod;
			} else break;
			if(f[Maxn - j] <= i) {
				if(j & 1) p[i] = (p[i] + p[i - f[Maxn - j]]) % Mod;
				else p[i] = (p[i] - p[i - f[Maxn - j]] + Mod) % Mod;
			} else break;
		}
}

ll Solve(int n, int k) {
	ll ret = p[n], dir = -1;
	for(int i = 1; ; i++) {
		ll val1 = 1LL * k * i * (3 * i - 1) / 2,
			val2 = 1LL * k * i * (3 * i + 1) / 2;
		if(val1 > n && val2 > n) break;
		if(val1 <= n) ret = (ret + dir * p[n - val1] + Mod) % Mod;
		if(val2 <= n) ret = (ret + dir * p[n - val2] + Mod) % Mod;
		dir *= -1;
	}
	return ret;
}

int main() {
#ifdef LOACL
	freopen("in.txt", "r", stdin);
	freopen("out.txt", "w", stdout);
#endif
	Init();
	int _;
	scanf("%d", &_);
	while(_--) {
		int n, k;
		scanf("%d %d", &n, &k);
		printf("%lld\n", Solve(n, k));
	}
	return 0;
}