1051.Pop Sequence

#include <stdio.h>
#include <string.h>
#include <stack>
using namespace std;

stack<int> ans;
int a[1005],b[1005];

int main()
{
int m,n,k,i;
while (scanf("%d%d%d",&m,&n,&k)!=EOF)        //m栈最大深度，n序列长度，k待检验序列个数
{
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
for (i=1;i<=n;i++)
{
a[i]=i;
}
while (k--)
{
while (!ans.empty())          //每次检测，清空栈
{
ans.pop();
}
bool flag=true;
for(i=1;i<=n;i++)
{
scanf("%d",&b[i]);
}
int *p=(a+1),*q=b+1;                   //注意数组从1~n保存值，所以指针后移一位
while (*q!=0)
{
if (ans.empty()&&*p!=0)                    //当前栈空，则进栈
{
ans.push(*p++);
}
while (ans.top()!=*q&&*p!=0)   //栈顶不等于出栈首元素，则继续进栈
{
ans.push(*p++);
if (ans.size()>m)      //超过栈的最大深度，false
{
flag=false;
}
}
if (ans.top()==*q)             //若相同，则出栈序列当前元素拿走，说明匹配
{
q++;
ans.pop();
continue;
}
if (*p==0&&ans.top()!=*q)      //若入栈序列到末尾与出栈首还不同，false
{
flag=false;
}
if (!flag)
{
break;
}
}
if (!flag)
{
printf("NO\n");
}
else
printf("YES\n");
}
}
return 0;
}