Submission #993658

#	Time	Username	Problem	Language	Result	Execution time	Memory
993658		LOLOLO	Krov (COCI17_krov)	C++17	140 / 140	1068 ms	19280 KiB

krov

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
 
#define           f     first
#define           s     second
#define           pb    push_back
#define           ep    emplace
#define           eb    emplace_back
#define           lb    lower_bound
#define           ub    upper_bound
#define       all(x)    x.begin(), x.end()
#define      rall(x)    x.rbegin(), x.rend()
#define   uniquev(v)    sort(all(v)), (v).resize(unique(all(v)) - (v).begin())
#define     mem(f,x)    memset(f , x , sizeof(f))
#define        sz(x)    (ll)(x).size()
#define  __lcm(a, b)    (1ll * ((a) / __gcd((a), (b))) * (b))
#define          mxx    *max_element
#define          mnn    *min_element
#define    cntbit(x)    __builtin_popcountll(x)
#define       len(x)    (int)(x.length())

const int N = 1e5 + 100;

ll h[N], x[N], y[N];

struct fen{
    vector <ll> f;
    int n;
    void ass(int N) {
        n = N;
        f.assign(n + 1, 0);
    }

    void upd(int i, int x) {
        for (; i <= n; i += i & (-i))
            f[i] += x;
    }

    ll get(int i) {
        ll s = 0;
        for (; i; i -= i & (-i))
            s += f[i];

        return s;
    }

    ll range(int l, int r) {
        return get(r) - get(l - 1);
    }
} suml, sumr, cntl, cntr;

map <ll, ll> m1, m2;
int n;

ll get(int p, ll v) {
    ll x = v + (n - p), y = v + p; 
    auto L = *m2.lower_bound(x);
    auto R = *m1.lower_bound(y);

    ll ans = suml.range(L.s, n + 1) - x * cntl.range(L.s, n + 1);
    ans += x * cntl.get(L.s - 1) - suml.get(L.s - 1);

    ans += sumr.range(R.s, n + 1) - y * cntr.range(R.s, n + 1);
    ans += y * cntr.get(R.s - 1) - sumr.get(R.s - 1);    
    return ans;
}

ll cal(int p) {
    ll l = max(p, n - p + 1), r = 1e9 + 1, ans = 1e18;
    while (l <= r) {
        ll m = (l + r) / 2;
        ll x = get(p, m) + abs(h[p] - m), y = get(p, m + 1) + abs(h[p] - (m + 1));
        if (y <= x) {
            l = m + 1;
        } else {
            r = m - 1;
        }

        ans = min({ans, x, y});
    }

    return ans;
}

ll solve() {
    cin >> n;

    for (int i = 1; i <= n; i++) {
        cin >> h[i];
        x[i] = h[i] + i;
        y[i] = h[i] + (n - i);
        m1[x[i]];
        m2[y[i]];
    }

    m1[1e18];
    m2[1e18];

    int timer = 1;
    for (auto &x : m1)
        x.s = timer++;

    timer = 1;
    for (auto &x : m2)
        x.s = timer++;

    suml.ass(n + 1);
    sumr.ass(n + 1);
    cntl.ass(n + 1); 
    cntr.ass(n + 1);

    for (int i = 1; i <= n; i++) {
        sumr.upd(m1[x[i]], x[i]);
        cntr.upd(m1[x[i]], 1);
    }

    ll ans = 1e18;
    for (int i = 1; i <= n; i++) {
        sumr.upd(m1[x[i]], -x[i]);
        cntr.upd(m1[x[i]], -1);
        ans = min(ans, cal(i));
        suml.upd(m2[y[i]], y[i]);
        cntl.upd(m2[y[i]], 1);
    }

    return ans;
}

int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    int t = 1;
    //cin >> t;

    while (t--) {
        cout << solve() << '\n';
    }

    return 0;
}  

• Subtask 1: 14 / 14
• Subtask 2: 14 / 14
• Subtask 3: 14 / 14
• Subtask 4: 14 / 14
• Subtask 5: 14 / 14
• Subtask 6: 14 / 14
• Subtask 7: 14 / 14
• Subtask 8: 14 / 14
• Subtask 9: 14 / 14
• Subtask 10: 14 / 14