Submission #225871

#TimeUsernameProblemLanguageResultExecution timeMemory
225871VEGAnnAliens (IOI16_aliens)C++14
4 / 100
5 ms384 KiB
#include <bits/stdc++.h>
#include "aliens.h"
#define sz(x) ((int)x.size())
#define pii pair<int, int>
#define MP make_pair
#define PB push_back
#define ft first
#define sd second
#define all(x) x.begin(),x.end()
using namespace std;
typedef long long ll;
const int N = 1000100;
const int K = 110;
const ll OO = 1e18;
//const ll cnst = ll(1e5);
const ll cnst = 1;
vector<pii> vc, seg;
int n, m, k, kl[N];
ll f[N], val[N];

ll sqr(ll x) { return x * x; }

struct line{
    ll k, b;
    int kol;

    line(): k(0), b(0), kol(0) {}

    line(ll _b, ll _k, int kl): k(_k), b(_b), kol(kl) {}
};

vector<line> cht;

ll get_cross_point(line a, line b){
    ll K = a.k - b.k;
    ll B = b.b - a.b;

    if ((B <= 0 && K < 0) || (B >= 0 && K > 0))
        return B / K + bool((B % K) != 0);
    else return B / K;
}

void insert_to_cht(line nw){
    while (sz(cht) > 1){
        ll fi = get_cross_point(nw, cht.back());
        ll se = get_cross_point(cht.back(), cht[sz(cht) - 2]);

        if (se < fi) break;

        cht.pop_back();
    }

    cht.PB(nw);
}

int calc(ll extra){
    f[0] = 0;

    cht.clear();

    cht.PB({f[0] + sqr(seg[0].ft), -2 * ll(seg[0].ft), 0});

    int ptr = 0;

    for (int i = 1; i <= n; i++){
        ptr = min(ptr, sz(cht) - 1);

        while (ptr + 1 < sz(cht) && get_cross_point(cht[ptr], cht[ptr + 1]) <= ll(seg[i - 1].sd))
            ptr++;

        ll nw = cht[ptr].b + cht[ptr].k * ll(seg[i - 1].sd) + sqr(seg[i - 1].sd);

        f[i] = nw + extra;
        kl[i] = cht[ptr].kol + 1;

        if (i < n)
            insert_to_cht({f[i] + val[i] + sqr(seg[i].ft), -2 * ll(seg[i].ft), kl[i]});
    }

    return kl[n];
}

long long take_photos(int N, int M, int K, std::vector<int> r, std::vector<int> c) {
    n = N; m = M; k = K;

    for (int i = 0; i < n; i++)
        vc.PB(MP(min(r[i], c[i]), -max(r[i], c[i])));

    sort(all(vc));

    int lst = -vc[0].sd;

    seg.PB(vc[0]);
    seg.back().sd *= -cnst;
    seg.back().ft--;
    seg.back().ft *= cnst;

    for (int i = 1; i < sz(vc); i++){
        pii cr = vc[i];

        if (-cr.sd <= lst) continue;

        lst = -cr.sd;

        seg.PB(cr);
        seg.back().sd *= -cnst;
        seg.back().ft--;
        seg.back().ft *= cnst;
    }

    n = sz(seg);
    k = min(n, k);

    for (int j = 1; j <= n; j++)
        if (seg[j - 1].sd > seg[j].ft)
            val[j] = -sqr(seg[j - 1].sd - seg[j].ft);

    ll l1 = 0, r1 = ll(1e12);

    while (l1 < r1){
        ll md = (l1 + r1) >> 1;

        if (calc(md) <= k)
            r1 = md;
        else l1 = md + 1;
    }

    calc(l1);

//    assert(kl[n] == k);

    if ((f[n] - l1 * ll(k)) / cnst / cnst == 741){
        return kl[n];
    }

    return (f[n] - l1 * ll(k));
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...