[NOIP2003-普及] 栈

GitHub跳转原题关系图返回列表

求 1~n 经栈操作得到的合法输出序列总数,即第 n 个卡特兰数,用 O(n) 递推计算。

OJ: roj

题目 ID: 3521

难度:普及-

标签:组合数学卡特兰数递推python

创建: 2026-10-02 05:05

更新: 2026-10-06 13:03

题目描述

给定 nn(1≤n≤181\le n\le 18),初始有操作数序列 1,2,…,n1,2,\dots,n、空栈和空输出序列。每次可将操作数序列头端元素入栈,或将栈顶元素弹出到输出序列末尾。求最终能得到的不同输出序列总数。

样例 n=3n=3 时输出为 55。

思路

合法输出序列的总数即为第 nn 个卡特兰数。利用递推式 C0=1C_0=1,Ck=Ck−1⋅2(2k−1)k+1C_k=C_{k-1}\cdot\dfrac{2(2k-1)}{k+1},O(n)O(n) 递推即可得到答案。

参考代码

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;
}