D. Knowledge Cards（思维/模拟/优先队列）

题目

题意

给定n*m的二维方格，下标(1,1)上有k个卡牌a，卡牌的数字是一个关于k的排列。现要将卡牌，按从大到小的顺序，放置到(n,m)上。要求

• 除了(1,1),(n,m)这两个方格，其他方格不能放置超过2个卡牌。
• 从(1,1)拿出的卡牌，不能放回(1,1)
• 放进(n,m)的卡牌，不能拿出(n,m)

问能否做到，能做到输出YA，否则输出TIDAK

思路

卡牌从(1,1)拿出后，如果它不能及时放进(n,m)，则它将会占据一个方格。
如果抽出的卡牌可以放置到(n,m)，则将它移动到(n,m)。移动的过程中，需要至少有一个空位置。

因此，我们模拟从(1,1)抽取卡牌的过程
如果当前卡牌可以放置到(n,m)，则将其放置到(n,m)，前提是有一个空方格。
并在放置卡牌后，检查剩下的卡牌是否可以继续放置到(n,m)
如果当前卡牌不可以放置到(n,m)，则将其放在临时方格。

在上述过程，如果出现没有空位置的情况，则说明无法移动了，此时无解。
注意特判下，可以从(1,1)直接走到(n,m)的情况。

我们可以用最大堆维护当前放在临时位置的卡牌的最大值。用堆的大小来检查是否有空位置。

代码

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pcc pair<char, char>
#define pii pair<int, int>
#define inf 0x3f3f3f3f
const int maxn = 200010;
const int mod = 998244353;


int n, m, k;
int a[maxn];
void solve() {
    scanf("%d%d%d", &n, &m, &k);
    for (int i = 0; i < k; ++i) {
        scanf("%d", &a[i]);
    }
    int s = n * m - 3;
    int cur = k;
    priority_queue<int> q;
	for (int i = 0; i < k; ++i) {
    	int &v = a[i];
        if (v == cur && (n - 1) + (m - 1) == 2) {
            --cur;
            continue;
        }
        q.push(v);
        if ((int)q.size() > s) {
            printf("TIDAK\n");
            return;
        }
        while (!q.empty() && q.top() == cur) {
            --cur;
            q.pop();
        }
    }
    printf("YA\n");

}
int main() {
    int t = 1;
    scanf("%d", &t);
    int cas = 1;
    while (t--) {
//		printf("cas %d:\n", cas++);
        solve();
    }

}