AcWing 129. 火车进栈

题目描述

这里有n列火车将要进站再出站，但是，每列火车只有1节，那就是车头。
这n列火车按1到n的顺序从东方左转进站，这个车站是南北方向的，它虽然无限长，只可惜是一个死胡同，而且站台只有一条股道，火车只能倒着从西方出去，而且每列火车必须进站，先进后出。
也就是说这个火车站其实就相当于一个栈，每次可以让右侧头火车进栈，或者让栈顶火车出站。
车站示意如图：
出站<—— <——进站
|车|
|站|
|__|
现在请你按《字典序》输出前20种可能的出栈方案。
输入格式
输入一个整数n，代表火车数量。
输出格式
按照《字典序》输出前20种答案，每行一种，不要空格。
数据范围
1≤n≤20
输入样例：
3
输出样例：
123
132
213
231
321

题解

维护好3个状态

1. state1表示出栈元素排队的状态，使用vector维护
   state2表示当前栈元素的状态，使用栈stack维护
   state3表示当前第几个元素准备进栈，使用整数int维护
2. 整个系统有两个操作情况
   一：将整数state3的元素进入到state2栈中
   二：将栈state2的元素出栈到state3链表中
3. 由于题目要求按《字典序》输出前20种可能的出栈方案，所以需要优先执行二操作

#include <iostream>
#include <stack>
#include <vector>

using namespace std;

int n;
int cnt = 20;
vector<int> state1;
stack<int> state2;
int state3 = 1;

void dfs()
{
    if(!cnt) return;
    if(state1.size() == n)
    {
        cnt--;
        for(int i = 0; i < n; i++) cout << state1[i];
        cout << endl;
        return;
    }
    
    if(state2.size())
    {
        state1.push_back(state2.top());
        state2.pop();
        dfs();
        state2.push(state1.back());
        state1.pop_back();
    }
    
    if(state3 <= n)
    {
        state2.push(state3);
        state3++;
        dfs();
        state3--;
        state2.pop();
    }
}

int main()
{
    cin >> n;
    dfs();
    return 0;
}

进阶

数据范围 1≤n≤60000该怎么求？

题解

首先规定 11 表示栈中进入一节车厢，00 表示栈中弹出一节车厢，那么每种火车进出栈方案都能用一串 0101 序列来表示。
由于每节车厢都要进栈一次出栈一次，所以每种方案所对应的 0101 序列的长度都是 2n。
那么火车进出栈方案的总数量就可以看成是所有合法的 0101 串的个数。
那咋算是合法的嘞？
要有车厢出栈，那么栈中就必须要有车厢。
在 0101 序列中，就对应了在任意位置，其前面的 1的个数大于等于其前面 0 的个数.
我们只需求卡特兰数即可。

#include <iostream>
#include <algorithm>

using namespace std;
typedef long long LL;
const int N = 60010, M = 120010;

LL res[N], tt;
int q[M];
bool st[M];

void multi(int b)
{
    LL t = 0;
    for(int i = 0; i <= tt; i++)
    {
        res[i] = res[i] * b + t;
        t = res[i] / 1000000000; //压位，相当于转化为1000000000数
        res[i] %= 1000000000;
    }
    
    while(t)
    {
        res[++tt] = t % 1000000000;
        t /= 1000000000;
    }
}


void out()
{
    printf("%lld", res[tt]);
    for (int i = tt - 1; i >= 0; i -- ) printf("%09lld", res[i]);
    cout << endl;
}

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

int main()
{
    int n;
    cin >> n;
    for(int i = 2; i <= 2 * n; i++)
        for(int j = 2 * i; j <= 2 * n; j += i)
        {
            st[j] = true;
        }
    
    for(int i = 2; i <= 2 * n; i++)
    {
        if(!st[i])
        {
            q[i] = get(2 * n, i) - get(n, i) * 2; //分子分母质因数的差
        }
    }
    
    
    //(n + 1)里各质因数的数目
    int k = n + 1;
    
    for(int i = 2; i <= k; i++)
    {
        while(k % i == 0)
        {
            k /= i;
            q[i]--;
        }
    }
    
    res[0] = 1;
    
    for(int i = 2; i <= 2 * n; i++)
    {
        while(q[i]--)
        {
            multi(i);
        }
    }
    
    out();
    
    return 0;
}