AcWing 129 火车进栈(模拟,dfs)

题目链接

解题思路

  火车进栈一共有两种状态,一种是后面的火车车厢进栈,一种是前面的火车出栈。可以发现,如果后面的车厢进栈了,因为先进先出,而他的编号较大,那么他出栈的形成的序列的字典序就一定比先让栈中的火车出栈要大。
  所以说如果要输出字典序最大的序列,应该优先让前面的火车出栈,然后再让后面的火车进栈。最后用回溯法模拟所有的情况。

代码

int n, cnt = 20;
stack<int> sk;
vector<int> res;
void dfs(int p) {
    if (!cnt) return;
    if (res.size()==n) {
        --cnt;
        for (auto num : res) cout << num;
        cout << endl;
        return;
    }
    if (!sk.empty()) {
        res.push_back(sk.top());
        sk.pop();
        dfs(p);
        sk.push(res.back());
        res.pop_back();
    }
    if (p<=n) {
        sk.push(p);
        dfs(p+1);
        sk.pop();
    }
}
int main() {
    cin >> n;
    dfs(1);
    return 0;
}
原文地址:https://www.cnblogs.com/shuitiangong/p/13328733.html