洛谷 P5375 【[THUPC2019]组合数据结构问题】题解

这个题看上去很有高论，但其实没有那么复杂。只需要分别模拟队列、栈、大根堆和小根堆的处理过程即可。

细节和注释都在代码里了。

$ m code $

# pragma GCC optimize (2)
# include <bits/stdc++.h>
using namespace std;
queue<int> que;
stack<int> stk;
priority_queue<int> heap1;
priority_queue<int, vector<int>, greater<int> > heap2;
// 使用 C++ 的 STL 库，分别定义四个数据结构：队列、栈、大根堆和小根堆.
int main() {
    ios::sync_with_stdio(0), cin.tie(0);
    int N; bool flag[5];
    memset(flag, true, sizeof(flag));
    cin >> N;
    while(N--) {
        int opt, x;
        cin >> opt >> x;
        if(opt == 1) {
            if(flag[1]) que.push(x);
            if(flag[2]) stk.push(x);
            if(flag[3]) heap1.push(x);
            if(flag[4]) heap2.push(x);
            // 如果黑箱里暂时可能是这个数据结构的话且opt为1，那么就插入x
            // 这里模拟的是插入操作.
        }
        else {
            if(que.empty()) flag[1] = false;
            if(stk.empty()) flag[2] = false;
            if(heap1.empty()) flag[3] = false;
            if(heap2.empty()) flag[4] = false;
            // 如果该数据结构已经为空，直接把flag改一下
            if(flag[1]) {
                if(que.front() != x) flag[1] = false;
                que.pop();
            }
            if(flag[2]) {
                if(stk.top() != x) flag[2] = false;
                stk.pop();
            }
            if(flag[3]) {
                if(heap1.top() != x) flag[3] = false;
                heap1.pop();
            }
            if(flag[4]) {
                if(heap2.top() != x) flag[4] = false;
                heap2.pop();
            }
            // 模拟删除操作.
        }
    }
    for(int i = 1; i <= 4; ++i)
        flag[i] ? puts("Yes") : puts("No"); // 输出
    return 0;
}