Submission #429279

#	Time	Username	Problem	Language	Result	Execution time	Memory
429279		JeanBombeur	Aliens (IOI16_aliens)	C++17	100 / 100	111 ms	6464 KiB

aliens

#include "aliens.h"
#include <bits/stdc++.h>
using namespace std;
 
#define ll long long
#define pii pair<int, int>
#define pli pair<ll, int>
#define fi first
#define se second
 
const int mxN=1e5;
int qh, qt, a[mxN], c[mxN];
ll b[mxN];
pii ps1[mxN];
vector<pii> ps2;
 
inline void al(int ai, ll bi, int ci) {
	while(qt-qh>1&&pli((b[qt-1]-b[qt-2])*(a[qt-1]-ai), c[qt-1])>pli((bi-b[qt-1])*(a[qt-2]-a[qt-1]), ci))
		--qt;
	a[qt]=ai, b[qt]=bi, c[qt]=ci;
	++qt;
}
 
inline pli qry(ll xi) {
	while(qt-qh>1&&pli(a[qh+1]*xi+b[qh+1], c[qh+1])<pli(a[qh]*xi+b[qh], c[qh]))
		++qh;
	return pli(a[qh]*xi+b[qh], c[qh]);
}
 
inline pli ddp(ll c) {
	qh=qt=0;
	pli dp;
	for(int i=0; i<ps2.size(); ++i) {
		ll d1=max(i?ps2[i-1].fi-ps2[i].se+1:0, 0), d2=ps2[i].se;
		al(-2*d2, dp.fi-d1*d1+d2*d2, dp.se);
		d1=ps2[i].fi+1;
		dp=qry(d1);
		dp.fi+=d1*d1+c, ++dp.se;
	}
	return dp;
}
 
ll take_photos(int n, int m, int k, vector<int> r, vector<int> c) {
	for(int i=0; i<n; ++i) {
		ps1[i]={r[i], c[i]};
		if(ps1[i].se>ps1[i].fi)
			swap(ps1[i].se, ps1[i].fi);
	}
	sort(ps1, ps1+n);
	ps2.clear();
	for(int i=0; i<n; ++i) {
		while(!ps2.empty()&&ps1[i].se<=ps2.back().se)
			ps2.pop_back();
		ps2.push_back(ps1[i]);
	}
	ll lb=0, rb=(ll)m*m, ans;
	while(lb<=rb) {
		ll mb=(lb+rb)/2;
		pli dp=ddp(mb);
		if(dp.se<=k) {
			ans=dp.fi;
			rb=mb-1;
		} else
			lb=mb+1;
	}
	return ans-lb*k;
}

Compilation message (stderr)

aliens.cpp: In function 'std::pair<long long int, int> ddp(long long int)':
aliens.cpp:33:16: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   33 |  for(int i=0; i<ps2.size(); ++i) {
      |               ~^~~~~~~~~~~
aliens.cpp: In function 'long long int take_photos(int, int, int, std::vector<int>, std::vector<int>)':
aliens.cpp:66:16: warning: 'ans' may be used uninitialized in this function [-Wmaybe-uninitialized]
   66 |  return ans-lb*k;
      |                ^

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