HDU - 5126: stars (求立方体内点数 CDQ套CDQ)

题意:现在给定空空的三维平面,有加点操作和询问立方体点数。

思路:考虑CDQ套CDQ。复杂度是O(NlogN*logN*logN),可以过此题。

具体的,这是一个四维偏序问题,4维分别是(times,x,y,z);我们知道cdq可以求出t<=T,x=X,y<=Y,在套一层就可以z<=Z了。那么一个立方体,我们拆为8个点来容斥。

然后现在的问题就是,求出(0,0,0)到(x,y,z)的点数。 第一维T已经默认排序了,我们先对X分治。 把所有问题分成两块,并且把左边这块标记o=-1,右边的标记o=1,然后塞到正常的CDQ里面,对y进行分治,对z进行统计。 此时一个z要插入树状数组,需要格外满足o=-1;而询问需要格外满足o=1;

对于CDQ:先对一维分治,两个小区间搞完后后,归并排序就可以了,这样避开了sort。 这里还可以为了方便用inplace_merge,但是比手写慢一些。

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
const int maxn=400010;
struct in{
    int x,y,z,id,opt,o; //编号,正负,第一维是否在左
}s[maxn],q[maxn],fcy[maxn];
bool cmp1(in w,in v) { return w.x<v.x;}
bool cmp2(in w,in v) { return w.y<v.y;}
int b[maxn],sz,tot,cnt,ans[maxn],sum[maxn];
void add(int x,int val)
{
    for(int i=x;i<=sz;i+=(-i)&i) sum[i]+=val;
}
int query(int x)
{
    int res=0;
    for(int i=x;i;i-=(-i)&i) res+=sum[i];
    return res;
}
void cdq2(int L,int R)
{
    if(L>=R) return ;
    int Mid=(L+R)>>1,i=L,j=Mid+1,now;
    cdq2(L,Mid); cdq2(Mid+1,R);
    for(;j<=R;j++){
        for(;i<=Mid&&q[i].y<=q[j].y;i++){
            if(q[i].o==-1&&q[i].opt==0) add(q[i].z,1);
        }
        if(q[j].o==1&&q[j].id) ans[q[j].id]+=q[j].opt*query(q[j].z);
    }
    while((--i)>=L) if(q[i].o==-1&&!q[i].id) add(q[i].z,-1);

    /*i=L,j=Mid+1,now=L; 手写归并,下面cdq1同理。
    while(j<=R&&i<=Mid){
        if(q[i].y<=q[j].y) fcy[now++]=q[i++];
        else  fcy[now++]=q[j++];
    }
    while(i<=Mid) fcy[now++]=q[i++];
    while(j<=R)  fcy[now++]=q[j++];
    rep(i,L,R) q[i]=fcy[i];*/
    inplace_merge(q+L,q+Mid+1,q+R+1,cmp2); //归并排序
}
void cdq1(int L,int R)
{
    if(L>=R) return ;
    int Mid=(L+R)>>1,i=L,j=Mid+1,now=L;
    cdq1(L,Mid);  cdq1(Mid+1,R);
    //sort(s+L,s+Mid+1,cmp1); sort(s+Mid+1,s+R+1,cmp1);
    tot=0;
    while(j<=R&&i<=Mid){
        if(s[i].x<=s[j].x){
            fcy[now++]=s[i];
            q[++tot]=s[i++],q[tot].o=-1;
            if(s[i-1].id) tot--;
        }
        else {
            fcy[now++]=s[j];
            q[++tot]=s[j++],q[tot].o=1;
            if(!s[j-1].id) tot--;
        }
    }
    while(i<=Mid) {
        fcy[now++]=s[i];
        q[++tot]=s[i++],q[tot].o=-1;
        if(s[i-1].id) tot--;
    }
    while(j<=R) {
        fcy[now++]=s[j];
        q[++tot]=s[j++],q[tot].o=1;
        if(!s[j-1].id) tot--;
    }
    rep(i,L,R) s[i]=fcy[i];
    //inplace_merge(s+L,s+Mid+1,s+R+1,cmp1);
    cdq2(1,tot);
}
void solve()
{
    int N,Q=0,opt,x1,y1,z1,x2,y2,z2;
    sz=0; tot=0;
    scanf("%d",&N);
    rep(i,1,N){
        scanf("%d",&opt);
        if(opt&1){
            scanf("%d%d%d",&x1,&y1,&z1);
            b[++sz]=z1;
            s[++tot]=in{x1,y1,z1,0,0,0};
        }
        else {
            scanf("%d%d%d%d%d%d",&x1,&y1,&z1,&x2,&y2,&z2);
            b[++sz]=z1; b[++sz]=z2; b[++sz]=z1-1; ans[++Q]=0;
            s[++tot]=in{x2,y2,z2,Q,1,0};
            s[++tot]=in{x1-1,y1-1,z1-1,Q,-1,0};
            s[++tot]=in{x1-1,y2,z2,Q,-1,0};
            s[++tot]=in{x1-1,y1-1,z2,Q,1,0};
            s[++tot]=in{x1-1,y2,z1-1,Q,1,0};
            s[++tot]=in{x2,y1-1,z2,Q,-1,0};
            s[++tot]=in{x2,y2,z1-1,Q,-1,0};
            s[++tot]=in{x2,y1-1,z1-1,Q,1,0};
        }
    }
    sort(b+1,b+sz+1);
    cnt=unique(b+1,b+sz+1)-(b+1);
    rep(i,1,tot) s[i].z=lower_bound(b+1,b+sz+1,s[i].z)-b;
    cdq1(1,tot);
    rep(i,1,Q) printf("%d\n",ans[i]);
}
int main()
{
    int T; scanf("%d",&T);
    while(T--) solve();
    return 0;
}

 

It is your time to fight!
posted @ 2019-09-13 09:05  nimphy  阅读(246)  评论(0编辑  收藏  举报