[BZOJ2683]简单题（cdq分治）

题目描述

题解

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define N 1000005

int n,T,cnt,x,y,p,q,v,opt,acnt,bcnt,pa,pb,tot;
int C
,ans
;
struct data{int x,y,v,ty,id;}opr
,a
,b
,ch
;
bool flag
;

int cmp(data a,data b)
{
return a.x<b.x;
}
void add(int loc,int val)
{
if (!loc) return;
for (int i=loc;i<=n;i+=i&-i)
C[i]+=val;
}
int query(int loc)
{
int ans=0;
for (int i=loc;i>=1;i-=i&-i)
ans+=C[i];
return ans;
}
void cdq(int l,int r)
{
if (l>=r) return;
int mid=(l+r)>>1;
cdq(l,mid);
acnt=0;
for (int i=l;i<=mid;++i) a[++acnt]=opr[i];
sort(a+1,a+acnt+1,cmp);
bcnt=0;
for (int i=mid+1;i<=r;++i) b[++bcnt]=opr[i];
sort(b+1,b+bcnt+1,cmp);
pa=pb=1;tot=0;
while (pb<=bcnt)
{
while (pa<=acnt&&a[pa].x<=b[pb].x)
{
if (!a[pa].ty)
{
add(a[pa].y,a[pa].v);
ch[++tot].x=a[pa].y,ch[tot].v=a[pa].v;
}
++pa;
}
ans[b[pb].id]+=query(b[pb].y)*b[pb].ty;
++pb;
}
for (int i=1;i<=tot;++i)
add(ch[i].x,-ch[i].v);
cdq(mid+1,r);
}
int main()
{
scanf("%d",&n);
while (~scanf("%d",&opt))
{
++T;
if (opt==3) break;
if (opt==1)
{
scanf("%d%d%d",&x,&y,&v);
opr[++cnt].x=x,opr[cnt].y=y,opr[cnt].ty=0,opr[cnt].v=v,opr[cnt].id=T;
}
else
{
flag[T]=1;
scanf("%d%d%d%d",&x,&y,&p,&q);
opr[++cnt].x=x-1,opr[cnt].y=y-1,opr[cnt].ty=1,opr[cnt].id=T;
opr[++cnt].x=p,opr[cnt].y=q,opr[cnt].ty=1,opr[cnt].id=T;
opr[++cnt].x=p,opr[cnt].y=y-1,opr[cnt].ty=-1,opr[cnt].id=T;
opr[++cnt].x=x-1,opr[cnt].y=q,opr[cnt].ty=-1,opr[cnt].id=T;
}
}
cdq(1,cnt);
for (int i=1;i<=T;++i)
if (flag[i]) printf("%d\n",ans[i]);
return 0;
}