uva514 栈的应用

uva514

• 开始以为是旋转某一截，就用奇怪的方法，以致后边发现思路错了甚至改不会来
• 但是进车站的车厢正好满足后进先出，应该考虑栈
• 在B轨运行的车箱，无非就是从A轨按顺序取，或者在栈顶取，
• 而A轨的车厢，如果不能与B轨道匹配，就说明他进栈了
• 这样考虑就会清晰明了

#include <iostream>
#include <fstream>
#include <stack>

using namespace std;

#define TEST 1

int main()
{
    int N;
    ofstream ofs;

    if (TEST)
    {
        ofs.open("/home/lixiaoqi/Documents/Code/C++/1.txt");
        if (!ofs.is_open())
            throw runtime_error("FILE NOT OPEN!");
    }
    ostream &os = TEST ? ofs : cout;

    while (cin >> N && N)
    {
        int *B = new int[N + 1];
        while (cin >> B[1] && B[1])
        {
            bool OK = true;
            stack<int> s;
            for (int i = 2; i <= N; ++i)
                cin >> B[i];

            int a, b;
            for (a = b = 1; b <= N;)
            {
                if (B[b] == a)
                    a++, b++;
                else if (s.size() && s.top() == B[b])
                    s.pop(), b++;
                else if (a <= N)
                    s.push(a++);
                else
                {
                    OK = false;
                    break;
                }
            }

            os << (OK ? "Yes" : "No") << '\n';
        }
        os << '\n';
        delete[] B;
    }
    return 0;
}