Submission #760156

#	Time	Username	Problem	Language	Result	Execution time	Memory
760156		AryanReddy	Aliens (IOI16_aliens)	C++14	12 / 100	76 ms	2292 KiB

aliens

#undef local
#include <bits/stdc++.h>
#define all(v) v.begin(),v.end()
#define allr(v) v.rbegin(),v.rend()
#define fori(i,n) for(int i=0;i<n;i++)
#define ford(i,n) for(int i = n-1;i >= 0;i--)
#define pb push_back
#define ll long long int
//#define mod 998244353
#define pi pair<int,int>
#define pll pair<ll,ll>
#define mp make_pair
#define fi first
#define se second
#define printVector(v) fori(;,v.size()) {cout << v[i] << " ";} cout << "\n";
#pragma GCC optimize("O3")
std::mt19937 rng((unsigned int) std::chrono::steady_clock::now().time_since_epoch().count());
using namespace std;
ll getAns(vector<ll> a,ll k){
    int n = a.size();
    vector<vector<ll>> dp(n+1,vector<ll> (k+1,1e18));
    for(int i = 0;i <= k;i++) dp[0][i] = 0;
    for(int i = 1;i <= n;i++){
        for(int j= k;j >= 1;j--){
            dp[i][j] = dp[i][j-1];
            for(int t = i;t >= 1;t--){
                ll d = a[i-1] - a[t-1]  + 1;
                d *= d;
                dp[i][j] = min(dp[i][j],dp[t-1][j-1] + d);
            }
        }
    }
    return dp[n][k];
}

long long take_photos(int n, int m, int k, vector<int> r, vector<int> c){
    vector<ll> a;
    for(int i = 0;i < n;i++){
        int x = r[i];
        int y = c[i];
        a.push_back(x);
        a.push_back(y);
    }
    sort(all(a));
    a.erase(unique(all(a)),a.end());
    ll ans = getAns(a,k);
    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