Submission #773737

#	Time	Username	Problem	Language	Result	Execution time	Memory
773737		Jomnoi	Aliens (IOI16_aliens)	C++17	12 / 100	104 ms	4176 KiB

aliens

#include <bits/stdc++.h>
#include "aliens.h"
using namespace std;

const int MAX_K = 1005;
const int MAX_N = 1005;
const int MAX_M = 1e6 + 5;
const long long INF = 1e18 + 7;

int N, M, K;
long long dp[MAX_K][MAX_N];
int qs[MAX_N][MAX_N];
bool dead[MAX_N];
int minL[MAX_M];

long long take_photos(int n, int m, int k, vector <int> r, vector <int> c) {
    N = n, M = m, K = k;
    for (int i = 0; i < m; i++) minL[i] = M;
    for (int i = 0; i < N; i++) {
        int x = min(r[i], c[i]), y = max(r[i], c[i]);
        minL[y] = min(minL[y], x);
    }

    stack <pair <int, int>> stk;
    for (int i = 0; i < m; i++) {
        if (minL[i] == M) continue;
        while (!stk.empty() and stk.top().first >= minL[i]) stk.pop();
        stk.emplace(minL[i], i);
    }

    vector <pair <int, int>> vec;
    while (!stk.empty()) {
        auto [x, y] = stk.top();
        stk.pop();

        vec.emplace_back(x, y);
    }

    N = vec.size();
    reverse(vec.begin(), vec.end());
    
    for (int i = 1; i <= N; i++) dp[0][i] = INF;
    for (int k = 1; k <= K; k++) {
        for (int i = 1; i <= N; i++) {
            dp[k][i] = INF;

            int minL = M;
            for (int j = i; j >= 1; j--) {
                minL = min(minL, vec[j - 1].first);
                // cout << minL << ' ' << Y[id[i - 1]] << endl;
                long long cost = (vec[i - 1].second - minL + 1) * (vec[i - 1].second - minL + 1);
                dp[k][i] = min(dp[k][i], cost + dp[k - 1][j - 1]);
            }
            // cout << k << ' ' << i << " : " << dp[k][i] << endl;
        }
    }

    long long ans = INF;
    for (int k = 1; k <= K; k++) ans = min(ans, dp[k][N]);
    return ans;
}

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