题解 | #火车进站#

火车进站

http://www.nowcoder.com/practice/97ba57c35e9f4749826dc3befaeae109

todo : 增加说明

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

void dfs(const vector<int>& in, int index, stack<int>& st, vector<int>& out, vector<vector<int>>& res)
{
    if (index >= in.size() && st.empty()) {
        res.push_back(out);
        return;
    }
    // 进
    if (index < in.size()) {
        st.push(in[index]);
        dfs(in, index + 1, st, out, res);
        st.pop();
    }
    // 出
    if (!st.empty()) {
        out.push_back(st.top());
        st.pop();
        dfs(in, index, st, out, res);
        st.push(out.back());
        out.pop_back();
    }
}

int main()
{
    int n;
    while (cin >> n) {
        vector<int> nums(n);
        for (int i = 0; i < n; ++i) { cin >> nums[i]; }
        stack<int> st;
        vector<int> path;
        vector<vector<int>> res;
        dfs(nums, 0, st, path, res);
        sort(res.begin(), res.end());
        for (auto p : res) {
            for (int a : p) { cout << a << " "; }
            cout << endl;
        }
    }
    return 0;
}
全部评论
回溯用的很牛
2 回复 分享
发布于 2022-08-10 15:09
可以把DFS函数几个参数设为全局变量,这样更简洁 https://blog.nowcoder.net/n/67d543f1567248398ccc36377b0fd5c6
1 回复 分享
发布于 2023-01-13 11:48 江苏
好精妙的算法,怎么想到的?调试了多久才成功的?真是天才
点赞 回复 分享
发布于 2022-06-03 19:22
有哪位老哥看懂了没有?
点赞 回复 分享
发布于 2022-08-01 14:30
真的牛批
点赞 回复 分享
发布于 2022-09-22 16:50 广东
莫非他真的是天才!
点赞 回复 分享
发布于 2023-10-05 21:02 河南

相关推荐

评论
23
3
分享
牛客网
牛客企业服务