题解 | #火车进站#

火车进站

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

解法:DFS++SET
DFS用于找所有可能的情况,栈用于模拟火车进出,set用于保存结果并自动排序。
在DFS中的每一步只有两种可能:
①火车出站,即从栈顶弹出一个,保存在输出数组中
②新火车进站,从输入列表中取一个火车压入栈,因此需要一个index来指示当前取到第几个火车了

DFS终止条件为当前输出数组的size等于n,随即将输出数组保存在set中并return。
具体实现如下

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


void dfs(set<vector<int>>& outputse, vector<int>& invec, stack<int> ist, vector<int> otvec, int index, int& n) {
    if (otvec.size() == n) {
        outputse.insert(otvec);
        return;
    }
    //每一步要不从栈中弹一个出栈,要么从输入invec中压入一个车入栈
    for (int j = 0; j < 2; j++) {
        if (j == 0 && !ist.empty()) {//出栈要保证当前栈不为空
            int temp = ist.top();
            ist.pop();
            otvec.push_back(temp);
            dfs(outputse, invec, ist, otvec, index, n);
            ist.push(temp);
            otvec.pop_back();


        }
        else if (j == 1 && index < n ) {//入栈保证索引没有超出范围
            int temp = invec[index];
            ist.push(temp);
            index++;
            dfs(outputse, invec, ist, otvec, index, n);
            index--;
            ist.pop();
        }
    }
}


int main() {
    int n;
    while (cin >> n) {
        vector<int> invec(n, 0);
        for (int i = 0; i < n; i++) {
            cin >> invec[i];

        }
        set<vector<int>> outputse;
        stack<int> ist;
        ist.push(invec[0]);//初始默认将第一辆车压入栈,因此下面函数 的index默认从1开始
        vector<int> ovec;
        dfs(outputse, invec, ist, ovec, 1, n);
        for (auto& ele : outputse) {
            for (auto& i : ele) {
                cout << i << " ";
            }
            cout << endl;
        }
    }
}
全部评论

相关推荐

评论
1
1
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务