Submission #1124764

#	Time	Username	Problem	Language	Result	Execution time	Memory
1124764		adaawf	Aliens (IOI16_aliens)	C++20	4 / 100	3 ms	328 KiB

aliens

#include <bits/stdc++.h>
using namespace std;
vector<pair<long long int, long long int>> vv, v;
pair<long long int, int> f[100005];
int mm;
struct LINE {
    long long int a, b, num;
} t[4000005];
long long int trya(LINE a, long long int x) {
    return a.a * x + a.b;
}
void update(int v, int tl, int tr, LINE x) {
    int mid = (tl + tr) / 2;
    long long int h = trya(t[v], mid), k = trya(x, mid);
    if (h > k) swap(t[v], x);
    if (h == k && x.num < t[v].num) swap(t[v], x);
    if (tl == tr) return;
    if (x.a < t[v].a) update(v * 2 + 1, mid + 1, tr, x);
    else update(v * 2, tl, mid, x);
}
LINE sum(int v, int tl, int tr, long long int x) {
    if (tl == tr) {
        return t[v];
    }
    int mid = (tl + tr) / 2;
    LINE g, l = t[v];
    if (x <= mid) g = sum(v * 2, tl, mid, x);
    else g = sum(v * 2 + 1, mid + 1, tr, x);
    long long int h = trya(g, x), k = trya(l, x);
    if (h > k) swap(g, l);
    if (h == k && g.num < l.num) swap(g, l);
    return g;
}
pair<long long int, int> check(long long int x) {
    for (int i = 0; i <= mm * 4 + 1; i++) {
        t[i] = {0, 1000000000000000000, 0};
    }
    for (int i = 1; i < v.size(); i++) {
        long long int h = (v[i].first - 1) * (v[i].first - 1) + f[i - 1].first - max(v[i - 1].second - v[i].first + 1, 0ll) * max(v[i - 1].second - v[i].first + 1, 0ll) + x;
        update(1, 1, mm, {-2 * (v[i].first - 1), h, f[i - 1].second});
        LINE k = sum(1, 1, mm, v[i].second);
        f[i].first = trya(k, v[i].second) + v[i].second * v[i].second;
        f[i].second = k.num + 1;
    }
    return f[v.size() - 1];
}
long long int take_photos(int n, int m, int k, vector<int> r, vector<int> c) {
    mm = m;
    for (int i = 0; i < n; i++) {
        vv.push_back({min(r[i], c[i]) + 1, max(r[i], c[i]) + 1});
    }
    sort(vv.begin(), vv.end());
    v.push_back({-1, -1});
    int ma = -1;
    for (auto w : vv) {
        if (w.second <= ma) continue;
        if (!v.empty() && w.first == v.back().first) v.pop_back();
        v.push_back(w);
        ma = w.second;
    }
    long long int l = -1e12, rr = 1e12, res;
    while (l <= rr) {
        long long int mid = (l + rr) / 2;
        if (check(mid).second <= k) {
            res = mid;
            rr = mid - 1;
        }
        else l = mid + 1;
    }
    pair<long long int, long long int> h = check(res);
    return h.first - h.second * res;
}

Compilation message (stderr)

aliens.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~
aliens_c.h:1:9: warning: #pragma once in main file
    1 | #pragma once
      |         ^~~~

• Subtask 1: 4 / 4
• Subtask 2: 0 / 12
• Subtask 3: 0 / 9
• Subtask 4: 0 / 16
• Subtask 5: 0 / 19
• Subtask 6: 0 / 40