Restoring the Permutation

cf题目链接
题意简述：原有一序列p，p中元素为[1,n],通过q[i]=max(p[1],p[2]…,p[i])得到一个新的序列q，现给出q，问满足规则的字典序最大和最小的p序列分别是什么

解法：通过规则，我们可以确定的是在q序列中第一次出现的数x，x在原序列p中也处于该位置(q[i]=max(p[1],p[2]…,p[i]))
字典序最小：我们把1-n中除了出现在q序列的数从小到大填入p序列的空位即可
字典序最大：我们先将确定好的q序列的数填入了p序列，那么我们将小于要填位置之前的第一个非空p序列的数从大到小填入即可

AcCode：

#include<iostream>
#include<algorithm>
#include<queue>
#include<map>
#include<set>

#define int long long

using namespace std;

const int N = 3e5 + 100;

int arr1[N], arr2[N], vis[N];

signed main() {
	ios::sync_with_stdio(false);
	int t;
	cin >> t;
	while (t--) {
		int n;
		cin >> n;
		set<int> st;//记录q序列的数
		set<int> init;//记录通过q序列确定的位置
		set<int> num;//记录未填入的1-n的数
		for (int i = 1; i <= n; i++) {
			cin >> arr1[i];
			arr2[i] = arr1[i];
			if (arr1[i] != arr1[i - 1]) {
				vis[i] = 1;
				st.insert(arr1[i]);
				init.insert(i);
			}
		}
		for (int i = 1; i <= n; i++) if (!st.count(i)) num.insert(i);
		set<int>::iterator iter;
		iter = num.begin();
		for (int i = 1; i <= n && iter != num.end(); i++) {//字典序最小
			if (vis[i]) continue;
			arr1[i] = *iter;
			iter++;
		}
		int rem = 0;
		for (int i = 1; i <= n; i++) {//字典序最大
			if (vis[i]) {
				rem = arr2[i];
				continue;
			}
			auto v = lower_bound(num.begin(), num.begin(), rem);
			iter = num.lower_bound(rem);
			if (iter != num.begin()) iter--;//得到未填入的符合条件的最大数
			arr2[i] = *iter;
			num.erase(iter);
		}
		for (int i = 1; i <= n; i++) cout << arr1[i] << " ";
		cout << endl;
		for (int i = 1; i <= n; i++) cout << arr2[i] << " ";
		cout << endl;
		for (auto v : init) vis[v] = 0;
	}
}