HDU 6127 Hard challenge（计算几何）

Description

给出二维平面$n$个点的坐标$(x_i,y_i)$和权值$val_i$，保证任意两点不重复且连线不经过原点，经过原点但不经过这$n$个点中任一点的直线，定义得分为这条直线两边点点权和的乘积，问最大得分

Input

第一行一整数$T$表示用例组数，每组用例首先输入一整数$n$表示点数，之后$n$行每行输入三个整数$x_i,y_i,val_i$表示点的坐标和权值$(1\le T\le 5,1\le n\le 5\cdot 10^4,|x_i|,|y_i|\le 10^9,1\le val_i\le 10^4)$

Output

输出最大得分

Sample Input

2
2
1 1 1
1 -1 1
3
1 1 1
1 -1 10
-1 0 100

Sample Output

1
1100

Solution

把所有点极角排序，用$y$轴开始逆时针旋转这条直线，每次把直线扫过的点放到另一侧，维护得分最大值即可

注：从$y$轴开始扫是因为极角最小值为$-\frac{\pi}{2}$，$atan(y/x)\in (-\frac{\pi}{2},\frac{\pi}{2})，atan2(y,x)\in(-\pi,\pi]$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define maxn 50005
#define PI acos(-1.0)
struct node
{
    int x,y,val;
    double angle;
    bool operator<(const node&b)const
    {
        return angle<b.angle;
    }
}p[maxn];
int T,n;
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(int i=0;i<n;i++)
        {
            scanf("%d%d%d",&p[i].x,&p[i].y,&p[i].val);
            if(p[i].x==0)p[i].angle=0.5*PI;
            else p[i].angle=atan(1.0*p[i].y/p[i].x);
        }
        sort(p,p+n);
        ll l=0,r=0;
        for(int i=0;i<n;i++)
            if(p[i].x>=0)r+=p[i].val;
            else l+=p[i].val;
        ll ans=l*r;
        for(int i=0;i<n;i++)
        {
            if(p[i].x>=0)r-=p[i].val,l+=p[i].val;
            else r+=p[i].val,l-=p[i].val;
            ans=max(ans,l*r);
        }
        printf("%I64d\n",ans);
    }
    return 0;
}