HDU 6127 Hard challenge(几何 多校第七场)

• 题目大意

  平面上有n个点,每个点有一个价值，每两个点之间都有一条线段，定义线段的值为两个点价值的乘积，现在让你找一条过原点的直线(直线不经过任何一个节点)，将这条直线所经过的所有线段的值求和，问最大的和是多少.

• 分析

  我们容易知道如果确定了直线的位置，那么将直线两边的点分别求和再乘起来就是答案

  那么现在只需要枚举直线的位置即可，只有当直线扫过点的时候值才会发生变化，所以我们离散化地取枚举每个节点，在跨过节点的时候只需要$O(1)$的复杂度就能对答案进行更新。

• 代码

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<queue>
#include<map>
#include<algorithm>
#include<set>
#include<stack>
using namespace std;
#define  LL long long int
const double pi=4*atan(1.0);
const int  MAXN=50009;
int T;
int n;
LL sum_up,sum_down;
struct Node
{
    int x,y,v;
    double r;
    bool is_up;//1表示在x轴上面
}node[MAXN];
double Get_r(int x,int y)//给定一个点的坐标，如果这个点在x轴上方，返回这个点和原点连线与轴正方向的夹角。如果这个点在x轴下方，返回这个点和原点连线与轴负方向的夹角
{
    if(x==0)return pi;
    if(y==0)return 0;
    if(x<0 && y>0)return pi-atan(-(double)y/(double)x);
    if(x>0 && y<0)return pi-atan(-(double)y/(double)x);
    return atan((double)y/(double)x);
}
bool Get_up(int x,int y)//判断一个节点在x轴上方还是下方，在x轴上的点正方向归为下方，负方向归为上方
{
    if(y>0)return 1;
    else if(y<0)return 0;
    else
    {
        if(x<0)return 1;
        else return 0;
    }
}
bool cmp(Node a,Node b)
{
    return a.r < b.r;
}
void In()
{
    sum_up=sum_down=0;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
         scanf("%d%d%d",&node[i].x,&node[i].y,&node[i].v);
         if(node[i].x==0 && node[i].y==0)node[i].v=0;
         node[i].r=Get_r(node[i].x,node[i].y);
         node[i].is_up=Get_up(node[i].x,node[i].y);
         if(node[i].is_up==1)sum_up+=node[i].v;
         else sum_down+=node[i].v;
    }
}
LL Work()
{
    LL ans=sum_up*sum_down;
    double cur_r=0;
    int i=1;
    while(node[i].r==0)i++;
    while(i<=n)
    {
        cur_r=node[i].r;
        while(node[i].r==cur_r)
        {
            if(node[i].is_up==1)
            {
                sum_up-=node[i].v;
                sum_down+=node[i].v;
            }
            else
            {
                sum_up+=node[i].v;
                sum_down-=node[i].v;
            }
            i++;
        }
        ans=max(ans,sum_down*sum_up);
    }
    return ans;
}
void Test()
{
    cout<<"sum_up="<<sum_up<<",sum_down="<<sum_down<<endl;
    for(int i=1;i<=n;i++)cout<<node[i].is_up<<" ";
    cout<<endl;
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        In();
        //Test();
        sort(node+1,node+n+1,cmp);
        cout<<Work()<<endl;
    }
}
/*
2
2
1 1 1
1 -1 1
3
1 1 1
1 -1 10
-1 0 100

1
4
1 0 4
0 1 5
0 -1 3
-2 0 6

*/