// Hallelujah, praise the one who set me free
// Hallelujah, death has lost its grip on me
// You have broken every chain, There's salvation in your name
// Jesus Christ, my living hope
#include <bits/stdc++.h>
#include "messy.h"
using namespace std;
#define REP(i, s, e) for (int i = (s); i < (e); i++)
#define RREP(i, s, e) for (int i = (s); i >= (e); i--)
template <class T>
inline bool mnto(T& a, T b) {return a > b ? a = b, 1 : 0;}
template <class T>
inline bool mxto(T& a, T b) {return a < b ? a = b, 1: 0;}
typedef unsigned long long ull;
typedef long long ll;
typedef long double ld;
#define FI first
#define SE second
typedef pair<int, int> ii;
typedef pair<ll, ll> pll;
typedef tuple<int, int, int> iii;
#define ALL(_a) _a.begin(), _a.end()
#define SZ(_a) (int) _a.size()
#define pb push_back
typedef vector<int> vi;
typedef vector<ll> vll;
typedef vector<ii> vii;
typedef vector<iii> viii;
#ifndef DEBUG
#define cerr if (0) cerr
#endif
const int INF = 1000000005;
const ll LINF = 1000000000000000005ll;
const int MAXN = 130;
int n;
vi ans;
void encode(int l, int r, int tl, int tr) {
int m = l + r >> 1;
string s(n, '0');
REP (i, tl, tr) {
s[i] = '1';
}
REP (i, l, m) {
string t = s;
t[i] = '1';
add_element(t);
}
if (l + 1 < r) {
encode(l, m, m, r); encode(m, r, l, m);
}
}
void decode(int l, int r, vi vt, vi id) {
if (l + 1 == r) {
ans[id[0]] = l;
return;
}
int m = l + r >> 1;
string s(n, '0');
for (int i : vt) {
s[i] = '1';
}
vi lv, rv;
for (int i : id) {
string t = s;
t[i] = '1';
if (check_element(t)) {
lv.pb(i);
} else {
rv.pb(i);
}
}
decode(l, m, rv, lv); decode(m, r, lv, rv);
}
vi restore_permutation(int N, int W, int R) {
n = N;
encode(0, n, 0, 0);
compile_set();
vi id(n);
iota(ALL(id), 0);
ans = vi(n);
decode(0, n, vi(), id);
return ans;
}
Compilation message
messy.cpp: In function 'void encode(int, int, int, int)':
messy.cpp:45:15: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
45 | int m = l + r >> 1;
| ~~^~~
messy.cpp: In function 'void decode(int, int, vi, vi)':
messy.cpp:64:15: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
64 | int m = l + r >> 1;
| ~~^~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
0 ms |
348 KB |
n = 8 |
2 |
Correct |
0 ms |
436 KB |
n = 8 |
3 |
Correct |
0 ms |
348 KB |
n = 8 |
4 |
Correct |
1 ms |
348 KB |
n = 8 |
5 |
Correct |
0 ms |
436 KB |
n = 8 |
6 |
Correct |
1 ms |
348 KB |
n = 8 |
7 |
Correct |
0 ms |
348 KB |
n = 8 |
8 |
Correct |
1 ms |
348 KB |
n = 8 |
9 |
Correct |
1 ms |
436 KB |
n = 8 |
10 |
Correct |
1 ms |
348 KB |
n = 8 |
11 |
Correct |
1 ms |
348 KB |
n = 8 |
12 |
Correct |
1 ms |
348 KB |
n = 8 |
13 |
Correct |
0 ms |
348 KB |
n = 8 |
14 |
Correct |
0 ms |
348 KB |
n = 8 |
15 |
Correct |
1 ms |
348 KB |
n = 8 |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
344 KB |
n = 32 |
2 |
Correct |
1 ms |
348 KB |
n = 32 |
3 |
Correct |
0 ms |
348 KB |
n = 32 |
4 |
Correct |
0 ms |
364 KB |
n = 32 |
5 |
Correct |
0 ms |
360 KB |
n = 32 |
6 |
Correct |
1 ms |
344 KB |
n = 32 |
7 |
Correct |
0 ms |
348 KB |
n = 32 |
8 |
Correct |
1 ms |
352 KB |
n = 32 |
9 |
Correct |
1 ms |
348 KB |
n = 32 |
10 |
Correct |
1 ms |
444 KB |
n = 32 |
11 |
Correct |
1 ms |
360 KB |
n = 32 |
12 |
Correct |
1 ms |
360 KB |
n = 32 |
13 |
Correct |
0 ms |
360 KB |
n = 32 |
14 |
Correct |
0 ms |
356 KB |
n = 32 |
15 |
Correct |
0 ms |
348 KB |
n = 32 |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
348 KB |
n = 32 |
2 |
Correct |
0 ms |
348 KB |
n = 32 |
3 |
Correct |
1 ms |
348 KB |
n = 32 |
4 |
Correct |
1 ms |
348 KB |
n = 32 |
5 |
Correct |
1 ms |
348 KB |
n = 32 |
6 |
Correct |
1 ms |
348 KB |
n = 32 |
7 |
Correct |
1 ms |
756 KB |
n = 32 |
8 |
Correct |
1 ms |
348 KB |
n = 32 |
9 |
Correct |
0 ms |
348 KB |
n = 32 |
10 |
Correct |
0 ms |
348 KB |
n = 32 |
11 |
Correct |
1 ms |
348 KB |
n = 32 |
12 |
Correct |
1 ms |
344 KB |
n = 32 |
13 |
Correct |
0 ms |
348 KB |
n = 32 |
14 |
Correct |
0 ms |
348 KB |
n = 32 |
15 |
Correct |
1 ms |
344 KB |
n = 32 |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
600 KB |
n = 128 |
2 |
Correct |
1 ms |
604 KB |
n = 128 |
3 |
Correct |
1 ms |
768 KB |
n = 128 |
4 |
Correct |
1 ms |
604 KB |
n = 128 |
5 |
Correct |
1 ms |
436 KB |
n = 128 |
6 |
Correct |
1 ms |
604 KB |
n = 128 |
7 |
Correct |
2 ms |
604 KB |
n = 128 |
8 |
Correct |
2 ms |
604 KB |
n = 128 |
9 |
Correct |
1 ms |
604 KB |
n = 128 |
10 |
Correct |
1 ms |
604 KB |
n = 128 |
11 |
Correct |
1 ms |
604 KB |
n = 128 |
12 |
Correct |
2 ms |
604 KB |
n = 128 |
13 |
Correct |
1 ms |
600 KB |
n = 128 |
14 |
Correct |
2 ms |
604 KB |
n = 128 |
15 |
Correct |
1 ms |
436 KB |
n = 128 |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
1 ms |
604 KB |
n = 128 |
2 |
Correct |
2 ms |
760 KB |
n = 128 |
3 |
Correct |
1 ms |
600 KB |
n = 128 |
4 |
Correct |
2 ms |
436 KB |
n = 128 |
5 |
Correct |
1 ms |
604 KB |
n = 128 |
6 |
Correct |
1 ms |
608 KB |
n = 128 |
7 |
Correct |
1 ms |
432 KB |
n = 128 |
8 |
Correct |
1 ms |
600 KB |
n = 128 |
9 |
Correct |
2 ms |
504 KB |
n = 128 |
10 |
Correct |
1 ms |
600 KB |
n = 128 |
11 |
Correct |
1 ms |
604 KB |
n = 128 |
12 |
Correct |
1 ms |
604 KB |
n = 128 |
13 |
Correct |
1 ms |
604 KB |
n = 128 |
14 |
Correct |
1 ms |
600 KB |
n = 128 |
15 |
Correct |
1 ms |
604 KB |
n = 128 |