彩虹瓶的制作過程(並不)是這樣的:先把一大批空瓶鋪放在裝填場地上,然后按照一定的順序將每種顏色的小球均勻撒到這批瓶子里。
假設彩虹瓶里要按順序裝 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<cstdio>
#include<iostream>
#include<algorithm>
#include<string>
#include<stack>
#include<cmath>
#include<cstring>
#include<cstdlib>
using namespace std;
#define all(x) (x).begin(),(x).end()
#define pb push_back
#define mp make_pair
#define ff first
#define ss second
typedef long long ll;
stack<int> st;
int s[2000];
int main() {
int n, m, k;
cin >> n >> m >> k;
while (k--) {
while (!st.empty())st.pop();
for (int i = 0; i < n; i++) {
cin >> s[i];
}
int now = 1;
int yes = 1;
for (int i = 0; i < n; i++) {
if (s[i] == now) {
now++;
while (st.empty()==false && st.top() == now) {
st.pop();
now++;
}
}
else {
st.push(s[i]);
if (st.size() > m) {
yes = 0;
}
}
}
if (yes == 0 || st.size() > 0) {
cout << "NO" << endl;
}
else {
cout << "YES" << endl;
}
}
}