Submission #471497

#	Time	Username	Problem	Language	Result	Execution time	Memory
471497		sofapuden	Aliens (IOI16_aliens)	C++14	0 / 100	1 ms	204 KiB

aliens

#include "aliens.h"
#include<bits/stdc++.h>

using namespace std;

typedef long long ll;

const ll INF = (1ll<<60);

bool cmp(pair<ll,ll> &a, pair<ll,ll> &b){
	if(a.second == b.second)return a.first < b.first;
	return a.second < b.second;

}

ll take_photos(int n, int m, int k, vector<int> r, vector<int> c) {
	vector<pair<ll,ll>> v(n);
	for(int i = 0; i < n; ++i){
		v[i] = {r[i],c[i]};
	}
	for(int i = 0; i < n; ++i){
		if(v[i].first > v[i].second)swap(v[i].first,v[i].second);
	}
	sort(v.begin(),v.end(),cmp);
	ll st = v[0].first, cu = v[0].second;
	ll su = (cu-st+1)*(cu-st+1);
	ll cn = 1;
	for(int i = 0; i < n; ++i){
		if(v[i].second <= cu)continue;
		if((v[i].second-v[i].first+1)*(v[i].second-v[i].first+1) < ((v[i].second-st+1)*(v[i].second-st+1)-(cu-st+1)*(cu-st+1))){
			cn++;
			su+=(v[i].second-v[i].first+1)*(v[i].second-v[i].first+1);
			st = v[i].first;
			cu = v[i].second;
		}
		else{
			su+=(v[i].second-st+1)*(v[i].second-st+1)-(cu-st+1)*(cu-st+1);
			cu = v[i].second;
		}
	}
	if(cn <= k)return su;
	ll l = 0, ri = INF, ans = INF;
	while(l <= ri){
		ll m = (l+ri)>>1;
		ll st = v[0].first, cu = v[0].second;
		ll su = (cu-st+1)*(cu-st+1);
		ll cn = 1;
		for(int i = 0; i < n; ++i){
			if(v[i].second <= cu)continue;
			if(((v[i].second-v[i].first+1)*(v[i].second-v[i].first+1)+m) < ((v[i].second-st+1)*(v[i].second-st+1)-(cu-st+1)*(cu-st+1))){
				cn++;
				su+=(v[i].second-v[i].first+1)*(v[i].second-v[i].first+1);
				st = v[i].first;
				cu = v[i].second;
			}
			else{
				su+=(v[i].second-st+1)*(v[i].second-st+1)-(cu-st+1)*(cu-st+1);
				cu = v[i].second;
			}
		}
		if(cn <= k){
			ans = min(ans,su);
			ri = m-1;
		}
		else{
			l = m+1;
		}
	}
	return ans;
}

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