Submission #337600

#TimeUsernameProblemLanguageResultExecution timeMemory
337600errorgornAliens (IOI16_aliens)C++17
60 / 100
768 ms8156 KiB
//雪花飄飄北風嘯嘯
//天地一片蒼茫

#include "aliens.h"
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/rope>
using namespace std;
using namespace __gnu_pbds;
using namespace __gnu_cxx;
#define ll long long
#define ii pair<ll,ll>
#define iii pair<ii,ll>
#define fi first
#define se second
#define endl '\n'
#define debug(x) cout << #x << " is " << x << endl

#define rep(x,start,end) for(auto x=(start)-((start)>(end));x!=(end)-((start)>(end));((start)<(end)?x++:x--))
#define all(x) (x).begin(),(x).end()
#define sz(x) (int)(x).size()

#define indexed_set tree<ll,null_type,less<ll>,rb_tree_tag,tree_order_statistics_node_update>
//change less to less_equal for non distinct pbds, but erase will bug

mt19937 rng(chrono::system_clock::now().time_since_epoch().count());

int n,m,k;
vector<ii> v;
ll extra[100005];
ll memo[2][50005];

ll sq(ll i){return i*i;}

struct line{
	ll m,c;
	
	ll eval(ll x){
		return m*x+c;
	}
};

struct hull{
	deque<line> dq;
	
	double POI(line i,line j){
		return (double)(i.c-j.c)/(j.m-i.m);
	}
	
	void add(line i){
		while (sz(dq)>=2 && POI(dq.back(),i)<=POI(dq[sz(dq)-2],dq.back())) dq.pop_back();
		dq.push_back(i);
	}
	
	ll query(ll i){
		while (sz(dq)>=2 && dq[1].eval(i)<dq[0].eval(i)) dq.pop_front();
		return dq.front().eval(i);
	}
};

long long take_photos(int N, int M, int K, std::vector<int> R, std::vector<int> C) {
    n=N,m=M,k=K;
    
    rep(x,0,n){
		int a=R[x],b=C[x];
		if (a>b) swap(a,b);
		v.push_back(ii(a,b+1));
	}
	
	sort(all(v),[](ii i,ii j){
		if (i.fi!=j.fi) return i.fi<j.fi;
		else return i.se>j.se;
	});
	
	vector<ii> temp;
	for (auto &it:v){
		if (temp.empty() || temp.back().se<it.se) temp.push_back(it);
	}
	
	swap(v,temp);
	
	//for (auto &it:v) cout<<it.fi<<" "<<it.se<<endl;
	
	rep(x,1,sz(v)) extra[x]=sq(max(0LL,v[x-1].se-v[x].fi));
	//rep(x,0,sz(v)) cout<<extra[x]<<" "; cout<<endl;
	
	int a=0,b=1;
	memset(memo,63,sizeof(memo));
	
	rep(zzz,0,k){
		memo[a][0]=0;
		memset(memo[b],63,sizeof(memo[b]));
		hull h;
		
		rep(x,0,sz(v)){
			h.add({-2*v[x].fi,memo[a][x]-extra[x]+sq(v[x].fi)});
			memo[b][x+1]=h.query(v[x].se)+sq(v[x].se);
		}
		swap(a,b);
	}
	
	return memo[a][sz(v)];
}
#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...