#include <bits/stdc++.h>
#define all(x) (x).begin(),(x).end()
using namespace std;
using ll = long long;
using ld = long double;
#define int ll
#define sz(x) ((int)(x).size())
using pii = pair<int,int>;
using tii = tuple<int,int,int>;
const int nmax = 1e6 + 5, inf = 1e18 + 5;
int n;
namespace G {
vector<vector<pii>> g;
void init(int dim) { g.resize(dim, vector<pii>()); }
int newnode() { g.emplace_back(); return sz(g); }
void addedge(int x, int y, int z) {
g[x].emplace_back(y, z);
}
namespace AINT {
int aint[(nmax + nmax / 2) * 2];
void init(int node = 1, int cl = 0, int cr = n - 1) {
if(cl == cr) {
aint[node] = cl;
return;
}
int mid = cl + cr >> 1;
init(2 * node, cl, mid);
init(2 * node + 1, mid + 1, cr);
aint[node] = newnode();
addedge(aint[node], aint[2 * node], 0);
addedge(aint[node], aint[2 * node + 1], 0);
return;
}
void pushto(int val, int l, int r, int node = 1, int cl = 1, int cr = n - 1) {
if(r < cl || cr < l) return;;
if(l <= cl && cr <= r) {
addedge(val, aint[node], 0);
return;
}
int mid = cl + cr >> 1;
pushto(val, l, r, 2 * node, cl, mid);
pushto(val, l, r, 2 * node + 1, mid + 1, cr);
}
}
priority_queue<pii, vector<pii>, greater<pii>> heap;
vector<int> arriv, occ;
vector<int> dijkstra(int s = -1) {
arriv.resize(sz(g));
occ.resize(sz(g));
fill(all(occ), 0);
if(s != -1) {
fill(all(arriv), inf);
heap.emplace(0, s);
arriv[s] = 0;
}
else {
for(int i = 0; i < sz(g); i++)
heap.emplace(arriv[i], i);
}
while(!heap.empty()) {
auto [c, node] = heap.top();
heap.pop();
if(occ[node]) continue;
occ[node] = 1;
for(auto [x, e] : g[node]) {
if(arriv[x] > c + e) {
arriv[x] = c + e;
heap.emplace(arriv[x], x);
}
}
}
return arriv;
}
}
long long minimum_walk(std::vector<signed> p, signed s) {
n = sz(p);
vector<int> col(n, 0);
vector<int> rs(n + 2, 0), ls(n + 2, 0);
int flag = 0;
multiset<int> lhf;
ll sum = 0;
G::init(n);
for(int i = 0; i < n; i++) {
if(col[i] == 0) {
flag++;
int x = i, cnt = 0;
set<int> vals;
while(col[x] != flag) {
col[x] = flag;
sum += abs(x - p[x]);
x = p[x];
cnt++;
vals.emplace(x);
}
int L = *vals.begin();
int R = *vals.rbegin();
flag++;
x = i;
while(col[x] != flag) {
col[x] = flag;
G::AINT::pushto(x, L, R);
x = p[x];
}
}
if(i > 0)
G::addedge(i - 1, i, 2),
G::addedge(i, i - 1, 2);
}
int sl = 0, sr = n - 1;
while(p[sl] == sl && sl < s) sl++;
while(p[sr] == sr && sr > s) sr--;
cerr << sr << ' ' << p[sr] << '\n';
auto fl = G::dijkstra(sl), fr = G::dijkstra(sr);
for(int i = 0; i < sz(G::g); i++) {
G::arriv[i] = fl[i] + fr[i];
}
auto rez = G::dijkstra();
return sum + rez[s];
}
#undef int
/**
Anul asta se da centroid.
-- Surse oficiale
*/
Compilation message
books.cpp: In function 'void G::AINT::init(ll, ll, ll)':
books.cpp:33:20: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
33 | int mid = cl + cr >> 1;
| ~~~^~~~
books.cpp: In function 'void G::AINT::pushto(ll, ll, ll, ll, ll, ll)':
books.cpp:47:20: warning: suggest parentheses around '+' inside '>>' [-Wparentheses]
47 | int mid = cl + cr >> 1;
| ~~~^~~~
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
0 ms |
212 KB |
3rd lines differ - on the 1st token, expected: '6', found: '4' |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
0 ms |
212 KB |
3rd lines differ - on the 1st token, expected: '6', found: '4' |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
0 ms |
212 KB |
3rd lines differ - on the 1st token, expected: '6', found: '4' |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
2 ms |
468 KB |
3rd lines differ - on the 1st token, expected: '3304', found: '3594' |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
Verdict |
Execution time |
Memory |
Grader output |
| 1 |
Incorrect |
0 ms |
212 KB |
3rd lines differ - on the 1st token, expected: '6', found: '4' |
| 2 |
Halted |
0 ms |
0 KB |
- |
