【栈】L2-032 彩虹瓶 (25分)

假设彩虹瓶里要按顺序装 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;
}