【kd-tree】bzoj2850 巧克力王国

分四种情况讨论：a,b>=0
a,b<0
a>=0,b<0
a<0,b>=0
然后每次检验是否进入一个矩形框或者是否直接利用这个矩形框的答案仅仅利用两个对角的坐标进行更新即可。
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
#define N 50001
#define INF 2147483647
#define KD 2//Î¬¶ÈÊý
int qp[KD];
ll lim;
int n,root,m;
bool dn;
struct Node
{
    int minn[KD],maxx[KD],p[KD],w;
    int ch[2];
    ll sumv;
    void Init()
      {
        for(int i=0;i<KD;++i)
          minn[i]=maxx[i]=p[i];
        sumv=(ll)w;
      }
}T[N];
void Update(int rt)
{
    for(int i=0;i<2;++i)
      if(T[rt].ch[i])
        {
          T[rt].sumv+=T[T[rt].ch[i]].sumv;
          for(int j=0;j<KD;++j)
            {
              T[rt].minn[j]=min(T[rt].minn[j],T[T[rt].ch[i]].minn[j]);
              T[rt].maxx[j]=max(T[rt].maxx[j],T[T[rt].ch[i]].maxx[j]);
            }
        }
}
bool operator < (const Node &a,const Node &b){return a.p[dn]<b.p[dn];}
int Buildtree(int l=1,int r=n,bool d=0)
{
    dn=d;
    int m=(l+r>>1);
    nth_element(T+l,T+m,T+r+1);
    T[m].Init();
    if(l!=m) T[m].ch[0]=Buildtree(l,m-1,d^1);
    if(m!=r) T[m].ch[1]=Buildtree(m+1,r,d^1);
    Update(m);
    return m;
}
ll ans;
void Query0(int rt=root)//a>0,b>0
{
    if((ll)T[rt].p[0] * (ll)qp[0] + (ll)T[rt].p[1] * (ll)qp[1] < lim)
      ans+=(ll)T[rt].w;
    for(int i=0;i<2;++i)
      if(T[rt].ch[i] &&
      (ll)T[T[rt].ch[i]].minn[0] * (ll)qp[0] +
      (ll)T[T[rt].ch[i]].minn[1] * (ll)qp[1] < lim)
        {
          if((ll)T[T[rt].ch[i]].maxx[0] * (ll)qp[0] +
          (ll)T[T[rt].ch[i]].maxx[1] * (ll)qp[1] < lim)
            ans+=T[T[rt].ch[i]].sumv;
          else
            Query0(T[rt].ch[i]);
        }
}
void Query1(int rt=root)//a>0,b<0
{
    if((ll)T[rt].p[0] * (ll)qp[0] + (ll)T[rt].p[1] * (ll)qp[1] < lim)
      ans+=(ll)T[rt].w;
    for(int i=0;i<2;++i)
      if(T[rt].ch[i] &&
      (ll)T[T[rt].ch[i]].minn[0] * (ll)qp[0] +
      (ll)T[T[rt].ch[i]].maxx[1] * (ll)qp[1] < lim)
        {
          if((ll)T[T[rt].ch[i]].maxx[0] * (ll)qp[0] +
          (ll)T[T[rt].ch[i]].minn[1] * (ll)qp[1] < lim)
            ans+=T[T[rt].ch[i]].sumv;
          else
            Query1(T[rt].ch[i]);
        }
}
void Query2(int rt=root)//a<0,b>0
{
    if((ll)T[rt].p[0] * (ll)qp[0] + (ll)T[rt].p[1] * (ll)qp[1] < lim)
      ans+=(ll)T[rt].w;
    for(int i=0;i<2;++i)
      if(T[rt].ch[i] &&
      (ll)T[T[rt].ch[i]].maxx[0] * (ll)qp[0] +
      (ll)T[T[rt].ch[i]].minn[1] * (ll)qp[1] < lim)
        {
          if((ll)T[T[rt].ch[i]].minn[0] * (ll)qp[0] +
          (ll)T[T[rt].ch[i]].maxx[1] * (ll)qp[1] < lim)
            ans+=T[T[rt].ch[i]].sumv;
          else
            Query2(T[rt].ch[i]);
        }
}
void Query3(int rt=root)//a<0,b<0
{
    if((ll)T[rt].p[0] * (ll)qp[0] + (ll)T[rt].p[1] * (ll)qp[1] < lim)
      ans+=(ll)T[rt].w;
    for(int i=0;i<2;++i)
      if(T[rt].ch[i] &&
      (ll)T[T[rt].ch[i]].maxx[0] * (ll)qp[0] +
      (ll)T[T[rt].ch[i]].maxx[1] * (ll)qp[1] < lim)
        {
          if((ll)T[T[rt].ch[i]].minn[0] * (ll)qp[0] +
          (ll)T[T[rt].ch[i]].minn[1] * (ll)qp[1] < lim)
            ans+=T[T[rt].ch[i]].sumv;
          else
            Query3(T[rt].ch[i]);
        }
}
int main()
{
//  freopen("bzoj2850.in","r",stdin);
//  freopen("bzoj2716.out","w",stdout);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i)
      scanf("%d%d%d",&T[i].p[0],&T[i].p[1],&T[i].w);
    Buildtree();
    root=(1+n>>1);
    for(int i=1;i<=m;++i)
      {
        ans=0;
        scanf("%d%d%lld",&qp[0],&qp[1],&lim);
        if(qp[0]>=0 && qp[1]>=0)
          Query0();
        else if(qp[0]>=0 && qp[1]<0)
          Query1();
        else if(qp[0]<0 && qp[1]>=0)
          Query2();
        else
          Query3();
        printf("%lld
",ans);
      }
    return 0;
}