L2-032 彩虹瓶

YuKiCheng發表於2024-03-20

純模擬,一次就AC了。

#define _CRT_SECURE_NO_WARNINGS
#include <bits/stdc++.h>
using namespace std;
vector<int> huoja;//貨架
queue<int> order;//發貨順序
int main() {
	int n, m, k;//顏色數量 貨架容量 發貨順序
	cin >> n >> m >> k;
	while (k--) {
		huoja.clear();
		while (!order.empty()) order.pop();
		for (int i = 0; i < n; i++) {
			int color;
			cin >> color;
			order.push(color);
		}
		int i = 1;
		for (i = 1; i <= n; i++) {//想得到的顏色
			if (!huoja.empty()) {//非空
				if (huoja.back() == i) {
					huoja.pop_back();
					continue;
				}
	            //如果在貨架上 但是不是最後一個
				if (find(huoja.begin(), huoja.end(), i) != huoja.end()) {
					cout << "NO" << '\n';
					break;
				}
			}
			int flag = 0;
			//貨架上沒有 去倉庫找
			while (!order.empty()) {
				if (order.front() == i) {
					order.pop();
					break;//將顏色使用掉
				}
				else {
					//取出來放到貨架上
					int color = order.front();
					order.pop();
					huoja.push_back(color);
				}
				if (huoja.size() > m) {
					flag = 1;
					cout << "NO" << '\n';
					break;
				}
			}
			if (flag) break;
		}
		if (i > n) {
			cout << "YES" << '\n';
		}
	}
	return 0;
}

相關文章