bzoj 5099: [POI2018]Pionek 双指针

Description

在无限大的二维平面的原点(0,0)放置着一个棋子。你有n条可用的移动指令，每条指令可以用一个二维整数向量表
示。每条指令最多只能执行一次，但你可以随意更改它们的执行顺序。棋子可以重复经过同一个点，两条指令的方
向向量也可能相同。你的目标是让棋子最终离原点的欧几里得距离最远，请问这个最远距离是多少？
Input

第一行包含一个正整数n(n<=200000)，表示指令条数。
接下来n行，每行两个整数x,y(|x|,|y|<=10000)，表示你可以从(a,b)移动到(a+x,b+y)。
Output

输出一行一个整数，即最大距离的平方。
Sample Input

5

2 -2

-2 -2

0 2

3 1

-3 1
Sample Output

26
HINT

分析：
先把向量极角排序，枚举最终向量方向，然后每一个180°的区域内的向量和就是答案。维护两个指针即可。
在加入一个向量或减掉一个向量时都要统计答案。

代码：

/**************************************************************
    Problem: 5099
    User: ypxrain
    Language: C++
    Result: Accepted
    Time:4024 ms
    Memory:7556 kb
****************************************************************/
 
#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#define LL long long
 
const int maxn=2e5+7;
const double pi=acos(-1);
 
using namespace std;
 
int n,p;
LL ans,sx,sy;
 
struct rec{
    int x,y;
    double ang;
}a[maxn*2];
 
bool cmp(rec a,rec b)
{
    return a.ang<b.ang;
}
 
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
    {
        scanf("%d%d",&a[i].x,&a[i].y);
        a[i].ang=atan2(a[i].x,a[i].y);
    }
    sort(a+1,a+n+1,cmp);
    for (int i=1;i<=n;i++)
    {
        a[i+n]=a[i];
        a[i+n].ang=a[i].ang+2*pi;
    }
    p=1;
    for (int i=1;i<=n;i++)
    {
        while ((p<i+n) && (a[p].ang-a[i].ang<pi))
        {
            sx+=(LL)a[p].x;
            sy+=(LL)a[p].y;
            ans=max(ans,sx*sx+sy*sy);
            p++;
        }
        sx-=(LL)a[i].x;
        sy-=(LL)a[i].y;
        ans=max(ans,sx*sx+sy*sy);
    }
    printf("%lld",ans);
}