This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include "aliens.h"
#include <iostream>
#include <algorithm>
using namespace std;
struct node{
long long int x,y;
}a[1000005];
double dp[1000005],t[1000005][3];
long long int ans,x[1000005],y[1000005],g[1000005];
bool cmp(node a,node b)
{
if(a.x<b.x)return 1;
if(a.x>b.x)return 0;
if(a.y>a.y)return 1;
return 0;
}
double check(long long int a,long long int b)
{
if(t[a][1]==t[b][1])return 99999999999999;
return (t[a][0]-t[b][0])/(t[a][1]-t[b][1]);
}
long long take_photos(int n, int m, int k, std::vector<int> r, std::vector<int> c) {
for(int i=0;i<n;i++)
{
a[i].x=r[i];
a[i].y=c[i];
if(a[i].y<a[i].x)
{
swap(a[i].x,a[i].y);
}
}
sort(a,a+n,cmp);
long long int cnt=1;
x[1]=a[0].x;
y[1]=a[0].y;
for(int i=1;i<n;i++)
{
if(a[i].x>x[cnt] && a[i].y>y[cnt])
{
cnt++;
x[cnt]=a[i].x;
y[cnt]=a[i].y;
}
else if(a[i].x==x[cnt] && a[i].y>y[cnt])
{
x[cnt]=a[i].x;
y[cnt]=a[i].y;
}
}
n=cnt;
//cout<<n<<endl;
//for(int i=1;i<=n;i++)cout<<x[i]<<" "<<y[i]<<endl;
//system("pause");
if(k>n)k=n;
x[0]=0;
y[0]=0;
double le=0,ri=m*m;
while(true)
{
double mid=(le+ri)/2;
int st=0,en=0;
//cout<<mid<<endl;
dp[1]=0;
t[0][0]=(x[1]-1)*(x[1]-1);
t[0][1]=x[1]-1;
t[0][2]=0;
for(int i=1;i<=n;i++)
{
//cout<<i<<" "<<check(st,st+1)<<" "<<2*y[i]<<endl;
while(en>=st+1 && check(st,st+1)<2*y[i])
{
st++;
}
int p=t[st][2];
//cout<<st<<" "<<p<<endl;
if(y[p]-x[p+1]+1>0 && p!=0)dp[i]=mid+(double)dp[p]+((y[i]-x[p+1]+1)*(y[i]-x[p+1]+1)-(y[p]-x[p+1]+1)*(y[p]-x[p+1]+1));
else dp[i]=mid+(double)(dp[p]+(y[i]-x[p+1]+1)*(y[i]-x[p+1]+1));
g[i]=g[p]+1;
en++;
//cout<<dp[i]+(x[i+1]-1)*(x[i+1]-1)<<endl;
t[en][0]=dp[i]+(x[i+1]-1)*(x[i+1]-1)-max((long long int)0,y[i]-x[i+1]+1)*max((long long int)0,y[i]-x[i+1]+1);
t[en][1]=x[i+1]-1;
t[en][2]=i;
while(en>=st+2 && check(en,en-2)<check(en-1,en-2))
{
en--;
t[en][0]=dp[i]+(x[i+1]-1)*(x[i+1]-1)-max((long long int)0,y[i]-x[i+1]+1)*max((long long int)0,y[i]-x[i+1]+1);
t[en][1]=x[i+1]-1;
t[en][2]=i;
}
}
//cout<<g[n]<<" "<<dp[n]-mid*g[n]<<endl;
if(g[n]==k || ri-le<0.00000000000001)return dp[n]-mid*g[n]+0.00001;
if(g[n]<k)
{
ri=mid;
}
else le=mid;
}
}
Compilation message (stderr)
aliens.cpp: In function 'bool cmp(node, node)':
aliens.cpp:14:8: warning: self-comparison always evaluates to false [-Wtautological-compare]
14 | if(a.y>a.y)return 1;
| ~~~^~~~
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |