解題報告-PAT-Pop Sequence 浙大1051

# include <stdio.h>
# include <stdlib.h>
# include <string.h>
# include <algorithm>
# include <iostream>
# include <stack> 
const int MAX_SIZE = 1000+10;
using namespace std;
stack<int >s;
int se[MAX_SIZE];

/*
5 7 2
1 7 6 5 4 3 2
5 6 4 3 7 2 1
*/

/*
問題描述:
	給定一個可以容納M個元素的棧,給定N,以1.2.3.4...N的次序進棧,出棧次序不定,問能否得到
給定的出棧序列。如果能得到,則輸出YES,不能得到輸出NO。 
輸入M,N,K。K是測試序列的個數
樣例輸入:
5 7 5
1 2 3 4 5 6 7
3 2 1 7 5 6 4
7 6 5 4 3 2 1
5 6 4 3 7 2 1
1 7 6 5 4 3 2
樣例輸出:
YES
NO
NO
YES
NO
算法思想:
	對於給定的序列,5 6 4 3 7 2 1 判斷是否是合法的出棧序列,申請指針p,用p依次指向序列每一個數;申請
指針f,記錄已經入棧的最大N值,即記錄1.2.3.4.5.6.7中入棧的最大N值,因爲在入棧的過程中可能會夾雜着出棧,
棧頂元素肯定不是最大的,需要f記錄。這樣是爲了下一次遇到  s.top() < p 需要從f+1開始到p,依次將數壓入棧中。
如果 s.top() == p,則棧頂元素出棧,順序比較下一個數。如果s.top() > p,則序列不合法。 中間如果遇到s.size() > M
的情況,則也要退出。 
*/

int main(){
	int m,n,k;
	cin >> m >> n >> k;
	int p,ok,f;
	while(k--){
		//輸入 
		for(int i=0;i<n;i++){
			cin>>se[i];
		}
		//初始化 
		ok = 1;
		f = 0;
		while(!s.empty())
			s.pop();
		
		//根據序列依次判斷 
		for(int j=0; j < n; ){
			p=se[j];   //當前判斷到的數字 
			if(s.empty() || s.top() < p){//如果棧空就不會執行s.top()
				for(int t = f+1; t <= p; t++){
					s.push(t);
					if(f<s.top())
						f=s.top();
				}
			}
			if(s.top() > p){ //s.top() > p一定要放在等於p之前,因爲如果棧空,s.top()取不到。 
				ok=0; break; //但是在中間,就保證了棧不爲空。 
			}
			if(s.top() == p){
				if(s.size() > m){
					ok=0;//出棧之前判斷size,入棧也可以判斷,但是跳不出兩層循環。 
					break;
				}
				if(f < s.top())
					f = s.top();
				s.pop();
				j++;
			}	
		}//for
		if(!ok) printf("NO\n");	
		else printf("YES\n");			
	}
	
	return 0;
} 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章