[BZOJ2683]简单题（cdq分治）

题目描述

传送门

题解

同BZOJ1176

代码

#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[N],ans[N];
struct data{int x,y,v,ty,id;}opr[N],a[N],b[N],ch[N];
bool flag[N];

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;
}