#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;
int n;
long long minimum_walk(std::vector<int> p, int s) {
n = sz(p);
vector<int> col(n, 0);
vector<int> reqs(n + 2, 0);
int flag = 0;
multiset<int> lhf;
ll sum = 0;
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);
}
if(cnt == 1) {
flag--;
col[i] = 0;
}
else {
int L = *vals.begin();
int R = *vals.rbegin();
reqs[L]++;
reqs[R]--;
}
}
}
for(int i = 1; i <= n; i++)
reqs[i] += reqs[i - 1];
int l = 0, r = n - 1;
while(l < n && p[l] == l) l++;
if(l == n) return 0;
while(p[r] == r) r--;
r--;
ll mn = 0;
for(int i = l; i <= r; i++) mn += reqs[i] == 0;
return sum + mn * 2;
}
/**
Anul asta se da centroid.
-- Surse oficiale
*/
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 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 |
1 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 |
1 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 |
1 ms |
212 KB |
3rd lines differ - on the 1st token, expected: '3304', found: '2744' |
2 |
Halted |
0 ms |
0 KB |
- |
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Incorrect |
1 ms |
212 KB |
3rd lines differ - on the 1st token, expected: '6', found: '4' |
2 |
Halted |
0 ms |
0 KB |
- |