求 1~n 经栈操作得到的合法输出序列总数,即第 n 个卡特兰数,用 O(n) 递推计算。
OJ: roj
题目 ID: 3521
难度:普及-
标签:组合数学卡特兰数递推python
创建: 2026-10-02 05:05
更新: 2026-10-06 13:03
题目描述
给定
样例
思路
合法输出序列的总数即为第
参考代码
Python 版:
python
#!/usr/bin/env python3
# Author by Rainboy blog: https://rainboylv.com github: https://github.com/rainboylvx
# rbook: -> https://rbook.roj.ac.cn https://rbook2.roj.ac.cn
# rainboy的学习导航网站: https://idx.roj.ac.cn
# create_at: 2026-10-02 05:05
# update_at: 2026-10-02 05:05
import sys
def catalan(n: int) -> int:
"""第 n 个卡特兰数:C_k = C_{k-1} * 2(2k-1) / (k+1),整除在乘法之后做。"""
count = 1 # C_0 = 1:空序列只有一种"什么都不做"的方案
for k in range(2, n + 1):
count = count * 2 * (2 * k - 1) // (k + 1)
return count
def solve() -> None:
data = iter(map(int, sys.stdin.buffer.read().split()))
n = next(data) # 题面只给一个整数 n
print(catalan(n))
if __name__ == "__main__":
solve()C++ 版(同一算法):
cpp
/**
* Author by Rainboy blog: https://rainboylv.com github: https://github.com/rainboylvx
* rbook: -> https://rbook.roj.ac.cn https://rbook2.roj.ac.cn
* rainboy的学习导航网站: https://idx.roj.ac.cn
* create_at: 2026-10-04 14:46
* update_at: 2026-10-04 14:46
*/
#include <iostream>
using namespace std;
typedef long long ll;
// 第 n 个卡特兰数:C_0=1, C_k = C_{k-1} * 2(2k-1) / (k+1),整除在乘法之后做。
ll catalan(int n) {
ll res = 1; // C_0 = 1
for (int k = 2; k <= n; ++k)
res = res * 2 * (2 * k - 1) / (k + 1);
return res;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr);
int n;
cin >> n;
cout << catalan(n) << "\n";
return 0;
}
