【ACWing】3172. 堆的计数

题目地址：

https://www.acwing.com/problem/content/description/3175/

我们知道包含$N$个元素的堆可以看成是一棵包含$N$个节点的完全二叉树。每个节点有一个权值。对于小根堆来说，父节点的权值一定小于其子节点的权值。假设$N$个节点的权值分别是$1\sim N$，你能求出一共有多少种不同的小根堆吗？例如对于$N=4$有如下$3$种：

    1
   / \
  2   3
 /
4

    1
   / \
  3   2
 /
4

    1
   / \
  2   4
 /
3

由于数量可能超过整型范围，你只需要输出结果除以$1000000009$的余数。

输入格式：
一个整数$N$。

输出格式：
一个整数表示答案。

数据范围：
对于$40$的数据，$1\le N\le 1000$；
对于$70$的数据，$1\le N\le 10000$；
对于$100$的数据，$1\le N\le 100000$。

$N$个节点的二叉堆的结构是确定的，设下标为$1$的点是堆顶，我们可以预处理出$s$数组，使得$s[i]$是下标为$i$的子堆的节点数量。接着设$f[i]$是下标为$i$的节点为堆顶的子堆的方案数量，如果$i$无左右孩子，显然$f[i]=1$；如果只有左孩子，则$f[i]=f[2i]$；如果左右孩子都有，则除了树根之外，可以随便从$s[i]$个数里选$s[2i]$个放到左子堆，左右子堆方案不影响，从而$f[i]=\left(\genfrac{}{}{0ex}{}{s[i]-1}{s[2i]}\right)f[2i]f[2i+1]$。从$N$到$1$递推出$f[i]$，最后返回$f[1]$即可。容易知道$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)$。