1176: [Balkan2007]Mokia/2683: 简单题

题目链接

题目大意：维护一个W*W的矩阵，每次操作可以增加某格子的权值,或询问某子矩阵的总权值

题解：CDQ分治

先按x排序保证每一层处理时x都有序，然后对时间分治

询问拆成4个差分计算

我的收获：orz CDQ

#include <bits/stdc++.h>
using namespace std;

#define ll long long
#define N 200005

int s,w,m;
int ans[10005];

struct Tree{
    int c[N*10];
    void add(int x,int v){for(;x<=w;x+=x&-x) c[x]+=v;}
    int ask(int x){int ret=0;for(;x;x-=x&-x) ret+=c[x];return ret;}
}T;

struct ques{
    int x,y,val,pos,id,opt;
    void readin(){scanf("%d%d%d",&x,&y,&val);id=m;}
}q[N],tmp[N];

bool operator<(ques a,ques b)
{
    if(a.x==b.x&&a.y==b.y) return a.opt<b.opt;
    if(a.x==b.x) return a.y<b.y;
    return a.x<b.x;
}

void addquery()
{
    int x1,y1,x2,y2;scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
    int now=++ans[0];
    q[++m].pos=now;q[m].id=m;q[m].x=x1-1;q[m].y=y1-1;q[m].val=1;q[m].opt=1;
    q[++m].pos=now;q[m].id=m;q[m].x=x2;q[m].y=y2;q[m].val=1;q[m].opt=1;
    q[++m].pos=now;q[m].id=m;q[m].x=x1-1;q[m].y=y2;q[m].val=-1;q[m].opt=1;
    q[++m].pos=now;q[m].id=m;q[m].x=x2;q[m].y=y1-1;q[m].val=-1;q[m].opt=1;
}

void cdq(int l,int r)
{
    if(l==r) return;
    int mid=(l+r)>>1,l1=l,l2=mid+1;
    for(int i=l;i<=r;i++)
    {
        if(q[i].id<=mid&&!q[i].opt) T.add(q[i].y,q[i].val);
        if(q[i].id>mid&&q[i].opt) ans[q[i].pos]+=q[i].val*T.ask(q[i].y);
    }
    for(int i=l;i<=r;i++) if(q[i].id<=mid&&!q[i].opt) T.add(q[i].y,-q[i].val);
    for(int i=l;i<=r;i++)
        if(q[i].id<=mid) tmp[l1++]=q[i];
        else tmp[l2++]=q[i];
    for(int i=l;i<=r;i++) q[i]=tmp[i];
    cdq(l,mid);cdq(mid+1,r);
}

void work()
{
    cdq(1,m);
    for(int i=1;i<=ans[0];i++) printf("%d\n",ans[i]);
}

void init()
{
    scanf("%d%d",&s,&w);
    int opt=0;
    while(opt!=3)
    {
        scanf("%d",&opt);
        if(opt==1) q[++m].readin();
        if(opt==2) addquery();
    }
    sort(q+1,q+m+1);
}

int main()
{
    init();
    work();
    return 0;
}