This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <bits/stdc++.h>
using namespace std;
struct xxx {
int l, r, val, ap;
bool operator < ( const xxx &x ) const {
return val < x.val;
}
};
const int MAX_N = 1e6;
int v[MAX_N + 1], leftt[MAX_N + 1], rightt[MAX_N + 1], value[MAX_N + 1], apar[MAX_N + 1];
set<xxx> seg;
vector<int> adaugiri[MAX_N + 1];
int main() {
int n, k;
cin >> n >> k;
for ( int i = 1; i <= n; i++ )
cin >> v[i];
int l = 1, r = 1;
while ( l <= n ) {
r = l;
while ( r + 1 <= n && v[l] == v[r + 1] )
r++;
leftt[r] = l;
rightt[l] = r;
value[l] = value[r] = v[l];
apar[l] = apar[r] = r - l + 1;
seg.insert( { l, r, v[l], r - l + 1 } );
l = r + 1;
}
while ( (!(seg.size() == 1 && seg.begin()->ap == 1)) ) {
int l = seg.begin()->l, r = seg.begin()->r, val = seg.begin()->val, ap = seg.begin()->ap;
seg.erase( { l, r, val, ap } );
if ( ap % 2 == 1 ) {
ap++;
k--;
adaugiri[r].push_back( val );
}
ap /= 2;
val++;
if ( val == value[l - 1] ) {
ap += apar[l - 1];
seg.erase( { leftt[l - 1], rightt[l - 1], value[l - 1], apar[l - 1] } );
l = leftt[l - 1];
}
if ( val == value[r + 1] ) {
ap += apar[r + 1];
seg.erase( { leftt[r + 1], rightt[r + 1], value[r + 1], apar[r + 1] } );
r = rightt[r + 1];
}
leftt[r] = l;
rightt[l] = r;
value[l] = value[r] = val;
apar[l] = apar[r] = ap;
seg.insert( { l, r, val, ap } );
}
for ( int i = 1; i <= n; i++ ) {
cout << v[i] << " ";
for ( int x: adaugiri[i] )
cout << x << " ";
}
int val = seg.begin()->val;
for ( int i = 0; i < k; i++ )
cout << val + i << " ";
return 0;
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |