#include <bits/stdc++.h>
using namespace std;
#define fr first
#define sc second
#define ll long long
const int mxn=1e5+5,szz=(1<<20);
int n,a[mxn],b[mxn],k,L[mxn],R[mxn],sz[mxn],mx[mxn];
vector<int>g[mxn],par[mxn];
set<int>s;
void solve(int m){
if(s.size()==0 && m<n+1){
cout<<-1;
exit(0);
}
if(m==n+1){
k--;
if(k==0){
for(int i=1;i<=n;i++)
cout<<a[i]<<" ";
cout<<'\n';
exit(0);
}
return ;
}
vector<int>v;
for(auto i:s){
v.push_back(i);
if(v.size()==10)
break;
}
for(auto i:v){
a[i]=m;
s.erase(i);
for(auto j:par[i]){
sz[j]--;
if(sz[j]==0)
s.insert(j);
}
solve(m+1);
s.insert(i);
for(auto j:par[i]){
sz[j]++;
if(sz[j]==1)
s.erase(j);
}
}
}
struct segment{
int val[szz];
void build(int id,int L,int R){
if(L+1==R){
val[id]=b[L];
return ;
}
int mid=(L+R)>>1;
build(id*2,L,mid);
build(id*2+1,mid,R);
val[id]=max(val[id*2],val[id*2+1]);
}
int get(int id,int L,int R,int l,int r){
if(L>=r || l>=R)
return 0;
if(l<=L && R<=r)
return val[id];
int mid=(L+R)>>1;
return max(get(id*2,L,mid,l,r),get(id*2+1,mid,R,l,r));
}
}seg;
int main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
cin>>n>>k;
int cnt=0;
vector<int>hey;
for(int i=1;i<=n;i++){
cin>>b[i];
if(b[i]==-1)hey.push_back(i);
}
if(b[n]!=-1){
cout<<-1;
return 0;
}
for(int i=1;i<hey.size();i++){
par[hey[i]].push_back(hey[i-1]);
g[hey[i-1]].push_back(hey[i]);
}
seg.build(1,1,n+1);
for(int i=0;i<hey.size();i++){
int l=(i==0)?1:hey[i-1]+1;
vector<int>v;
for(int j=hey[i]-1;j>=l;j--){
par[j].push_back(hey[i]);
g[hey[i]].push_back(j);
R[j]=j+1;
while(R[j]<hey[i] && b[R[j]]<b[j]){
if(seg.get(1,1,n+1,j+1,R[j])<b[j]){
par[R[j]].push_back(j);
g[j].push_back(R[j]);
}
R[j]=R[R[j]];
}
par[j].push_back(R[j]);
g[R[j]].push_back(j);
if(seg.get(1,1,n+1,j+1,R[j])+1!=b[j]){
cout<<-1;
return 0;
}
}
}
for(int i=1;i<=n;i++){
sz[i]=g[i].size();
if(sz[i]==0)
s.insert(i);
}
solve(1);
cout<<-1;
return 0;
}
Compilation message
Main.cpp: In function 'int main()':
Main.cpp:87:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
87 | for(int i=1;i<hey.size();i++){
| ~^~~~~~~~~~~
Main.cpp:92:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
92 | for(int i=0;i<hey.size();i++){
| ~^~~~~~~~~~~
Main.cpp:77:6: warning: unused variable 'cnt' [-Wunused-variable]
77 | int cnt=0;
| ^~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
8540 KB |
Output is correct |
| 2 |
Correct |
1 ms |
8540 KB |
Output is correct |
| 3 |
Incorrect |
1 ms |
8536 KB |
Output isn't correct |
| 4 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
8540 KB |
Output is correct |
| 2 |
Correct |
1 ms |
8540 KB |
Output is correct |
| 3 |
Incorrect |
1 ms |
8536 KB |
Output isn't correct |
| 4 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
33 ms |
26188 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
33 ms |
26188 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
1 ms |
8540 KB |
Output is correct |
| 2 |
Correct |
1 ms |
8540 KB |
Output is correct |
| 3 |
Incorrect |
1 ms |
8536 KB |
Output isn't correct |
| 4 |
Halted |
0 ms |
0 KB |
- |
