#include <bits/stdc++.h>
using namespace std;
#define max3(a, b, c) max(a, max(b, c))
#define min3(a, b, c) min(a, min(b, c))
#define mp make_pair
#define f first
#define se second
#define pb push_back
#define ppb pop_back
#define ll long long
#define ull unsigned long long
#define cntbit(x) __builtin_popcount(x)
#define uset unordered_set
#define umap unordered_map
#define pii pair<int, int>
#define ld long double
#define pll pair<long long, long long>
const int inf = 2e9;
const int N = 1e6 + 15;
int n, k, a[N];
struct node {
node *prev, *next;
int val;
node(int val) : val(val) {
prev = next = NULL;
}
};
typedef node* pnode;
map <int, vector <pnode> > is;
vector <pnode> fr;
pnode root, last;
stack <pair <int, pnode> > st, temp;
void processnext() {
while(last) {
int now = last->val;
while(!st.empty() && st.top().f == now) {
st.pop();
++now;
}
while(k && !st.empty() && st.top().f < now) {
pnode x = new node(st.top().f);
x->prev = st.top().se;
x->next = st.top().se->next;
if(x->next)
x->next->prev = x;
if(x->prev)
x->prev->next = x;
fr.pb(x);
int curval = x->val;
while(!st.empty() && st.top().f == curval)
st.pop(), ++curval;
st.push(mp(curval, x));
--k;
}
while(!st.empty() && st.top().f == now) {
st.pop();
++now;
}
st.push(mp(now, last));
last = last->next;
}
}
void addelse() {
while(!st.empty()) {
int now = st.top().f;
temp.push(st.top());
st.pop();
while(!st.empty() && st.top().f > now + 1) {
int need = st.top().f - 1;
pnode x = new node(need);
x->prev = st.top().se;
x->next = st.top().se->next;
if(x->next)
x->next->prev = x;
if(x->prev)
x->prev->next = x;
fr.pb(x);
st.push(mp(need, x));
--k;
}
}
}
void checkend() {
deque <pair <int, pnode> > s;
while(!temp.empty()) {
st.push(temp.top());
s.push_back(temp.top());
temp.pop();
}
for(int i = 0; i < n; ++i) {
if(s.size() >= 2 && s[0].f == s[1].f) {
int x = s[0].f;
cout << x << endl;
pnode d = s[0].se;
s.pop_front();
s.pop_front();
s.push_front(mp(x + 1, d));
}
if(s.size() >= 2 && s[s.size() - 1].f == s[s.size() - 2].f) {
int x = s.back().f;
cout << x << endl;
pnode d = s.back().se;
s.pop_back();
s.pop_back();
s.push_back(mp(x + 1, d));
}
}
while(!s.empty() && s.front().f != 30) {
if(s.front() < s.back()) {
last = s.front().se;
pnode x = new node(s.front().f);
x->next = last;
int now = x->val;
if(x->next)
x->next->prev = x;
fr.pb(x);
--k;
while(!s.empty() && s.front().f == now) {
s.pop_front();
++now;
}
}
else {
last = s.back().se;
pnode x = new node(s.back().f);
x->prev = last;
int now = x->val;
if(x->prev)
x->prev->next = x;
fr.pb(x);
--k;
while(!s.empty() && s.back().f == now) {
s.pop_back();
++now;
}
}
}
}
void extend_fr() {
while(!fr.empty() && k) {
pnode x = fr.back();
fr.ppb();
if(x->val <= 1)
continue;
pnode a = new node(x->val - 1), b = new node(x->val - 1);
a->prev = x->prev;
a->next = b;
b->prev = a;
b->next = x->next;
if(b->next)
b->next->prev = b;
if(a->prev)
a->prev->next = a;
fr.pb(a);
fr.pb(b);
--k;
}
}
int main() {
scanf("%d%d", &n, &k);
for(int i = 1; i <= n; ++i)
scanf("%d", &a[i]);
root = new node(a[1]);
last = root;
for(int i = 2; i <= n; ++i) {
pnode x = new node(a[i]);
x->prev = last;
last->next = x;
last = x;
}
last = root;
processnext();
addelse();
checkend();
extend_fr();
while(root) {
printf("%d ", root->val);
root = root->next;
}
return 0;
}
Compilation message
zalmoxis.cpp: In function 'void processnext()':
zalmoxis.cpp:53:13: warning: this 'if' clause does not guard... [-Wmisleading-indentation]
if(x->prev)
^~
zalmoxis.cpp:55:10: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'if'
fr.pb(x);
^~
zalmoxis.cpp: In function 'void extend_fr()':
zalmoxis.cpp:162:6: warning: this 'if' clause does not guard... [-Wmisleading-indentation]
if(a->prev)
^~
zalmoxis.cpp:164:3: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'if'
fr.pb(a);
^~
zalmoxis.cpp: In function 'int main()':
zalmoxis.cpp:171:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d%d", &n, &k);
~~~~~^~~~~~~~~~~~~~~~
zalmoxis.cpp:173:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d", &a[i]);
~~~~~^~~~~~~~~~~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
258 ms |
37752 KB |
Output is correct |
| 2 |
Correct |
262 ms |
37564 KB |
Output is correct |
| 3 |
Correct |
262 ms |
37624 KB |
Output is correct |
| 4 |
Correct |
273 ms |
37704 KB |
Output is correct |
| 5 |
Correct |
258 ms |
37752 KB |
Output is correct |
| 6 |
Correct |
256 ms |
37624 KB |
Output is correct |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Correct |
259 ms |
37668 KB |
Output is correct |
| 2 |
Correct |
251 ms |
37904 KB |
Output is correct |
| 3 |
Correct |
251 ms |
37724 KB |
Output is correct |
| 4 |
Correct |
265 ms |
37624 KB |
Output is correct |
| 5 |
Correct |
258 ms |
37736 KB |
Output is correct |
| 6 |
Correct |
258 ms |
37580 KB |
Output is correct |
| 7 |
Correct |
258 ms |
37624 KB |
Output is correct |
| 8 |
Correct |
258 ms |
37752 KB |
Output is correct |
| 9 |
Correct |
248 ms |
40164 KB |
Output is correct |
| 10 |
Incorrect |
225 ms |
53732 KB |
not a zalsequence |
| 11 |
Correct |
233 ms |
47624 KB |
Output is correct |
| 12 |
Incorrect |
2 ms |
376 KB |
Unexpected end of file - int32 expected |
| 13 |
Incorrect |
2 ms |
376 KB |
Unexpected end of file - int32 expected |
| 14 |
Incorrect |
2 ms |
376 KB |
Unexpected end of file - int32 expected |
