算法竞赛入门——铁轨

某城市有一个火车站，铁轨铺设如图所示。有n节车厢从A方向驶入车站，按进站顺序编号为1~n。你的任务是让它们按照某种特定的顺序进入B方向的铁轨并驶出车站。为了重组车厢，你可以借助中转站C。这是一个可以停放任意多节车厢的车站，但由于末端封顶，驶入C的车厢必须按照相反的顺序驶出C。对于每个车厢，一旦从A移入C，就不能再回到A了；一旦从C移入B，就不能回到C了。换句话说，在任意时刻，只有两种选择：A-->C和C-->B。
输入样例：
5
1 2 3 4 5
5
5 4 1 2 3
6 
6 5 4 3 2 1
样例输出：
Yes
No
Yes

这里用了两种方式，一种是没用STL库的，一种是用了STL库的
代码如下：

#include<stdio.h>
#include<iostream>
const int MAX = 1000;

int S[MAX];

int main()
{
    int i, n;
    int a[1000];
    while (scanf("%d", &n))
    {
        for (i = 0; i<n; i++)
        {
            S[i] = i + 1;
        }
        for (i = 1; i <= n; i++)
        {
            scanf("%d", &a[i]);
        }
        int top = n-1;
        int flag = 0, j = 1, k = n;
        while (top)
        {
            printf("%d", S[top]);
            if (S[top] == a[j])
            {
                top--;
                j++;
                flag = 1;
            }
            if (S[top] == a[k])
            {
                top--;
                k--;
                flag=1;
            }
            else
            {
                break;
            }
        }
        if (flag)
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;

}

#include<iostream>
#include<stack>

using namespace std;

int main()
{
    stack<int>S;
    int n;
    int a[1000];
    while (cin>>n)
    {
        for (int i = 1; i <= n; i++)
        {
            S.push(i);
            cin >> a[i];
        }
        int j = 1, k = n, flag=0;
        while (!S.empty())
        {
            if (S.top() == a[j])
            {
                flag = 1;
                j++;
            }
            if (S.top() == a[k])
            {
                flag = 1;
                k--; 
            }
            else
            {
                flag = 0;
                break;
            }
            S.pop();
        }
        if (flag)
            cout << "Yes" << endl;
        else
            cout << "No" << endl;
    }

}