#include <bits/stdc++.h>
using namespace std;
vector<int> v[1000001];
int n,k,arr[1000001];
void solve(int l , int r , int val , int freq){
if(r == l-1){
for(int i = 0 ; i < 31 ; i += 1){
if(freq & (1<<i)){
v[r].push_back(val+i);
k -= 1;
}
}
return ;
}
int pos = -1;
for(int i = l ; i <= r ; i += 1){
if(arr[i] == val){
pos = i;
break;
}
}
if(pos == -1){
solve(l,r,val-1,2*freq);
return ;
}
int farr[31];
memset(farr,0,sizeof farr);
for(int i = l ; i < pos ; i += 1){
farr[arr[i]] += 1;
}
int tag = 0;
for(int i = 0 ; i < val ; i += 1){
farr[i+1] += farr[i]/2;
if(farr[i]%2 == 1){
tag = 1;
}
}
freq -= 1;
solve(l,pos-1,val,farr[val]+tag),solve(pos+1,r,val,freq-farr[val]-tag);
}
int main(){
cin >> n >> k;
for(int i = 1 ; i <= n ; i += 1){
cin >> arr[i];
}
solve(1,n,30,1);
vector<pair<int,int>> varr;
for(int i = 0 ; i <= n ; i += 1){
if(i){
varr.push_back({arr[i],1});
}
for(auto j : v[i]){
varr.push_back({j,0});
}
}
vector<int> ans;
while(k > 0){
if(varr.back().second == 1){
ans.push_back(varr.back().first);
varr.pop_back();
continue;
}
if(varr.back().first > 0){
k -= 1;
int a = varr.back().first;
varr.pop_back();
varr.push_back({a-1,0}),varr.push_back({a-1,0});
}else{
ans.push_back(varr.back().first);
varr.pop_back();
}
}
while(varr.size()){
ans.push_back(varr.back().first);
varr.pop_back();
}
reverse(ans.begin(),ans.end());
for(auto i : ans){
assert(i <= 30);
cout << i << " ";
}cout << endl;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
476 ms |
45604 KB |
Output is correct |
2 |
Correct |
458 ms |
45604 KB |
Output is correct |
3 |
Correct |
475 ms |
45544 KB |
Output is correct |
4 |
Correct |
488 ms |
45648 KB |
Output is correct |
5 |
Correct |
475 ms |
45544 KB |
Output is correct |
6 |
Correct |
498 ms |
45604 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
472 ms |
45572 KB |
Output is correct |
2 |
Runtime error |
536 ms |
89648 KB |
Execution killed with signal 6 |
3 |
Correct |
459 ms |
45620 KB |
Output is correct |
4 |
Correct |
469 ms |
45700 KB |
Output is correct |
5 |
Correct |
489 ms |
45668 KB |
Output is correct |
6 |
Correct |
516 ms |
45552 KB |
Output is correct |
7 |
Correct |
467 ms |
45640 KB |
Output is correct |
8 |
Correct |
526 ms |
45556 KB |
Output is correct |
9 |
Runtime error |
399 ms |
109328 KB |
Execution killed with signal 6 |
10 |
Runtime error |
186 ms |
86360 KB |
Execution killed with signal 6 |
11 |
Runtime error |
293 ms |
104948 KB |
Execution killed with signal 6 |
12 |
Correct |
141 ms |
29744 KB |
Output is correct |
13 |
Correct |
142 ms |
29748 KB |
Output is correct |
14 |
Correct |
137 ms |
29736 KB |
Output is correct |