BZOJ 5099: Pionek(双指针)(占位)

时间:2021-12-10 05:05:56

pro:有N个向量,你可以选择一些向量,使得其向量和离原点最远。 输出这个欧几里得距离的平方。

sol:(感觉网上的证明都不是很充分,我自己也是半信半疑吧)日后证明了再补。

#include<bits/stdc++.h>
#define ll long long
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int maxn=;
const double pi=acos(-1.0);
struct point{
int x,y;
double angle;
};
bool cmp(point w,point v){ return w.angle<v.angle; }
point a[maxn]; ll ans,x,y;
void update(){ ans=max(ans,1LL*x*x+y*y); }
int main()
{
int N,head=;
scanf("%d",&N);
rep(i,,N){
scanf("%d%d",&a[i].x,&a[i].y);
a[i].angle=atan2(a[i].y,a[i].x);
}
sort(a+,a+N+,cmp);
rep(i,,N){
a[i+N]=a[i];
a[i+N].angle=a[i].angle+pi*;
}
rep(i,,N){
x-=a[i-].x; y-=a[i-].y;
update();
while(head+-i+<=N&&head+<=N+N&&a[head+].angle-a[i].angle<=pi){
head++; x+=a[head].x; y+=a[head].y;
update();
}
}
printf("%lld\n",ans);
return ;
}