Codeup墓地-1094

Codeup墓地-1094
src:http://codeup.cn/problem.php?id=1094

1094: 出栈合法性
时间限制: 1 Sec 内存限制: 32 MB
题目描述
已知自然数1，2，…，N（1<=N<=100）依次入栈，请问序列C1，C2，…，CN是否为合法的出栈序列。
输入
输入包含多组测试数据。
每组测试数据的第一行为整数N（1<=N<=100），当N=0时，输入结束。
第二行为N个正整数，以空格隔开，为出栈序列。
输出
对于每组输入，输出结果为一行字符串。
如给出的序列是合法的出栈序列，则输出Yes，否则输出No。
样例输入
5
3 4 2 1 5
5
3 5 1 4 2
0
样例输出
Yes
No

#include<iostream>
#include<cstdio>
#include<stack>
#include<stdio.h>
using namespace std;
int main(){
    stack<int> s;
    int a[10000]={0};
    int n,flag,i,j,top;
    while(1){
        scanf("%d",&n);
        for(i=0;i<=n-1;++i)
            scanf("%d",&a[i]);
        if(n==0)
            return 0;
        flag=1;
        for(i=1;i<=a[0];++i)
            s.push(i);
        s.pop();
        for(j=1;j<=n-1;++j){
            if(a[j]>a[j-1])
                for(i=a[j-1]+1;i<=a[j];++i){
                    s.push(i);
                    s.pop();
                }
            else{
                top=s.top();
                if(top==a[j])
                    s.pop();
                else{
                    flag=0;
                    break;
                }
            }
        }
        if(flag==1)
            printf("Yes\n");
        else
            printf("No\n");
    }
    return 0;
}