#include <bits/stdc++.h>
using namespace std;
const int maxn = 200000;
int n, A[maxn+10];
vector<int> memo[maxn+10][36];
void merge(vector<int> &res, int root, vector<int> &L, vector<int> &R) {
res.push_back(root);
int lidx = 0, ridx = 0;
int cur = 1;
while (lidx < L.size() || ridx < R.size()) {
for (int i = 0; i < cur; i++) {
if (lidx < L.size()) res.push_back(L[lidx++]);
}
for (int i = 0; i < cur; i++) {
if (ridx < R.size()) res.push_back(R[ridx++]);
}
cur *= 2;
}
}
vector<int> &dp(int u, int par) {
if (memo[u][par].size() > 0) return memo[u][par];
int idx = u, p = par;
while (p >= 2) {
p -= 2;
idx /= 2;
}
if (p == 1) idx *= 2;
int cur = A[idx];
int l = u*2, r = u*2+1;
if (l > n && r > n) {
memo[u][par].push_back(cur);
return memo[u][par];
} else if (r > n) {
if (A[l] < cur) {
memo[u][par].push_back(A[l]);
memo[u][par].push_back(cur);
} else {
memo[u][par].push_back(cur);
memo[u][par].push_back(A[l]);
}
return memo[u][par];
}
int L = A[l], R = A[r];
// no swap
if (cur < L && cur < R) {
merge(memo[u][par], cur, dp(l, 0), dp(r, 0));
return memo[u][par];
}
// swap left
if (cur > L && L < R) {
merge(memo[u][par], L, dp(l, par+2), dp(r, 0));
return memo[u][par];
}
if (R < cur && R < L) {
// swap right
merge(memo[u][par], R, dp(l, 0), dp(r, par+2));
vector<int> tmp;
// swap both
merge(tmp, R, dp(l, par+2), dp(r, 3));
if (tmp < memo[u][par]) {
memo[u][par] = tmp;
}
return memo[u][par];
}
assert(false);
}
int main() {
cin.tie(0); ios_base::sync_with_stdio(0);
cin >> n; for (int i = 1; i <= n; i++) cin >> A[i];
vector<int> ans = dp(1, 0);
for (int v : ans) cout << v << " ";
cout << "\n";
return 0;
}
Compilation message
swap.cpp: In function 'void merge(std::vector<int>&, int, std::vector<int>&, std::vector<int>&)':
swap.cpp:14:17: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
while (lidx < L.size() || ridx < R.size()) {
~~~~~^~~~~~~~~~
swap.cpp:14:36: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
while (lidx < L.size() || ridx < R.size()) {
~~~~~^~~~~~~~~~
swap.cpp:16:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
if (lidx < L.size()) res.push_back(L[lidx++]);
~~~~~^~~~~~~~~~
swap.cpp:19:22: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
if (ridx < R.size()) res.push_back(R[ridx++]);
~~~~~^~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
100 ms |
169464 KB |
Output is correct |
2 |
Correct |
103 ms |
169592 KB |
Output is correct |
3 |
Correct |
99 ms |
169464 KB |
Output is correct |
4 |
Correct |
103 ms |
169464 KB |
Output is correct |
5 |
Correct |
117 ms |
169464 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
100 ms |
169464 KB |
Output is correct |
2 |
Correct |
103 ms |
169592 KB |
Output is correct |
3 |
Correct |
99 ms |
169464 KB |
Output is correct |
4 |
Correct |
103 ms |
169464 KB |
Output is correct |
5 |
Correct |
117 ms |
169464 KB |
Output is correct |
6 |
Correct |
114 ms |
169464 KB |
Output is correct |
7 |
Correct |
105 ms |
169464 KB |
Output is correct |
8 |
Correct |
112 ms |
169464 KB |
Output is correct |
9 |
Correct |
113 ms |
169464 KB |
Output is correct |
10 |
Correct |
103 ms |
169560 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
100 ms |
169464 KB |
Output is correct |
2 |
Correct |
103 ms |
169592 KB |
Output is correct |
3 |
Correct |
99 ms |
169464 KB |
Output is correct |
4 |
Correct |
103 ms |
169464 KB |
Output is correct |
5 |
Correct |
117 ms |
169464 KB |
Output is correct |
6 |
Correct |
114 ms |
169464 KB |
Output is correct |
7 |
Correct |
105 ms |
169464 KB |
Output is correct |
8 |
Correct |
112 ms |
169464 KB |
Output is correct |
9 |
Correct |
113 ms |
169464 KB |
Output is correct |
10 |
Correct |
103 ms |
169560 KB |
Output is correct |
11 |
Correct |
100 ms |
169592 KB |
Output is correct |
12 |
Correct |
103 ms |
169596 KB |
Output is correct |
13 |
Correct |
104 ms |
169592 KB |
Output is correct |
14 |
Correct |
106 ms |
169848 KB |
Output is correct |
15 |
Correct |
118 ms |
169848 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
100 ms |
169464 KB |
Output is correct |
2 |
Correct |
103 ms |
169592 KB |
Output is correct |
3 |
Correct |
99 ms |
169464 KB |
Output is correct |
4 |
Correct |
103 ms |
169464 KB |
Output is correct |
5 |
Correct |
117 ms |
169464 KB |
Output is correct |
6 |
Correct |
114 ms |
169464 KB |
Output is correct |
7 |
Correct |
105 ms |
169464 KB |
Output is correct |
8 |
Correct |
112 ms |
169464 KB |
Output is correct |
9 |
Correct |
113 ms |
169464 KB |
Output is correct |
10 |
Correct |
103 ms |
169560 KB |
Output is correct |
11 |
Correct |
100 ms |
169592 KB |
Output is correct |
12 |
Correct |
103 ms |
169596 KB |
Output is correct |
13 |
Correct |
104 ms |
169592 KB |
Output is correct |
14 |
Correct |
106 ms |
169848 KB |
Output is correct |
15 |
Correct |
118 ms |
169848 KB |
Output is correct |
16 |
Correct |
160 ms |
178612 KB |
Output is correct |
17 |
Correct |
187 ms |
178292 KB |
Output is correct |
18 |
Correct |
185 ms |
178868 KB |
Output is correct |
19 |
Correct |
433 ms |
209648 KB |
Output is correct |
20 |
Correct |
467 ms |
209392 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
100 ms |
169464 KB |
Output is correct |
2 |
Correct |
103 ms |
169592 KB |
Output is correct |
3 |
Correct |
99 ms |
169464 KB |
Output is correct |
4 |
Correct |
103 ms |
169464 KB |
Output is correct |
5 |
Correct |
117 ms |
169464 KB |
Output is correct |
6 |
Correct |
114 ms |
169464 KB |
Output is correct |
7 |
Correct |
105 ms |
169464 KB |
Output is correct |
8 |
Correct |
112 ms |
169464 KB |
Output is correct |
9 |
Correct |
113 ms |
169464 KB |
Output is correct |
10 |
Correct |
103 ms |
169560 KB |
Output is correct |
11 |
Correct |
100 ms |
169592 KB |
Output is correct |
12 |
Correct |
103 ms |
169596 KB |
Output is correct |
13 |
Correct |
104 ms |
169592 KB |
Output is correct |
14 |
Correct |
106 ms |
169848 KB |
Output is correct |
15 |
Correct |
118 ms |
169848 KB |
Output is correct |
16 |
Correct |
160 ms |
178612 KB |
Output is correct |
17 |
Correct |
187 ms |
178292 KB |
Output is correct |
18 |
Correct |
185 ms |
178868 KB |
Output is correct |
19 |
Correct |
433 ms |
209648 KB |
Output is correct |
20 |
Correct |
467 ms |
209392 KB |
Output is correct |
21 |
Correct |
349 ms |
209536 KB |
Output is correct |
22 |
Correct |
352 ms |
210308 KB |
Output is correct |
23 |
Correct |
345 ms |
210560 KB |
Output is correct |
24 |
Runtime error |
741 ms |
262148 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |
25 |
Halted |
0 ms |
0 KB |
- |