题意:
序列p为1~n排列的一种,数组b由序列p构造而成,𝑏𝑖=max(𝑝2𝑖−1,𝑝2𝑖) for 1≤𝑖≤𝑛/2。给定n和数组b,求字典序最小的序列p并输出;如果序列不存在,则输出-1。
思路:
先将b中元素填入ans,ans2𝑖=𝑏𝑖(1≤𝑖≤𝑛/2),从大到小遍历i(n~1)填数到ans中构造p序列,如果i在b中出现过则进队,没出现过则判断是否队空,若是则表示p序列不存在,否则取队头元素,该数为b数组中已遍历过的数在ans数组的位置的最大值,将i放入该数前一位置。
代码:
#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
const int N = 200001;
int main() {
int t, n;
cin>>t;
while(t--) {
cin>>n;
int ans[N] = {0}, b[N] = {0};
bool flag = true;
for(int i=2; i<=n; i+=2) {
cin>>ans[i]; //填入ans
b[ans[i]] = i; //记录位置
}
priority_queue<int> q;
for(int i=n; i>0; i--) {
if(b[i])
q.push(b[i]); //i在b中进队
else {
if(q.empty()) { //不满足ans2i=bi(1≤i≤n/2)
flag = false;
break;
}
ans[q.top()-1] = i;
q.pop();
}
}
if(flag) {
for(int i=1; i<=n; i++)
cout<<ans[i]<<' ';
}
else
cout<<"-1";
cout<<endl;
}
}