【BZOJ】1176: [Balkan2007]Mokia-cdq分治

传送门：bzoj1176

---

代码

cdq模板，不解释

#include<bits/stdc++.h>
using namespace std;
const int M=2e6+10;
const int N=17e4+19;
int n,m,op,xa,ya,xb,yb,cnt;
int s,w,as[10004],tmp[N<<2],cur;
int tre[M];

struct P{
    int x,y,val,op,id,bel;
    P(int x=0,int y=0,int val=0,int op=0,int id=0,int bel=0):x(x),y(y),val(val),op(op),id(id),bel(bel){}
}t[N<<2],tp[N<<2];

bool cmp(const P& a,const P& b)
{if(a.x==b.x) if(a.y==b.y) return a.op<b.op;else return a.y<b.y;else return a.x<b.x;}

inline int rd(){
    char ch;int x=0,f=1;
    while(!isdigit(ch=getchar())){if(ch=='-') f=0;}
    x=(ch^48);
    while(isdigit(ch=getchar())){x=(x<<3)+(x<<1)+(ch^48);}
    return f?x:-x;
}

inline void ad(int x,int val)
{
   for(;x<=w;x+=x&(-x)) 
     tre[x]+=val;
     }
inline int get(int x)
{int ret=0;for(;x;x-=x&(-x)) ret+=tre[x];return ret;}

inline void cdq(int l,int r)
{
    if(l==r) return;cur=0;
    int mid=(l+r)>>1,L=l,R;R=mid+1;
    for(int i=l;i<=r;i++){
        if(t[i].id<=mid && !t[i].op) ad(t[i].y,t[i].val),tmp[++cur]=i;
        else if(t[i].id>mid && t[i].op) as[t[i].bel]+=t[i].val*get(t[i].y);
    }
    for(int i=1;i<=cur;i++) ad(t[tmp[i]].y,-t[tmp[i]].val);
    for(int i=l;i<=r;i++){
        if(t[i].id<=mid) tp[L++]=t[i];
        else tp[R++]=t[i];
    }
    for(int i=l;i<=r;i++) t[i]=tp[i]; 
    cdq(l,mid);cdq(mid+1,r);
}

int main(){
   s=rd();w=rd();
   while(1){
       op=rd();
       if(op==3) break;
       if(op==1){
         t[++m].x=rd();t[m].y=rd();
         t[m].val=rd();t[m].id=m;
       }else{
          ++cnt;xa=rd()-1,ya=rd()-1,xb=rd(),yb=rd();
          t[++m]=P(xa,ya,1,1,m,cnt);
          t[++m]=P(xb,ya,-1,1,m,cnt);
          t[++m]=P(xa,yb,-1,1,m,cnt);
          t[++m]=P(xb,yb,1,1,m,cnt);
       }
   }    
   sort(t+1,t+m+1,cmp);
   cdq(1,m);
   for(int i=1;i<=cnt;i++) printf("%d\n",as[i]);
   return 0;
}