Submission #773627

#	Time	Username	Problem	Language	Result	Execution time	Memory
773627		PoonYaPat	Aliens (IOI16_aliens)	C++14	12 / 100	78 ms	1332 KiB

aliens

#include "aliens.h"
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;

int n,m,k;
int dp[505][505]; //n,k
vector<pii> v;
set<pii> h;

long long take_photos(int N, int M, int K, vector<int> y, vector<int> x) {
    n=N; m=M; k=K;
    for (int i=0; i<n; ++i) h.insert(pii(x[i],y[i]));
    for (auto u : h) v.push_back(u);
    sort(v.begin(),v.end());

    for (int i=0; i<n; ++i) for (int j=0; j<=k; ++j) dp[i][j]=1e9;
    for (int j=1; j<=k; ++j) dp[0][j]=1;

    for (int i=1; i<v.size(); ++i) {
        for (int j=1; j<=k; ++j) {

            //find dp[i][j]
            if (j==1) {
                dp[i][j]=min(dp[i][j],(v[i].first-v[0].first+1)*(v[i].first-v[0].first+1));
            } else {
                for (int x=1; x<=i; ++x) { //capture i to x
                    dp[i][j]=min(dp[i][j],dp[x-1][j-1]+(v[i].first-v[x].first+1)*(v[i].first-v[x].first+1));
                }
            }

        }
    }

    return dp[v.size()-1][k];
}

Compilation message (stderr)

aliens.cpp: In function 'long long int take_photos(int, int, int, std::vector<int>, std::vector<int>)':
aliens.cpp:21:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   21 |     for (int i=1; i<v.size(); ++i) {
      |                   ~^~~~~~~~~

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