题解 | #火车进站#
火车进站
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; } } }