假设彩虹瓶里要按顺序装 N 种颜色的小球(不妨将顺序就编号为 1 到 N)。现在工厂里有每种颜色的小球各一箱,工人需要一箱一箱地将小球从工厂里搬到装填场地。如果搬来的这箱小球正好是可以装填的颜色,就直接拆箱装填;如果不是,就把箱子先码放在一个临时货架上,码放的方法就是一箱一箱堆上去。当一种颜色装填完以后,先看看货架顶端的一箱是不是下一个要装填的颜色,如果是就取下来装填,否则去工厂里再搬一箱过来。
货架的容量有限,如果要堆积的货物超过容量,工人也没办法顺利完成任务。
输入格式:
输入首先在第一行给出 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
通过样例:cpp
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
int a[1005];//栈,货架
int main()
{
int n,m,k,i,j;
while(scanf("%d%d%d",&n,&m,&k) != EOF){
while(k--){
memset(a,0,sizeof(a));
int x;
int top = 0; //栈顶
int ans = 0; //放入货架的货物数量
int index = 1; //标记放到哪个编号的货物
int flag = 0; //标记是否栈溢出,超出货架容量
for(i = 0 ; i < n ; i++)
{
scanf("%d",&x);
a[top++] = x; //入栈
ans++; //货架货物数量增加
if(ans == m+1 && index == a[top-1]){ //特殊判断当货架满时,但是下一件货物刚好符合工人装填
index++;
top--;
ans--;
}
while(top > 0 && a[top-1] == index && ans <= m){ //循环判断栈顶是否符合工人要搬运货物
index++;
top--;
ans--;
}
}
if(ans >= m ){ //栈满,货架有货物,标记一下
flag = 1;
}
if(top == 0 && flag == 0){
printf("YES\n");
}else{
printf("NO\n");
}
}
}
return 0;
}
STL
#include<iostream>
#include<stack>
using namespace std;
int main() {
int n, m, k, tmp, count, full;
cin >> n >> m >> k;
for (int i = 0; i < k; i++) {
stack<int> stk;
full = 1; count = 1;
for (int j = 0; j < n; j++) {
cin >> tmp;
if (tmp != count) { //不相等就进栈
stk.push(tmp);
if (stk.size() > m)
full = 0;
}
else {
count++; //相等就把栈顶能出的全出了
while (!stk.empty() && full && stk.top() == count && ++count)
stk.pop();
}
}
cout << (stk.empty() ? "YES" : "NO") << endl;
}
return 0;
}