1051.Pop Sequence

http://pat.zju.edu.cn/contests/pat-a-practise/1051

#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;
}