acwing 129. 火车进栈 [dfs搜索]

这里有n列火车将要进站再出站,但是,每列火车只有1节,那就是车头。

这n列火车按1到n的顺序从东方左转进站,这个车站是南北方向的,它虽然无限长,只可惜是一个死胡同,而且站台只有一条股道,火车只能倒着从西方出去,而且每列火车必须进站,先进后出。

也就是说这个火车站其实就相当于一个栈,每次可以让右侧头火车进栈,或者让栈顶火车出站。

车站示意如图:

        出站<——    <——进站
                 |车|
                 |站|
                 |__|

现在请你按《字典序》输出前20种可能的出栈方案。

输入格式
输入一个整数n,代表火车数量。

输出格式
按照《字典序》输出前20种答案,每行一种,不要空格。

数据范围
1≤n≤20
输入样例:
3
输出样例:
123
132
213
231
321

输出前20种出站的顺序;
设未进站的车位state3,站种的位state2,已经出站的位state1;
图来自yxc dalao的直播;
在这里插入图片描述
可以看出,有2种操作,因为state3中的元素递增,且栈中的元素一定比state3中的小,如果先进行2操作再进行3,那么2操作出栈的元素一定小于state3中的元素,字典序最小,所以每次dfs是要先进行2操作再进行1操作;

#include <cstdio>
#include <iostream>
#include<stack>
using namespace std;
int n,st3=1,cnt=20;
vector<int> st1;
stack<int> st2;

void dfs()
{
    if(!cnt)return ;
    if(st1.size() == n)
    {
        cnt --;
        for(int i = 0;i < n; i ++)
            cout << st1[i];
        cout << endl;
        return ;
    }
    if(st2.size())//进行操作2
    {
        st1.push_back(st2.top());
        st2.pop();
        dfs();
        st2.push(st1.back());//dfs后恢复原状
        st1.pop_back();
    }
    if(st3 <= n)
    {
        st2.push(st3);
        st3 ++;
        dfs();
        st3 --;
        st2.pop();
    }
}
int main() {
    cin >> n;
    dfs();
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
为了判断通过一个栈能否得到指定的出栈序列,可以采用模拟法。具体做法如下: 1. 定义一个栈,初始为空; 2. 依次将1~n压入栈中,每次操作后判断是否需要弹出栈顶元素; 3. 如果当前栈顶元素等于出栈序列中下一个应该弹出的元素,就弹出它并将出栈序列指针后移一位; 4. 最终如果栈为空并且出栈序列指针已经指向了最后一个元素,说明能够得到指定的出栈序列,否则不能。 下面是实现上述算法的 Python 代码: ```python def is_valid_pop_order(push_order, pop_order): stack = [] push_index, pop_index = 0, 0 while push_index < len(push_order): stack.append(push_order[push_index]) push_index += 1 while stack and stack[-1] == pop_order[pop_index]: stack.pop() pop_index += 1 return len(stack) == 0 and pop_index == len(pop_order) # 测试 print(is_valid_pop_order([1, 2, 3, 4, 5], [4, 5, 3, 2, 1])) # True print(is_valid_pop_order([1, 2, 3, 4, 5], [4, 3, 5, 1, 2])) # False ``` 在上述代码中,is_valid_pop_order函数接收两个参数push_order和pop_order,分别表示进栈序列和出栈序列。函数中定义了一个栈stack,以及两个指针push_index和pop_index,分别用于记录进栈序列和出栈序列中已经处理过的位置。 具体实现时,每次将一个元素压入栈中,并检查是否需要弹出栈顶元素。如果当前栈顶元素等于出栈序列中下一个应该弹出的元素,就弹出它并将出栈序列指针后移一位。最终如果栈为空并且出栈序列指针已经指向了最后一个元素,说明能够得到指定的出栈序列,否则不能。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值