5-6 判断出栈序列

【问题描述】对于一个栈，已知元素的进栈序列，判断一个由栈中所有元素组成的排列是否是可能的出栈序列。
比如，进栈序列为1 2 3 4，则可能的出栈序列有4 3 2 1，1 4 3 2等。而1 4 2 3就不是。
【输入形式】从标准输入读取第一行是一个整数N（3≤N≤10），代表有N个元素，其进栈序列是1 2 3 …… N。
第二行是空格分隔的1~N的数字的一个排列。
【输出形式】向标准输出打印结果。如果该排列是可能的出栈序列，则打印“YES”，否则打印“NO”。在行末要输出一个回车符。
【输入样例】
3
3 1 2
【输出样例】
NO
【样例说明】进栈序列为1 2 3的出栈序列里没有3 1 2

• 自做答案

#include<stdio.h>
#include<string.h>
int judge(int num[],int n)
{
    int i,j;
    int ci = 1;
    for(i = 0;i<n;i++,ci++)
    {
        //printf("%d %d %d\n",ci,num[i],i);
        if(ci == num[i])
        {
            continue;
        }
        else if(ci < num[i])
        {
            int flag = 1;
            for(j = 1;j<=num[i]-ci;j++)
            {
                if(num[i + j-1] != num[i + j] + 1)
                {
                    flag = 0;
                }
            }
            if(flag == 0)
                return 0;
            else //if (flag == 1)
            {
                int t = 0;
                t = num[i] - ci;
                ci = num[i];
                i = i + t;
            }
        }
        else
        {
            return 0;
        }
    }
    if(i == n) return 1;
    else return 0;
}
int main()
{
    int i,j;
    int N;
    int num[100];
    scanf("%d",&N);
    for(i = 0;i<N;i++)
    {
        scanf("%d",&num[i]);
    }
    int flag = 0;
    flag = judge(num,N);
    if(flag ==1) printf("YES\n");
    else printf("NO\n");
    return 0;
}