HDU5126 stars(CDQ分治)

传送门

大意：

向三维空间中加点，询问一个三维区间中点的个数。

解题思路：

带修改CDQ，将修改和询问一起插入CDQ分治询问。

（询问可以由8个前缀和加减操作实现）

其中第一层CDQ维护x有序。

第二层CDQ维护y有序并且将z离线处理完更新答案。

注意要将原数组的辅助数组推入第二层CDQ否则会将顺序毁坏。

代码：

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 const int N=;
 const int M=;
 struct data{
     int t;
     int x;
     int y;
     int tz;
     int z;
     int c;
     bool left;
 }d[N],o[N],tmp[N],stdata;
 int n,T;
 int cnt;
 int tot;
 int ans[N];
 int sek[N];
 int line[M];
 bool cmpz(data a,data b){return a.tz<b.tz;}
 bool cmpb(data a,data b){return a.t<b.t;}
 int lowbit(int x){return x&(-x);}
 void update(int pos,int v)
 {
     while(pos<=tot)
     {
         line[pos]+=v;
         pos+=lowbit(pos);
     }
     return ;
 }
 int query(int pos)
 {
     int ans=;
     while(pos)
     {
         ans+=line[pos];
         pos-=lowbit(pos);
     }
     return ans;
 }
 void Get(void)
 {
     tot=;
     std::sort(d+,d+cnt+,cmpz);
     d[].z=;
     for(int i=;i<=cnt;i++)
     {
         if(d[i].tz!=d[i-].tz)
             tot++;
         d[i].z=tot;
     }
     std::sort(d+,d+cnt+,cmpb);
     return ;
 }
 void cdq(int l,int r)
 {
     if(l==r)
         return ;
     int mid=(l+r)>>;
     cdq(l,mid);
     cdq(mid+,r);
     int j=l;
     for(int i=mid+;i<=r;i++)
     {
         for(;j<=mid&&o[j].y<=o[i].y;j++)
             if(!o[j].c&&o[j].left)
                 update(o[j].z,);
         if(o[i].left)
             continue;
         ans[o[i].t]+=o[i].c*query(o[i].z);
     }
     for(int i=l;i<j;i++)
         if(!o[i].c&&o[i].left)
             update(o[i].z,-);
     int sta1=l,sta2=mid+;
     for(int i=l;i<=r;i++)
         if(sta1<=mid&&(sta2>r||o[sta1].y<=o[sta2].y))
             tmp[i]=o[sta1++];
         else
             tmp[i]=o[sta2++];
     for(int i=l;i<=r;i++)
         o[i]=tmp[i];
     return ;
 }
 void CDQ(int l,int r)
 {
     if(l==r)
         return ;
     int mid=(l+r)>>;
     CDQ(l,mid);
     CDQ(mid+,r);
     for(int i=l;i<=r;i++)
         d[i].left=(i<=mid);
     int sta1=l,sta2=mid+;
     for(int i=l;i<=r;i++)
         if(sta1<=mid&&(sta2>r||d[sta1].x<=d[sta2].x))
             o[i]=d[sta1++];
         else
             o[i]=d[sta2++];
     for(int i=l;i<=r;i++)
         d[i]=o[i];
     cdq(l,r);
     for(int i=l;i<=r;i++)
         d[i].left=false;
     return ;
 }
 int main()
 {
     //freopen("a.in","r",stdin);
     scanf("%d",&T);
     while(T--)
     {
         for(int i=;i<=n;i++)
             ans[i]=;
         for(int i=;i<=cnt;i++)
             d[i]=tmp[i]=stdata;
         cnt=n=;
         scanf("%d",&n);
         for(int i=;i<=n;i++)
         {
             int cmd;
             scanf("%d",&cmd);
             sek[i]=cmd;
             if(cmd==)
             {
                 int x,y,z;
                 scanf("%d%d%d",&x,&y,&z);
                 d[++cnt]=(data){i,x,y,z,,,};
             }else{
                 int ax,ay,az,bx,by,bz;
                 scanf("%d%d%d%d%d%d",&ax,&ay,&az,&bx,&by,&bz);
                 if(ax>bx)
                     std::swap(ax,bx);
                 if(ay>by)
                     std::swap(ay,by);
                 if(az>bz)
                     std::swap(az,bz);
                 ax--,ay--,az--;
                 d[++cnt]=(data){i,ax,ay,az,,-,};
                 d[++cnt]=(data){i,ax,ay,bz,,,};
                 d[++cnt]=(data){i,ax,by,az,,,};
                 d[++cnt]=(data){i,bx,ay,az,,,};
                 d[++cnt]=(data){i,ax,by,bz,,-,};
                 d[++cnt]=(data){i,bx,ay,bz,,-,};
                 d[++cnt]=(data){i,bx,by,az,,-,};
                 d[++cnt]=(data){i,bx,by,bz,,,};
             }
         }

         Get();
         CDQ(,cnt);
         for(int i=;i<=n;i++)
             if(sek[i]==)
                 printf("%d\n",ans[i]);
     }
     return ;
 }