2017西安交大ACM小学期数据结构 [分块、二维矩阵]

Problem B

发布时间: 2017年6月28日 10:06   最后更新: 2017年6月28日 16:35   时间限制: 2000ms   内存限制: 32M

描述

给定一个 n×m 的矩形, 其中第 i 行第 j 列的值为 ai,j

给出 q 个操作, 操作有两种

对于形如 1   x1   y1   x2   y2   z  的操作, 将 (x1,y1) - (x2,y2) 这段矩形区域的所有元素加上 z , 满足 1≤x1≤x2≤n ,  1≤y1≤y2≤m ,  1≤z≤105

对于形如 2   x   y 的操作, 输出 ax,y , 满足 1≤x≤n ,  1≤y≤m

998≤n≤1000 ,  998≤m≤1000 ,  9×104≤q≤105 ,  1≤ai,j≤105

输入

第一行三个整数 n ,  m ,  q , 意义如上所述。
接下来 n 行, 每行 m 个整数, 表示序列 a 。
再接下来 q 行, 每行第一个数为 opt , 如果 opt=1 , 则后面紧跟五个数, 意义如上所述; 如果 opt=2 , 则后面紧跟两个数, 意义如上所述。

输出

对于所有操作 2 , 输出答案, 一行一个。

样例输入1  复制

3 4 3
3 1 4 1
5 9 2 6
5 3 5 8
2 2 2
1 1 1 3 4 6
2 2 2

样例输出1

9
15

题解：

这道题目给出的操作是对要做区间修改，端点查询。我们可以仿照《数据结构训练1》的思路，构造出一个序列b，然后单点查询变成了对序列b进行求和，区间修改变成了

对序列b进行的单点修改。

经过尝试，我们定义b[i][j] = a[i][j]+a[i-1][j-1]-a[i][j-1]-a[i-1][j]

这样的话 sum(b[i][j])就等于a[i][j]

而我们对于区间x1,y1,x2,y2进行修改(加上在z)就相当于

在点b[x1][y1]处加上z

在b[x1][y2+1]处减去z

在b[x2+1][y1]处减去z

在b[x2+1][y2+1]处加上z

做以上4步操作。

这样的话求和就很简单了，现在可以用两种方法来实现求和

比较简单的一种想法是用二维树状数组来维护，这里就不多说了。

还有一种思路就是用分块的方法，在O(q(sqrt n*m))的复杂度内完成

由于二维矩阵在分块上比较复杂。

我们转向对询问进行分块。

每块地大小是s = sqrt(n*m)

每经过s个询问，我们都要对b数组和sum数组进行重构。

时间复杂度就是O(q(sqrt n*m))

代码：

#include <cstdio>
#include <cmath>
const int MAX = 1007;
typedef long long LL; 
LL cx1[100007];
LL cy1[100007];
LL cx2[100007];
LL cy2[100007];
LL cv[100007];
LL arr[MAX][MAX];
LL brr[MAX][MAX];
LL sum2[MAX][MAX];
LL tot = 0;
LL las = 0;
LL n,m,q;
LL s;
void cg(LL x1,LL y1,LL x2,LL y2,LL val){
	cx1[tot] = x1;
	cy1[tot] = y1;
	cx2[tot] = x2;
	cy2[tot] = y2;
	cv[tot] = val;
	tot ++;
	if(las + s == tot ){
		//重构 
		for(LL i = las;i < tot;i++){
			brr[cx1[i]][cy1[i]] += cv[i];
			brr[cx2[i]+1][cy2[i]+1] += cv[i];
			brr[cx1[i]][cy2[i]+1] -= cv[i];
			brr[cx2[i]+1][cy1[i]] -= cv[i];
		}
		las = tot;
		for(LL i = 1;i <= n;i++){
			for(LL j = 1;j <= m;j++){
				sum2[i][j] = sum2[i][j-1] + sum2[i-1][j] - sum2[i-1][j-1] + brr[i][j];
			}
		} 
	}
}
LL sums(LL x,LL y){
	LL res = sum2[x][y];
	for(LL i = las;i < tot;i++){
		if(cx1[i] <= x  && x <= cx2[i] && cy1[i] <= y && y <= cy2[i]){
			res += cv[i];
		}
	}
	return res;
}

int main(){
	//freopen("2.in","r",stdin);
	//freopen("2.out","w",stdout);
	scanf("%lld%lld%lld",&n,&m,&q);
	s = (LL)sqrt(n*m);
	for(LL i = 1;i <= n;i++){
		for(LL j = 1;j <= m;j++){
			scanf("%lld",&arr[i][j]);
		}
	}
	for(LL i = 1;i <= n;i++){
		for(LL j = 1;j <= m;j++){
			brr[i][j] = arr[i][j] + arr[i-1][j-1] - arr[i][j-1] - arr[i-1][j];
		}
	}
	for(LL i = 1;i <= n;i++){
		for(LL j = 1;j <= m;j++){
			sum2[i][j] = sum2[i][j-1] + sum2[i-1][j] - sum2[i-1][j-1] + brr[i][j];
		}
	} 
	while(q--){
		LL opt;scanf("%lld",&opt);
		if(opt == 2){
			//查询  
			LL a,b;
			scanf("%lld%lld",&a,&b);
			printf("%lld\n",sums(a,b));
			sums(a,b);
		}
		else if(opt == 1){
			LL a,b,c,d,z;
			scanf("%lld%lld%lld%lld%lld",&a,&b,&c,&d,&z);
			cg(a,b,c,d,z);
		}
	}
	return 0;

} 
/*
2 2 4
1 1 
1 1
1
1 1 1 1 10
1
2 2 2 2 10
2
1 1 
2
2 2
*/