BZOJ1935: [Shoi2007]Tree 园丁的烦恼

BZOJ1935传送门

$1176,2683$三倍$EXP$。
戳这里–>BZOJ1176

【代码】

#include <cstdio>
#include  <iostream>
#include <algorithm>
#define N 500005
#define M 10000005
using namespace std;
typedef long long ll;
int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,m,tot,numa,maxy;
int szsz[M],ans[N];

class Query{
    public:
        int type,x,y,w,id;
    Query(){}
    Query(int tt,int xx,int yy,int zz,int ii){
        type=tt,x=xx,y=yy,w=zz,id=ii;
    }
}Q[2500005],tmp[2500005]; 

bool operator <(Query a,Query b){
    return a.x<b.x||(a.x==b.x&&a.type<b.type);
}

int lowbit(int x){return x&-x;}
void Sum_Up(int x){
    for(int i=x;i<=maxy;i+=lowbit(i)) szsz[i]++;
}
int query(int x){
    int rtn=0;
    for(int i=x;i;i-=lowbit(i)) rtn+=szsz[i];
    return rtn;
}
void Clear(int x){
    for(int i=x;i<=maxy&&szsz[i];i+=lowbit(i)) szsz[i]=0;
}

void CDQ(int L,int R)
{
    if(L==R) return;
    int mid=L+R>>1;CDQ(L,mid);CDQ(mid+1,R);
    int p=L,q=mid+1,o=0;
    while(p<=mid&&q<=R) {
        if(Q[p]<Q[q]) {
            if(!Q[p].type) Sum_Up(Q[p].y);
            tmp[o++]=Q[p++];
        }
        else {
            if(Q[q].type) ans[Q[q].id]+=query(Q[q].y)*Q[q].w;
            tmp[o++]=Q[q++];
        }
    }
    while(p<=mid) tmp[o++]=Q[p++];
    while(q<=R) {
        if(Q[q].type) ans[Q[q].id]+=query(Q[q].y)*Q[q].w;
        tmp[o++]=Q[q++];
    }
    for(int i=0;i<o;i++) Q[i+L]=tmp[i],Clear(tmp[i].y);
}

int main()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++)
    {
        static int x,y;x=read()+1,y=read()+1;
        Q[++tot]=Query(0,x,y,0,0);maxy=max(maxy,y);
    }   
    for(int i=1;i<=m;i++)
    {
        static int xx,yy,x,y;
        x=read()+1,y=read()+1,xx=read()+1,yy=read()+1;maxy=max(maxy,yy);
        Q[++tot]=Query(1,x-1,y-1,1,i);Q[++tot]=Query(1,x-1,yy,-1,i);
        Q[++tot]=Query(1,xx,y-1,-1,i);Q[++tot]=Query(1,xx,yy,1,i);
    }
    CDQ(1,tot);
    for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
    return 0;
}