Submission #761729

#TimeUsernameProblemLanguageResultExecution timeMemory
761729Dan4LifeAliens (IOI16_aliens)C++17
12 / 100
77 ms2260 KiB
#include "aliens.h"
#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define sz(a) (int)a.size()
using ll = long long;
using ar = array<int,2>;

ll take_photos(int n, int m, int k, vector<int> r, vector<int> c) {
	for(int i = 0; i < n; i++)
		if(r[i]>c[i]) r[i]=m-r[i]-1,c[i]=m-c[i]-1;
	vector<ar> v,w; int mn=m;
	for(int i = 0; i < n; i++) w.pb({c[i],r[i]});
	sort(begin(w),end(w),[&](ar a, ar b){
		if(a[0]!=b[0]) return a[0]<b[0];
		return a[1]>b[1];
	}); 
	for(int i = n-1; i >=0; i--){
		if(mn>w[i][1]) v.pb(w[i]);
		mn = min(mn,w[i][1]);
	}
	n = sz(v); ll dp[k+1][n+1], ans = (ll)4e18;
	memset(dp,63,sizeof(dp)); sort(begin(v),end(v));
	for(int i = 1; i <= n; i++){
		dp[1][i]=v[i-1][0]-v[0][1]+1;
		dp[1][i]*=dp[1][i];
	}
	for(int j = 2; j <= k; j++)
		for(int i = 1; i <= n; i++)
			for(int x = 1; x < i; x++)
				dp[j][i] = min(dp[j][i], dp[j-1][x]+
					(v[i-1][0]-v[x][1]+1)*(v[i-1][0]-v[x][1]+1));	
	for(int i = 1; i <= k; i++) ans = min(ans, dp[i][n]);
    return ans;
}
#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...