Rail_UVa514_栈

/*
    https://vjudge.net/problem/UVA-514
*/
#include "pch.h"
#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) // 输入，一共n辆火车
    {
        int isOnePart = 1; // 是否是同一次测试
        if (n == 0) // 程序应当结束
            break;
        while (isOnePart) // 是同一次测试时
        {
            stack<int> s;
            int A = 1, B = 1; // A:进C站顺序 1,2,3,...,n   B:数组target的下标
            for (int i = 1; i <= n; i++)
            {
                scanf("%d", &target[i]); // 输入出C站顺序
                if (target[i] == 0)
                {
                    isOnePart = 0;
                    printf("\n"); // 注意！
                    break;
                }
            }
            if (isOnePart == 0) // 退出此次测试，进入另一部分的测试
                break;
            int ok = 1;   // 标志，是否可以按顺序出站
            while (B <= n) // 判断当前出C站的车辆数是否达到总数
            {
                if (A == target[B]) // 要进站的火车与要出站的火车为同一辆，则进站后直接出站
                {
                    A++; B++;
                }
                else if (!s.empty() && s.top() == target[B])  // 栈（站）非空，栈顶与要出站的火车相同，则出站
                {
                    s.pop();
                    B++;  // 要出站的++
                }
                else if (A <= n)  // 要进站的与出站的不同，只进站，先等着，当栈顶与目标相同时再出（上一个if）
                {
                    s.push(A++);
                }
                else
                {
                    ok = 0;  //正常结束循环为b=n,  若前几种情况都不符合，则不能按照要求顺序出站，break
                    break;
                }
            }
            printf("%s\n", ok ? "Yes" : "No");
        }
        
    }
    return 0;
}