彩虹瓶的製作過程(並不)是這樣的:先把一大批空瓶鋪放在裝填場地上,然後按照一定的順序將每種顏色的小球均勻撒到這批瓶子裏。
假設彩虹瓶裏要按順序裝 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 <iostream>
#include <stack>
using namespace std;
int main() {
int N, M, K;
cin >> N >> M >> K;
while (K--) {
int num = 1, tmp, flag = 1;
stack<int>st;
for (int i = 0; i < N; ++i) {
cin >> tmp;
st.push(tmp);
while (!st.empty()&&st.top()==num){
st.pop();
num++;
}
if (st.size() > M) flag = 0;
}
cout << (st.empty() && flag ? "YES" : "NO") << endl;
}
return 0;
}