题目地址:
https://www.acwing.com/problem/content/description/3175/
我们知道包含 N N N个元素的堆可以看成是一棵包含 N N N个节点的完全二叉树。每个节点有一个权值。对于小根堆来说,父节点的权值一定小于其子节点的权值。假设 N N N个节点的权值分别是 1 ∼ N 1∼N 1∼N,你能求出一共有多少种不同的小根堆吗?例如对于 N = 4 N=4 N=4有如下 3 3 3种:
1
/ \
2 3
/
4
1
/ \
3 2
/
4
1
/ \
2 4
/
3
由于数量可能超过整型范围,你只需要输出结果除以 1000000009 1000000009 1000000009的余数。
输入格式:
一个整数
N
N
N。
输出格式:
一个整数表示答案。
数据范围:
对于
40
40%
40的数据,
1
≤
N
≤
1000
1≤N≤1000
1≤N≤1000;
对于
70
70%
70的数据,
1
≤
N
≤
10000
1≤N≤10000
1≤N≤10000;
对于
100
100%
100的数据,
1
≤
N
≤
100000
1≤N≤100000
1≤N≤100000。
N N N个节点的二叉堆的结构是确定的,设下标为 1 1 1的点是堆顶,我们可以预处理出 s s s数组,使得 s [ i ] s[i] s[i]是下标为 i i i的子堆的节点数量。接着设 f [ i ] f[i] f[i]是下标为 i i i的节点为堆顶的子堆的方案数量,如果 i i i无左右孩子,显然 f [ i ] = 1 f[i]=1 f[i]=1;如果只有左孩子,则 f [ i ] = f [ 2 i ] f[i]=f[2i] f[i]=f[2i];如果左右孩子都有,则除了树根之外,可以随便从 s [ i ] s[i] s[i]个数里选 s [ 2 i ] s[2i] s[2i]个放到左子堆,左右子堆方案不影响,从而 f [ i ] = ( s [ i ] − 1 s [ 2 i ] ) f [ 2 i ] f [ 2 i + 1 ] f[i]={s[i]-1\choose s[2i]}f[2i]f[2i+1] f[i]=(s[2i]s[i]−1)f[2i]f[2i+1]。从 N N N到 1 1 1递推出 f [ i ] f[i] f[i],最后返回 f [ 1 ] f[1] f[1]即可。容易知道 1000000009 1000000009 1000000009是素数,求逆元需要用到费马小定理和快速幂。代码如下:
#include <iostream>
using namespace std;
const int N = 100010, MOD = 1e9 + 9;
int n;
long f[N], s[N];
long fact[N], inv[N];
long fast_pow(long x, int k) {
long res = 1;
while (k) {
if (k & 1) res = res * x % MOD;
x = x * x % MOD;
k >>= 1;
}
return res;
}
void init() {
fact[0] = inv[0] = 1;
for (int i = 1; i <= n; i++) {
fact[i] = fact[i - 1] * i % MOD;
inv[i] = fast_pow(fact[i], MOD - 2);
}
}
long C(int a, int b) {
return fact[a] * inv[b] % MOD * inv[a - b] % MOD;
}
int main() {
scanf("%d", &n);
init();
for (int i = n; i; i--) {
s[i] = 1;
if (i * 2 <= n) s[i] += s[i * 2];
if (i * 2 + 1 <= n) s[i] += s[i * 2 + 1];
}
for (int i = n; i; i--) {
f[i] = 1;
if (i * 2 <= n) f[i] = f[i * 2];
if (i * 2 + 1 <= n) f[i] = f[i] * C(s[i] - 1, s[i * 2]) % MOD * f[i * 2 + 1] % MOD;
}
printf("%ld\n", f[1]);
}
时空复杂度 O ( n ) O(n) O(n)。