#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i = a; i<int(b);++i)
#define all(v) v.begin(),v.end()
#define sz(v) v.size()
#define trav(a,c) for(auto a: c)
typedef long long ll;
typedef vector<ll> vi;
typedef pair<ll,ll> pii;
int main(){
cin.sync_with_stdio(false);
ll START = 30;
ll n,k;
cin>>n>>k;
vi z(n);
rep(i,0,n) cin>>z[i];
/*vi z = {START};
rep(i,0,n+k-1){
ll rIndex = rand()%z.size();
ll rVal = z[rIndex];
z.erase(z.begin()+rIndex);
z.insert(z.begin()+rIndex,rVal-1);
z.insert(z.begin()+rIndex,rVal-1);
}
rep(i,0,z.size()) cout<<z[i]<<" ";
cout<<endl;
rep(i,0,k){
z.erase(z.begin()+rand()%z.size());
}
rep(i,0,z.size()) cout<<z[i]<<" ";
cout<<endl;*/
vi s;
s.push_back(z[0]);
vector<vi> inserts(n);
rep(i,1,n){
while(z[i]>s[s.size()-1]){
inserts[i-1].push_back(s[s.size()-1]);
--k;
s[s.size()-1]++;
while(s.size()>=2&&s[s.size()-2]==s[s.size()-1]){
s.pop_back();
s[s.size()-1]++;
}
}
s.push_back(z[i]);
while(s.size()>=2&&s[s.size()-2]==s[s.size()-1]){
s.pop_back();
s[s.size()-1]++;
}
}
while(s[0]!=START){
inserts[n-1].push_back(s[s.size()-1]);
--k;
s[s.size()-1]++;
while(s.size()>=2&&s[s.size()-2]==s[s.size()-1]){
s.pop_back();
s[s.size()-1]++;
}
}
//k+=10;
rep(i,0,n){
list<ll> newInserts;
list<ll>::iterator it = newInserts.end();
rep(j,0,inserts[i].size()){
if(it == newInserts.end()){
newInserts.push_back(inserts[i][j]);
it = prev(newInserts.end());
}else{
newInserts.push_back(inserts[i][j]);
}
while(k&&*it>0){
--k;
(*it)--;
newInserts.insert(next(it),*it);
while(it!=newInserts.end()&&(*it)==0)
++it;
}
}
inserts[i].clear();
for(auto e:newInserts)
inserts[i].push_back(e);
}
/*if(k!=0){
rep(i,0,n)
rep(j,0,inserts[i].size())
assert(inserts[i][j]==0);
}*/
assert(k==0);
rep(i,0,n){
cout<<z[i]<<" ";
rep(j,0,inserts[i].size()) cout<<inserts[i][j]<<" ";
}
cout<<endl;
/*cout<<endl;
rep(i,0,n) {
cout<<i<<":";
rep(j,0,inserts[i].size()) cout<<inserts[i][j]<<" ";
cout<<endl;
}*/
}
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
227 ms |
33760 KB |
Output is correct |
2 |
Correct |
218 ms |
33808 KB |
Output is correct |
3 |
Correct |
233 ms |
33856 KB |
Output is correct |
4 |
Correct |
192 ms |
33904 KB |
Output is correct |
5 |
Correct |
211 ms |
33904 KB |
Output is correct |
6 |
Correct |
217 ms |
33904 KB |
Output is correct |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
254 ms |
33904 KB |
Output is correct |
2 |
Correct |
253 ms |
33972 KB |
Output is correct |
3 |
Correct |
246 ms |
33972 KB |
Output is correct |
4 |
Correct |
214 ms |
33988 KB |
Output is correct |
5 |
Correct |
310 ms |
33988 KB |
Output is correct |
6 |
Correct |
221 ms |
34172 KB |
Output is correct |
7 |
Correct |
220 ms |
34172 KB |
Output is correct |
8 |
Correct |
205 ms |
34172 KB |
Output is correct |
9 |
Incorrect |
293 ms |
34172 KB |
not a zalsequence |
10 |
Incorrect |
209 ms |
38884 KB |
not a zalsequence |
11 |
Incorrect |
228 ms |
38884 KB |
not a zalsequence |
12 |
Incorrect |
174 ms |
42924 KB |
not a zalsequence |
13 |
Incorrect |
165 ms |
42924 KB |
not a zalsequence |
14 |
Incorrect |
237 ms |
42924 KB |
not a zalsequence |