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<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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章