【bzoj2850】巧克力王国

Description

巧克力王国里的巧克力都是由牛奶和可可做成的。但是并不是每一块巧克力都受王国人民的欢迎，因为大家都不喜
欢过于甜的巧克力。对于每一块巧克力，我们设x和y为其牛奶和可可的含量。由于每个人对于甜的程度都有自己的
评判标准，所以每个人都有两个参数a和b，分别为他自己为牛奶和可可定义的权重，因此牛奶和可可含量分别为x
和y的巧克力对于他的甜味程度即为ax + by。而每个人又有一个甜味限度c，所有甜味程度大于等于c的巧克力他都
无法接受。每块巧克力都有一个美味值h。现在我们想知道对于每个人，他所能接受的巧克力的美味值之和为多少
Input

第一行两个正整数n和m，分别表示巧克力个数和询问个数。接下来n行，每行三个整数x,y,h，含义如题目所示。再
接下来m行，每行三个整数a,b,c，含义如题目所示。
Output

输出m行，其中第i行表示第i个人所能接受的巧克力的美味值之和。
Sample Input

3 3

1 2 5

3 1 4

2 2 1

2 1 6

1 3 5

1 3 7
Sample Output

5

0

4
HINT

1 <= n, m <= 50000，1 <= 10^9，-10^9 <= a, b, x, y <= 10^9。

Source

Violet 0

题解
kd-tree 模板题。

代码

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#define inf 100000000
#define ll long long
#define mo 1000000007
#define N 50005
using namespace std;
struct P
{
    int d[2],mn[2],mx[2],l,r,v;
    ll sum;
}p[N],t[N];
int n,m,D,rt;
ll ans,a,b,c;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
bool operator<(P a,P b)
{
    return a.d[D]<b.d[D];
}
void update(int k)
{
    int l=t[k].l,r=t[k].r;
    for (int i=0;i<2;i++)
    {
        if (l) t[k].mn[i]=min(t[k].mn[i],t[l].mn[i]),t[k].mx[i]=max(t[k].mx[i],t[l].mx[i]);
        if (r) t[k].mn[i]=min(t[k].mn[i],t[r].mn[i]),t[k].mx[i]=max(t[k].mx[i],t[r].mx[i]);
    }
    t[k].sum=t[l].sum+t[r].sum+t[k].v;
}
int build(int l,int r,int now)
{
    D=now;
    int mid=(l+r)>>1;
    nth_element(p+l,p+mid,p+r+1);
    t[mid]=p[mid];
    for (int i=0;i<2;i++)
        t[mid].mx[i]=t[mid].mn[i]=p[mid].d[i];
    if (l<mid) t[mid].l=build(l,mid-1,now^1);
    if (mid<r) t[mid].r=build(mid+1,r,now^1);
    update(mid);
    return mid;
}
bool check(int x,int y)
{
    return x*a+b*y<c;
}
int cal(P p)
{
    int tmp=0;
    tmp+=check(p.mn[0],p.mn[1]);
    tmp+=check(p.mx[0],p.mn[1]);
    tmp+=check(p.mn[0],p.mx[1]);
    tmp+=check(p.mx[0],p.mx[1]);
    return tmp;
}
void query(int k)
{
    int l=t[k].l,r=t[k].r;
    if (check(t[k].d[0],t[k].d[1])) ans+=t[k].v;
    int tmpl,tmpr;
    if (l) tmpl=cal(t[l]);
    if (r) tmpr=cal(t[r]);
    if (tmpl==4) ans+=t[l].sum;
    else if (tmpl) query(l);
    if (tmpr==4) ans+=t[r].sum;
    else if (tmpr) query(r);
}
int main()
{
    n=read();m=read();
    for (int i=1;i<=n;i++)
    {
        p[i].d[0]=read();p[i].d[1]=read();p[i].v=read();
    }
    rt=build(1,n,0);
    while (m--)
    {
        a=read(),b=read(),c=read();
        ans=0;
        query(rt);
        printf("%lld\n",ans);
    }
    return 0;
}