矩阵树状数组(矩阵加矩阵求和)

将矩阵分为很多由lowbit 组成的小矩阵 , 然后就跟树状数组一样维护了

求和的时候用矩阵前缀和的思想(s[x2][y2]-s[x1-1][y2]-s[x2][y1-1]+s[x1-1][y1-1]) 

单点修改时将(x1,y1)加 , (x2+1,y2) 减 , (x1,y2+1) 减 , (x2+1,y2+1) 加

复杂度是 O((logn)*(logm)) 


我们考虑如何矩阵加

我们考虑差分数组 d[x][y]=d[x-1][y]+d[x][y-1]-d[x-1][y-1]+a[x][y]

 现在放一个矩阵修改 , 矩阵求和的代码

#include<bits/stdc++.h>
#define N 2050
#define LL long long
using namespace std;
LL t1[N][N],t2[N][N],t3[N][N],t4[N][N];
int n,m;
void Up(int x,int y,int val){
	for(int i=x;i<=n;i+=i&-i)
		for(int j=y;j<=m;j+=j&-j){ 
			t1[i][j] += (LL)val;
			t2[i][j] += (LL)val * x;
			t3[i][j] += (LL)val * y;
			t4[i][j] += (LL)val * x * y;
		}
}
LL Q(int x,int y){
	LL ans = 0;
	for(int i=x;i;i-=i&-i)
		for(int j=y;j;j-=j&-j){
			ans += (LL)t1[i][j] * (x+1) * (y+1);
			ans -= (LL)t2[i][j] * (y+1);
			ans -= (LL)t3[i][j] * (x+1);
			ans += (LL)t4[i][j];
		}
	return ans;
}
int main(){
	scanf("%d%d",&n,&m); int op,x1,y1,x2,y2;
	while(~scanf("%d%d%d%d%d",&op,&x1,&y1,&x2,&y2)){
		if(op==1){
			int val; scanf("%d",&val);
			Up(x1,y1,val); Up(x1,y2+1,-val);
			Up(x2+1,y1,-val); Up(x2+1,y2+1,val);
		}
		if(op==2){
			LL ans = Q(x2,y2) - Q(x2,y1-1) - Q(x1-1,y2) + Q(x1-1,y1-1);
			printf("%lld",ans); 
		}
	} return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

FSYo

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值