自用模板,原理网上一大堆,不过建议自己跟着题目敲几遍,有部分所谓题解代码是错的。

<树状数组区间更新与查询>

#include<bits/stdc++.h>
using namespace std;
const int maxn=50005;
int n,t,s;	//n数组长度,t区间修改次数,s查询次数
int a[maxn]= {0};
int sum1[maxn]= {0};
int sum2[maxn]= {0};

int lowbit(int x) {
	return x&(-x);
}

int update(int i,int k) {
	int rec=i;
	while(i<=n) {
		sum1[i]+=k;
		sum2[i]+=k*rec;
		i+=lowbit(i);
	}
}

int getsum(int i) {
	int res=0,cnt=i;
	while(i>0) {
		res+=(cnt+1)*sum1[i]-sum2[i];
		i-=lowbit(i);
	}
	return res;
}

int main() {
	cin>>n;
	for(int i=1; i<=n; i++) {
		cin>>a[i];
		update(i,a[i]-a[i-1]);
	}
	cin>>t;
	for(int i=1; i<=t; i++) {
		int l,r,k;	//[l,r]区间 ,k为修改的价值
		cin>>l>>r>>k;
		update(l,k);
		update(r+1,-k);
	}
	/*for(int i=0;i<=n+1;i++){
		printf("%d\t%d\t%d\t%d\n",a[i],sum1[i],sum2[i],getsum(i));
	}*/
	cin>>s;
	for(int i=1; i<=s; i++) {
		int l,r;
		cin>>l>>r;
		cout<<getsum(r)-getsum(l-1)<<endl;	
		//如果查询较多的话可以这个结果也用一个数组存起来 
	}
}


点赞(2)
 

0.0分

0 人评分

C语言网提供由在职研发工程师或ACM蓝桥杯竞赛优秀选手录制的视频教程,并配有习题和答疑,点击了解:

一点编程也不会写的:零基础C语言学练课程

解决困扰你多年的C语言疑难杂症特性的C语言进阶课程

从零到写出一个爬虫的Python编程课程

只会语法写不出代码?手把手带你写100个编程真题的编程百练课程

信息学奥赛或C++选手的 必学C++课程

蓝桥杯ACM、信息学奥赛的必学课程:算法竞赛课入门课程

手把手讲解近五年真题的蓝桥杯辅导课程

评论列表 共有 0 条评论

暂无评论