Submission #130553

#	Time	Username	Problem	Language	Result	Execution time	Memory
130553		fefe	Aliens (IOI16_aliens)	C++17	4 / 100	33 ms	504 KiB

aliens

#include<bits/stdc++.h>
#include "aliens.h"
#define MAX_N 100005
#define sq(x) ((x)*(x))
#define inf (1LL<<60)
using namespace std;
typedef long long LL;
typedef pair<LL,LL> pil;
struct node{
	LL s,e;
}V[MAX_N];
struct line{
	double a,b;
	LL c;
}L[MAX_N];
LL n,k,ans;
LL from[MAX_N];
double dp[MAX_N];
bool is_ok(line a,line b,line c){
	return (b.b-a.b)*(b.a-c.a)<(a.a-b.a)*(c.b-b.b);
}
LL CHT(double c){
	LL s,e;
	LL i;
	s=e=0;
	L[e++]={-2*V[1].s,sq(V[1].s)-2*V[1].s,0};
	for(i=1;i<=n;i++){
		while(e-s>1 && L[s].a*V[i].e+L[s].b>=L[s+1].a*V[i].e+L[s+1].b)	s++;
		dp[i]=L[s].a*V[i].e+L[s].b+sq(V[i].e+1)+c;
		from[i]=L[s].c;
		line p={-2*V[i+1].s,sq(V[i+1].s)-2*V[i+1].s+dp[i]};
		if(V[i].e-V[i+1].s+1>0)	p.b-=sq(V[i].e-V[i+1].s+1);
		p.c=i;
		while(e-s>1 && !is_ok(L[e-2],L[e-1],p))	e--;
		L[e++]=p;
	}
	for(c=0,i=n;i;i=from[i],c++);
	return c;
}
long long take_photos(int N, int m, int K, std::vector<int> r, std::vector<int> c) {	
	LL i;
	LL s=0,e=(LL)m*m;
	n=N;
	k=K;
	for(i=1;i<=n;i++){
		V[i].s=min(r[i-1],c[i-1]);
		V[i].e=max(r[i-1],c[i-1]);
	}
	sort(V+1,V+n+1,[&](const node x,const node y){return (x.s==y.s)?(x.e>y.e):(x.s<y.s);});
	
	m=0;
	LL x=-10000;
	for(i=1;i<=n;i++){
		if(x<V[i].e)	V[++m]=V[i];
		x=max(x,V[i].e);
	}
	
	n=m;
	sort(V+1,V+n+1,[&](const node x,const node y){return x.e<y.e;});
	ans=inf;
	k=min(n,k);
	bool Z=false;
	
	while(e-s>10){
		LL mid=(s+e)>>1;
		x=CHT(mid);
		if(x==k){
			break;
		}else if(x<k){
			e=mid-1;
		}else{
			s=mid+1;
		}
	}
	double j;
	for(j=s;j<=e;j+=0.5){
		x=CHT(j);
		if(x>k)	continue;
		ans=min(ans,(LL)(dp[n]-j*x));
	}
	return ans;
}

Compilation message (stderr)

aliens.cpp: In function 'LL CHT(double)':
aliens.cpp:26:12: warning: narrowing conversion of '(-2 * V[1].node::s)' from 'LL {aka long long int}' to 'double' inside { } [-Wnarrowing]
  L[e++]={-2*V[1].s,sq(V[1].s)-2*V[1].s,0};
          ~~^~~~~~~
aliens.cpp:26:30: warning: narrowing conversion of '((V[1].node::s * V[1].node::s) - (2 * V[1].node::s))' from 'LL {aka long long int}' to 'double' inside { } [-Wnarrowing]
  L[e++]={-2*V[1].s,sq(V[1].s)-2*V[1].s,0};
                              ^
aliens.cpp:31:13: warning: narrowing conversion of '(-2 * V[(i + 1)].node::s)' from 'LL {aka long long int}' to 'double' inside { } [-Wnarrowing]
   line p={-2*V[i+1].s,sq(V[i+1].s)-2*V[i+1].s+dp[i]};
           ~~^~~~~~~~~
aliens.cpp: In function 'long long int take_photos(int, int, int, std::vector<int>, std::vector<int>)':
aliens.cpp:62:7: warning: unused variable 'Z' [-Wunused-variable]
  bool Z=false;
       ^

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