铁路（栈）

某城市有一个火车站，有n节车厢从A方向驶入车站，按进站的顺序编号为1~n。现让它们按照某种特定的顺序进入B方向的铁轨并驶出车站。为了重组车厢，你可以借助中转站C。在程序中输入车厢数目和出站的特定顺序，如果可以则输出Yes，否者输出No。
样例输入：
5
1 2 3 4 5
5
5 4 1 2 3
6
6 5 4 3 2 1
样例输出：
Yes
No
Yes

分析：
考虑好入栈时会有哪些情况。

#include <iostream>#include <cstdio>#include <stack>using namespace std;const int MAXN = 1000 + 10;int n, target[MAXN];int main() {    while (scanf("%d", &n) == 1) {        stack<int> s;        int A = 1,B = 1;        for (int i = 1; i <= n; i++)            scanf("%d", &target[i]); //输入的出栈顺序         int ok = 1;        while (B <= n) {            if (A == target[B]) { A++; B++;}   //当入栈后又直接出栈的情况             else if (!s.empty() && s.top() == target[B]) { s.pop(); B++;} //当栈顶元素等于出栈顺序元素             else if (A <= n) s.push(A++);     //入栈后没有直接出栈               else { ok = 0; break; }        }        printf("%s\n", ok ? "YES" : "NO");    }    return 0;}