codeforces 1506 E Restoring the Permutation（优先队列）

题面

题意

给定一个序列q,它是通过p序列变化得来的，让你求p序列的字典序最小和最大
假设原序列为p，q[i]=max(p[1],p[2]…p[i])

题解

1. 我们分析字典序最小的，字典序最大的同理。对于字典序最小的，第一个肯定是q[1],对于q[2],如果等于q[1]的话，说明p[2]的值没有p[1]大，如果q[2]!=q[1],说明p[2]>p[1]

2. 那么对于字典序最小的序列，从前向后遍历q序列，如果是第一个出现，那么q[i]=p[i],然后用优先队列存小于q[i]的数，如果不是第一次出现，说明这个p[i]就可以用优先队列中的数填补

代码

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>

using namespace std;
const int N = 2e5 + 10;

int t, n;
int q[N];
int minn[N], maxn[N];
int st[N];

int main() {
    
    

    cin >> t;
    while (t--) {
    
    
        memset(st, 0, sizeof st);
        cin >> n;
        for (int i = 1; i <= n; i++) cin >> q[i];

        priority_queue<int, vector<int>, greater<int>> pmin;
        priority_queue<int> pmax;

        int index = 1;
        for (int i = 1; i <= n; i++) {
    
    

            while (index < q[i] && st[index] == 0) {
    
    
                pmin.push(index);
                pmax.push(index);
                index++;
            }
            index = q[i] + 1;
            if (st[q[i]] == 0) {
    
    
                minn[i] = maxn[i] = q[i];
                st[q[i]] = 1;
            } else {
    
    
                minn[i] = pmin.top();
                pmin.pop();
                maxn[i] = pmax.top();
                pmax.pop();
            }
        }

        for (int i = 1; i <= n; i++) cout << minn[i] << " ";
        cout << endl;
        for (int i = 1; i <= n; i++) cout << maxn[i] << " ";
        cout << endl;
    }
    return 0;
}