pta-L2-032 彩虹瓶 （25 分)

L2-032 彩虹瓶 （25 分)

 

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

假設彩虹瓶里要按順序裝 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）、臨時貨架的容量 M（<）、以及需要判斷的發貨順序的數量 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 <iostream>
#include <stack>
using namespace std;
stack<int> sta;
int main(){
    int n,m,k,t,cnt;
    cin>>n>>m>>k;
    while(k--){
        int f=1,flag=1,cnt=0;
        for(int i=1;i<=n;i++){
            cin>>t;
            sta.push(t);
            cnt++;
            while(!sta.empty()&&sta.top()==f){//若棧頂元素正確，循環
                f++;
                sta.pop();
                cnt--;
            }
            if(cnt>m) flag=0;//棧溢出
        }
        if(sta.empty()&&flag) cout<<"YES";//棧為空且棧沒有溢出過
        else cout<<"NO";
        if(k>0) cout<<'\n';//格式控制
        while(!sta.empty())
            sta.pop();
    }
    return 0;
}