题意:给出一个W*W的矩阵,有两个操作:1.对矩阵中的某一个点进行增加a的操作 2.询问矩阵中(x1,y1)为左下角(x2,y2)为右上角的矩阵包含的点权值和为多少
题解:首先,操作2可以分解为4个关于区间的查询如下图。
由于增加点权值的操作对其他操作互不影响,且可以离线,所以可以采用CDQ分治的离线做法。
对(L,R)的操作进行横坐标排序,对于插入操作按横坐标次序对操作时间处于(L,mid)操作按y坐标插入到树状数组中,对于查询操作按横坐标次序对操作时间处于(mid+1,R)的操作进行查询。
之后将树状数组中的操作还原以免对后续运算产生影响。
再将已经排好序的操作分为时间段(L,mid)与时间段(mid+1,R)的两个区间,分别继续进行以上操作,直到为一个元素。
由于每一层插入与查询总操作的时间为nlogw,一共需要进行logn轮,所以总的复杂度为nlognlogw,空间复杂度为nlogw
AC代码:
#include<stdio.h>
#include<algorithm>
using namespace std;
struct node
{
int op,x,y;
int t,id,A;
node(){}
node(int op,int x,int y,int t,int id,int A)
{
this->op=op;
this->x=x;
this->y=y;
this->t=t;
this->id=id;
this->A=A;
}
}qu[800005],newq[800005];
int initnum,w;
int c[2000005];
int ans[2000005];
int tot,top;
bool cmp(node a,node b)
{
if(a.x==b.x&&a.y==b.y)return a.op<b.op;
if(a.x==b.x)return a.y<b.y;
return a.x<b.x;
}
int lowbit(int i)
{
return i&(-i);
}
void add(int i,int k)
{
while(i<=w)
{
c[i]+=k;
i+=lowbit(i);
}
}
int sum(int i)
{
int ans=0;
while(i)
{
ans+=c[i];
i-=lowbit(i);
}
return ans;
}
void solve(int l,int r)
{
if(l==r)return ;
int mid=l+r>>1;
for(int i=l;i<=r;i++)
{
if(qu[i].op==1&&qu[i].t<=mid)add(qu[i].y,qu[i].A);
if(qu[i].op==2&&qu[i].t>mid)ans[qu[i].id]+=sum(qu[i].y)*qu[i].A;
}
for(int i=l;i<=r;i++)
if(qu[i].op==1&&qu[i].t<=mid)add(qu[i].y,-qu[i].A);
int l1=l,l2=mid+1;
for(int i=l;i<=r;i++)
{
if(qu[i].t>mid)newq[l2++]=qu[i];
if(qu[i].t<=mid)newq[l1++]=qu[i];
}
for(int i=l;i<=r;i++)
qu[i]=newq[i];
solve(l,mid);
solve(mid+1,r);
}
int main()
{
scanf("%d%d",&initnum,&w);
for(int i=1;i<=w;i++)add(i,initnum);
for(;;)
{
int op,x,y,v;
scanf("%d",&op);
if(op==1)
{
scanf("%d%d%d",&x,&y,&v);
qu[tot]=node(op,x,y,tot,0,v);
tot++;
}
else if(op==2)
{
int x2,y2;
scanf("%d%d%d%d",&x,&y,&x2,&y2);
qu[tot]=node(op,x2,y2,tot,top,1);
tot++;
qu[tot]=node(op,x-1,y2,tot,top,-1);
tot++;
qu[tot]=node(op,x2,y-1,tot,top,-1);
tot++;
qu[tot]=node(op,x-1,y-1,tot,top,1);
tot++;top++;
}
else break;
}
sort(qu,qu+tot,cmp);
solve(0,tot-1);
for(int i=0;i<top;i++)
printf("%d\n",ans[i]);
}