浙大PAT 1051題 1051. Pop Sequence

設置一個變量tmp，由1開始自增。當棧頂元素與輸入的元素不匹配時將當前tmp入棧並tmp自加，當匹配時棧頂出棧。

#include<stdio.h>
#include<stack>
using namespace std;
int main(){
	int i,j,M,N,K;
	int input,tmp,flag;
	scanf("%d %d %d",&M,&N,&K);
	for(i=0;i<K;i++){
		flag=1;
		stack<int> ss;
		tmp=1;
		for(j=0;j<N;j++){
			scanf("%d",&input);
			while(flag){
				if(ss.empty()||ss.top()!=input){
					ss.push(tmp);
					//printf("push:%d ",ss.top());
					if(ss.size()>M){
						flag=0;
						break;
					}
					tmp++;
				}
				else if(ss.size()>=1&&ss.top()==input){
					//printf("pop:%d\n",ss.top());
					ss.pop();
					break;
				}
			}
		}
		if(ss.size()>0) flag=0;
		if(flag) printf("YES\n");
		else printf("NO\n");
	}
	return 0;
}