BZOJ 1176: [Balkan2007]Mokia

BZOJ 1176: [Balkan2007]Mokia

标签（空格分隔）： OI BZOJ cdq分治 分治 树状数组

---

Time Limit: 30 Sec
Memory Limit: 162 MB

---

Description

维护一个W*W的矩阵，初始值均为S.每次操作可以增加某格子的权值,或询问某子矩阵的总权值.修改操作数M<=160000,询问数Q<=10000,W<=2000000.

Input

第一行两个整数,S,W;其中S为矩阵初始值;W为矩阵大小

接下来每行为一下三种输入之一(不包含引号):

"1 x y a"

"2 x1 y1 x2 y2"

"3"

输入1:你需要把(x,y)(第x行第y列)的格子权值增加a

输入2:你需要求出以左上角为(x1,y1),右下角为(x2,y2)的矩阵内所有格子的权值和,并输出

输入3:表示输入结束

Output

对于每个输入2,输出一行,即输入2的答案

Sample Input

0 4

1 2 3 3

2 1 1 3 3

1 2 2 2

2 2 2 3 4

3
Sample Output

3

5
HINT

保证答案不会超过int范围

---

Solution####

cdq分治+树状数组 (O(n*log_2n))。2008年cdq的国家集训队作业有相关资料

---

Code####

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<queue>
#include<set>
#include<map>
#include<bitset>
#include<vector>
using namespace std;
#define PA pair<int,int>
const int N=0,M=0;
int read()
{int s=0,f=1;char ch=getchar();
 while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
 while(ch>='0'&&ch<='9'){s=(s<<1)+(s<<3)+ch-'0';ch=getchar();}
 return s*f;
}
//smile please
int n,T,m;
struct que
{
	bool ask;int x,y,s,t,h;
}q[800005];int qs;
bool operator<(que a,que b)
{
	return a.x<b.x||(a.x==b.x&&a.ask<b.ask);
}
int s[2000005];
que *a[800005];
que *aa[800005];
int ans[200005];
void gb(int l,int r)
{
	int mid=l+r>>1;
	int i=l,j=mid+1,k=l;
	for(int i=l;i<=r;i++)
	   aa[i]=a[i];
	while(i!=mid+1||j!=r+1)
	  {if(i==mid+1)a[k++]=aa[j++];
	   else if(j==r+1)a[k++]=aa[i++];
	   else if(*aa[i]<*aa[j])a[k++]=aa[i++];
	   else a[k++]=aa[j++];
	  }
}
void cdq(int l,int r)
{
	if(l==r)return;
	int mid=l+r>>1;
	cdq(l,mid);cdq(mid+1,r);
	gb(l,r);
	for(int i=l;i<=r;i++)
	   {que now=*a[i];
	    if(now.ask)
	     {if(now.h<=mid)continue;
	      int sum=0;
		  for(int x=now.y;x;x-=x&-x)
	         sum+=s[x];
	      ans[now.t]+=sum*now.s;
		 }
	   else
	     {if(now.h>mid)continue;
	      for(int x=now.y;x<=n;x+=x&-x)
	         s[x]+=now.s;
		 }
	   }
	for(int i=l;i<=r;i++)
	   {que now=*a[i];
	    if(!now.ask)
	      {if(now.h>mid)continue;
	       for(int x=now.y;x<=n;x+=x&-x)
	          s[x]=0;
		  }
	   }
}
int main()
{
	m=read();n=read();T=1;
	for(int op=read();op!=3;op=read(),++T)
	    if(op==1)
	      q[++qs]=(que){0,read(),read(),read(),T,qs},ans[T]=-1;
		else
		  {int x1=read(),y1=read(),x2=read(),y2=read();
		   q[++qs]=(que){1,x2,y2,1,T,qs};
		   q[++qs]=(que){1,x2,y1-1,-1,T,qs};
		   q[++qs]=(que){1,x1-1,y2,-1,T,qs};
		   q[++qs]=(que){1,x1-1,y1-1,1,T,qs};
		   ans[T]+=(x2-x1)*(y2-y1)*m;
		  }
	for(int i=1;i<=qs;i++)a[i]=&q[i];
	cdq(1,qs);
	for(int i=1;i<T;i++)
	   if(ans[i]!=-1)
	     printf("%d
",ans[i]);
	return 0;
}