2019 天梯賽 PTA L2-032 彩虹瓶 (25 分) -堆疊的簡單模擬

L2-4 彩虹瓶

彩虹瓶的製作過程（並不）是這樣的：先把一大批空瓶鋪放在裝填場地上，然後按照一定的順序將每種顏色的小球均勻撒到這批瓶子裡。

假設彩虹瓶裡要按順序裝 N 種顏色的小球（不妨將順序就編號為 1 到 N）。現在工廠裡有每種顏色的小球各一箱，工人需要一箱一箱地將小球從工廠裡搬到裝填場地。如果搬來的這箱小球正好是可以裝填的顏色，就直接拆箱裝填；如果不是，就把箱子先碼放在一個臨時貨架上，碼放的方法就是一箱一箱堆上去。當一種顏色裝填完以後，先看看貨架頂端的一箱是不是下一個要裝填的顏色，如果是就取下來裝填，否則去工廠裡再搬一箱過來。

如果工廠裡發貨的順序比較好，工人就可以順利地完成裝填。例如要按順序裝填 7 種顏色，工廠按照 7、6、1、3、2、5、4 這個順序發貨，則工人先拿到 7、6 兩種不能裝填的顏色，將其按照 7 在下、6 在上的順序堆在貨架上；拿到 1 時可以直接裝填；拿到 3 時又得臨時碼放在 6 號顏色箱上；拿到 2 時可以直接裝填；隨後從貨架頂取下 3 進行裝填；然後拿到 5，臨時碼放到 6 上面；最後取了 4 號顏色直接裝填；剩下的工作就是順序從貨架上取下 5、6、7 依次裝填。

但如果工廠按照 3、1、5、4、2、6、7 這個順序發貨，工人就必須要憤怒地折騰貨架了，因為裝填完 2 號顏色以後，不把貨架上的多個箱子搬下來就拿不到 3 號箱，就不可能順利完成任務。

另外，貨架的容量有限，如果要堆積的貨物超過容量，工人也沒辦法順利完成任務。例如工廠按照 7、6、5、4、3、2、1 這個順序發貨，如果貨架夠高，能碼放 6 只箱子，那還是可以順利完工的；但如果貨架只能碼放 5 只箱子，工人就又要憤怒了……

本題就請你判斷一下，工廠的發貨順序能否讓工人順利完成任務。

輸入格式：
輸入首先在第一行給出 3 個正整數，分別是彩虹瓶的顏色數量 N（1<N≤10
​3
​​ ）、臨時貨架的容量 M（<N）、以及需要判斷的發貨順序的數量 K。

隨後 K 行，每行給出 N 個數字，是 1 到N 的一個排列，對應工廠的發貨順序。

一行中的數字都以空格分隔。

輸出格式：
對每個發貨順序，如果工人可以愉快完工，就在一行中輸出 YES；否則輸出 NO。

輸入樣例：
7 5 3
7 6 1 3 2 5 4
3 1 5 4 2 6 7
7 6 5 4 3 2 1
輸出樣例：
YES
NO
NO

程式碼如下：

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e3+1;
int a[maxn];
int main()
{
	stack<int>c;
	int n,m,k;
	cin>>n>>m>>k;
	while(k--)
	{
		int ans=1;
		for(int i=1;i<=n;i++)
		cin>>a[i];
		bool flag=false;
		for(int i=1;i<=n;i++)
		{
			if(a[i]==ans)       
			{
				ans++;
				while(!c.empty()&&c.top()==ans)   //連續符合條件的情況
				{
					c.pop();  //箱子從臨時貨架搬出
					ans++;
					if(ans==n+1) {   //順利完工
					cout<<"YES"<<endl;
					flag=1;
					break;}
				}
			}
			else if(a[i]>ans&&c.empty()) //第一個入棧的元素
			{
			c.push(a[i]);}
			else if(a[i]>ans&&!c.empty())  //如果新元素大於棧中的元素就直接NO了
			{
				if(c.size()==m||c.top()<a[i])  //臨時貨架超容量直接NO
				{
					cout<<"NO"<<endl;
					break;
				}
				else 
				c.push(a[i]);
			}
		}
	if(!flag&&ans==n+1) cout<<"YES"<<endl;
	while(!c.empty())c.pop();     //清空棧中元素
	}
	return 0;
}