純模擬,一次就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;
}