bzoj 5099: [POI2018]Pionek

题意

给你一些向量
选出其中一些使得他们的和最大

题解

这题做了一个上午

一个结论：向量和的长度等于所有向量在其方向上投影的长度和。
因此想要向量和的长度最大，即要选择所有在其方向上投影长度为正的向量。
由于与一个向量夹角在 (−π/2,π/2)(−π/2,π/2) 范围内的向量在其方向上投影为正，因此所求的就是对于任何一个长度为 π 的区间包含的所有向量的和长度的最大值。

怎么弄出所有方向呢？

对于区间左端点为某个给定向量的，可以通过双指针法来维护向量和。
对于区间左端点不为某个给定向量的，可以在双指针每一步（尾部加向量、头部删向量）后都统计一遍答案。容易发现这样一定是正确的。

为什么是正确的呢？
假设我们有当前的芳香，不想选择前面的一条直线
那么他们的夹角肯定是比$π/2$大的
后面同理
当两个都比$π/2$的时候，就肯定比 π π <script type="math/tex" id="MathJax-Element-36">π</script>大
就可以统计答案了

CODE:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>
typedef long long LL;
using namespace std;
const double pi=acos(-1);
const LL N=200005*2;
LL n;
struct qq
{
    LL x,y;
    double a;
}s[N];
bool cmp(qq x,qq y){return x.a>y.a;}
LL tx[N],ty[N];
LL ans=0;
void check (LL l,LL r)
{
    if (l<=r)   ans=max(ans,(tx[r]-tx[l-1])*(tx[r]-tx[l-1])+(ty[r]-ty[l-1])*(ty[r]-ty[l-1]));
}
int main()
{
    scanf("%lld",&n);
    for (LL u=1;u<=n;u++)
    {
        scanf("%lld%lld",&s[u].x,&s[u].y);
        s[u].a=atan2(s[u].y,s[u].x);
        s[u+n]=s[u];
        s[u+n].a=s[u].a+2*pi;
    }
    sort(s+1,s+1+2*n,cmp);
    for (LL u=1;u<=2*n;u++) tx[u]=tx[u-1]+s[u].x,ty[u]=ty[u-1]+s[u].y;
    LL i=1;
    for (LL u=1;u<=2*n;u++)
    {
        while (i<u&&s[u].a<=s[i].a-pi) i++,check(i,u-1);
        check(i,u);
    }
    while (i<=2*n) check(i,2*n),i++;
    printf("%lld\n",ans);
    return 0;
}