Submission #657841

#TimeUsernameProblemLanguageResultExecution timeMemory
657841benjaminkleynAliens (IOI16_aliens)C++17
41 / 100
2081 ms4136 KiB
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

#define COL first
#define ROW second

vector<pair<ll,ll>> pts, dp;
pair<ll,ll> eval(int i, int j, ll cost)
{
    ll new_square = (pts[i].COL - pts[j+1].ROW + 1);
    ll overlap = max(pts[j].COL - pts[j+1].ROW + 1, 0LL);
    ll total_area = new_square * new_square - overlap * overlap;

    return {dp[j].first + total_area + cost, dp[j].second + 1};
}

pair<ll,ll> calc(int n, ll cost)
{
    dp[0] = {0, 0};
    for (int i = 1; i <= n; i++)
    {
        dp[i] = {LLONG_MAX, LLONG_MAX};
        for (int j = 0; j < i; j++)
            dp[i] = min(dp[i], eval(i, j, cost));
    }

    return dp[n];
}

ll take_photos(int n, int m, int k, vector<int> r, vector<int> c)
{
    vector<pair<ll,ll>> temp_pts(n);
    for (int i = 0; i < n; i++)
    {
        if (r[i] > c[i])
            swap(r[i], c[i]);
        temp_pts[i] = {c[i], r[i]};
    }
    sort(temp_pts.begin(), temp_pts.end());

    pts.push_back({-1, -1});
    for (pair<ll,ll> pt : temp_pts)
    {
        while (pts.size() && pts.back().ROW >= pt.ROW)
            pts.pop_back();
        pts.push_back(pt);
    }

    n = pts.size() - 1;
    dp.resize(n + 1);

    ll lo = 0, hi = 1e12, res = 1e12, mid;
    while (lo <= hi)
    {
        mid = (lo + hi) / 2;

        calc(n, mid);

        if (dp[n].second <= k)
            hi = mid - 1, res = mid;
        else
            lo = mid + 1;

    }

    calc(n, res);
    return dp[n].first - k * res;
}
#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...