//{ <defines>
#include <bits/stdc++.h>
using namespace std;
//#pragma comment(linker, "/stack:200000000")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
//#pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math,-O3")
#define fr(i, n) for(int i = 0; i < n; ++i)
#define fo(n) fr(i, n)
#define re return
#define ef else if
#define ifn(x) if(!(x))
#define _ << ' ' <<
#define ft first
#define sd second
#define ve vector
#define pb push_back
#define eb emplace_back
#define sz(x) int((x).size())
#define bnd(x) x.begin(), x.end()
#define clr(x, y) memset((x), (y), sizeof (x))
typedef long long ll;
typedef long double ld;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
typedef ve<int> vi;
inline ll time() {re chrono :: system_clock().now().time_since_epoch().count();}
mt19937 rnd(time());
mt19937_64 RND(time());
template<typename t> inline void umin(t &a, t b) {a = min(a, b);}
template<typename t> inline void umax(t &a, t b) {a = max(a, b);}
int md = 998244353;
inline int m_add(int&a, int b) {a += b; if(a < 0) a += md; if(a >= md) a -= md; re a;}
inline int m_sum(int a, int b) {a += b; if(a < 0) a += md; if(a >= md) a -= md; re a;}
inline int m_mul(int&a, int b) {re a = 1ll * a * b % md;}
inline int m_prod(int a, int b) {re 1ll * a * b % md;}
int m_bpow(ll A, ll b) {
int a = A % md;
ll ans = 1;
for(ll p = 1ll << 63 - __builtin_clzll(b); p; p >>= 1) {
(ans *= ans) %= md;
if(p & b)
(ans *= a) %= md;
}
re ans;
}
//const ld pi = arg(complex<ld>(-1, 0));
//const ld pi2 = pi + pi;
const int oo = 2e9;
const ll OO = 4e18;
//} </defines>
const int N = 1e5 + 5;
int n;
int a[N];
void input() {
cin >> n;
fo(n) cin >> a[i];
}
struct tree {
vi v, c, fc;
ve<ll> fv;
int n;
ll s;
tree(): v(0), c(0), fc(0), fv(0), n(0), s(0) {};
void init() {
c = v;
sort(bnd(c));
fc.assign(sz(v) + 1, 0);
fv.assign(sz(v) + 1, 0);
}
void fcupd(int i, int v) { for(i++; i < sz(fc); i += i & -i) fc[i] += v; }
void fvupd(int i, int v) { for(i++; i < sz(fv); i += i & -i) fv[i] += v; }
int fcget(int i) { int v {}; for(i++; i; i -= i & -i) v += fc[i]; re v; }
int fvget(int i) { ll v {}; for(i++; i; i -= i & -i) v += fv[i]; re v; }
int index(int v) { re upper_bound(bnd(c), v) - c.begin() - 1; }
void add(int i) {
int j = index(v[i]);
fcupd(j, +1);
fvupd(j, +v[i]);
++n, s += v[i];
}
void del(int i) {
int j = index(v[i]);
fcupd(j, -1);
fvupd(j, -v[i]);
--n, s -= v[i];
}
ll get(int c) {
int j = index(c);
ll le = fcget(j);
ll ri = n - le;
ll sle = fvget(j);
ll sri = s - sle;
re le * c - sle +
sri - ri * c;
}
};
void solve() {
input();
ll ans = +OO;
tree le {}, ri {};
fo(n) le.v.pb(a[i] - i); le.init();
fo(n) ri.v.pb(a[i] + i); ri.init();
for(int i = 1; i < n; ++i) ri.add(i);
typedef unsigned int ui;
for(int i = 0; i < n; ++i) {
ui l = max(i + 1, n - i), r = 1.505e9;
while(l < r) {
int m0 = (l + r) >> 1;
int m1 = m0 + 1;
ll f0 = le.get(m0 - i) + ri.get(m0 + i) + abs(a[i] - m0);
ll f1 = le.get(m1 - i) + ri.get(m1 + i) + abs(a[i] - m1);
if(f0 == f1) l = r = m0;
if(f0 < f1) r = m0;
if(f1 < f0) l = m1;
}
umin<ll>(ans, le.get(l - i) + ri.get(l + i) + abs(a[i] - (int)l));
le.add(i);
if(i + 1 < n)
ri.del(i + 1);
}
cout << ans << endl;
}
int main() {
#ifdef _LOCAL
freopen("in.txt", "r", stdin);
int tests; cin >> tests;
for(int test = 1; test <= tests; ++test) {
cerr << "case #" << test << endl;
solve();
cerr << endl;
}
#else
// freopen("input.txt", "r", stdin); freopen("output.txt", "w", stdout);
ios_base :: sync_with_stdio(0); cin.tie(0); cout.tie(0);
solve();
#endif
return 0;
}
Compilation message
krov.cpp: In function 'int m_bpow(ll, ll)':
krov.cpp:50:26: warning: suggest parentheses around '-' inside '<<' [-Wparentheses]
for(ll p = 1ll << 63 - __builtin_clzll(b); p; p >>= 1) {
~~~^~~~~~~~~~~~~~~~~~~~
krov.cpp: In function 'void solve()':
krov.cpp:9:18: warning: this 'for' clause does not guard... [-Wmisleading-indentation]
#define fr(i, n) for(int i = 0; i < n; ++i)
^
krov.cpp:10:15: note: in expansion of macro 'fr'
#define fo(n) fr(i, n)
^~
krov.cpp:120:5: note: in expansion of macro 'fo'
fo(n) le.v.pb(a[i] - i); le.init();
^~
krov.cpp:120:30: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'for'
fo(n) le.v.pb(a[i] - i); le.init();
^~
krov.cpp:9:18: warning: this 'for' clause does not guard... [-Wmisleading-indentation]
#define fr(i, n) for(int i = 0; i < n; ++i)
^
krov.cpp:10:15: note: in expansion of macro 'fr'
#define fo(n) fr(i, n)
^~
krov.cpp:121:5: note: in expansion of macro 'fo'
fo(n) ri.v.pb(a[i] + i); ri.init();
^~
krov.cpp:121:30: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'for'
fo(n) ri.v.pb(a[i] + i); ri.init();
^~
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
9 ms |
384 KB |
Output is correct |
| 2 |
Correct |
9 ms |
384 KB |
Output is correct |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
11 ms |
512 KB |
Output is correct |
| 2 |
Correct |
9 ms |
384 KB |
Output is correct |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
11 ms |
384 KB |
Output is correct |
| 2 |
Correct |
12 ms |
512 KB |
Output is correct |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Incorrect |
13 ms |
512 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
19 ms |
512 KB |
Output is correct |
| 2 |
Correct |
18 ms |
512 KB |
Output is correct |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Incorrect |
22 ms |
760 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
163 ms |
1528 KB |
Output is correct |
| 2 |
Correct |
168 ms |
1784 KB |
Output is correct |
| 3 |
Correct |
178 ms |
1712 KB |
Output is correct |
| 4 |
Correct |
203 ms |
1920 KB |
Output is correct |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
301 ms |
2292 KB |
Output is correct |
| 2 |
Correct |
304 ms |
2820 KB |
Output is correct |
| 3 |
Incorrect |
144 ms |
2556 KB |
Output isn't correct |
| 4 |
Halted |
0 ms |
0 KB |
- |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Incorrect |
586 ms |
3700 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Incorrect |
424 ms |
4980 KB |
Output isn't correct |
| 2 |
Halted |
0 ms |
0 KB |
- |
