【ACWing】130. 火车进出栈问题

记录算法题解

已于 2024-01-22 06:29:23 修改

阅读量826

点赞数 22

分类专栏： AC 数学文章标签：算法 c++

于 2024-01-22 05:44:27 首次发布

本文链接：https://blog.csdn.net/qq_46105170/article/details/135738892

版权

AC 数学专栏收录该内容

109 篇文章 7 订阅

订阅专栏

题目地址：

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

一列火车 $n$ 节车厢，依次编号为 $1, 2, 3, \dots, n$ 。每节车厢有两种运动方式，进栈与出栈，问 $n$ 节车厢出栈的可能排列方式有多少种。

输入格式：
输入一个整数 $n$ ，代表火车的车厢数。

输出格式：
输出一个整数 $s$ 表示 $n$ 节车厢出栈的可能排列方式数量。

数据范围：
$1 \leq n \leq 60000$

答案就是Catalan数，公式如下： $C_n = \frac{1}{n+1} \binom{2n}{n}$ 本题数据范围太大，我们可以采取分解质因数的方式来算。由于 $C_n = \frac{1}{n+1} \binom{2n}{n}=\frac{(2n)!}{n!(n+1)!}$ ，所以我们先用欧拉筛筛出 $[1, 2 n]$ 之间的素数（参考https://blog.csdn.net/qq_46105170/article/details/103760318），这样答案分解质因数之后的素数有哪些就确定了，接着我们就计算答案里每个素数的幂是多少，容易知道 $n!$ 里素数 $p$ 的幂次就是 $\sum_{k=1}^{\infty} \left\lfloor \frac{n}{p^k} \right\rfloor$ ，我们直接求出 $C_n$ 里各个素数的幂次之后，最后做一次高精度乘法即可。由于答案可能非常长，我们需要压位。代码如下：

#include <iostream>
#include <vector>
using namespace std;

const int N = 6e4 + 10;
int n;
int ps[N], cnt;
bool st[N];
int sum[N];

// 欧拉筛素数
void get_primes(int n) {
  for (int i = 2; i <= n; i++) {
    if (!st[i]) ps[cnt++] = i;
    for (int j = 0; ps[j] <= n / i; j++) {
      st[ps[j] * i] = true;
      if (i % ps[j] == 0) break;
    }
  }
}

int get(int n, int p) {
  int res = 0;
  while (n) {
    res += n / p;
    n /= p;
  }
  return res;
}

void mul(vector<int> &a, int b) {
  int t = 0;
  for (int i = 0; i < a.size(); i++) {
    t += a[i] * b;
    a[i] = t % 10000;
    t /= 10000;
  }

  while (t) {
    a.push_back(t % 10000);
    t /= 10000;
  }
}

int main() {
  scanf("%d", &n);
  int a = 2 * n, b = n;

  get_primes(a);

  for (int i = 0; i < cnt; i++) {
    int p = ps[i];
    sum[i] = get(a, p) - get(b, p) - get(a - b + 1, p);
  }

  vector<int> res{1};

  for (int i = 0; i < cnt; i++)
    for (int j = 0; j < sum[i]; j++) mul(res, ps[i]);

  printf("%d", res.back());
  for (int i = res.size() - 2; i >= 0; i--) printf("%04d", res[i]);
}