POJ 3468区间修改区间查询

最新推荐文章于 2022-03-20 12:39:28 发布

明月千里赴迢遥

最新推荐文章于 2022-03-20 12:39:28 发布

阅读量102

点赞数

分类专栏：数据结构 ACM 文章标签： c++ acm竞赛

本文链接：https://blog.csdn.net/qq_54886579/article/details/119495056

版权

ACM 同时被 2 个专栏收录

76 篇文章 0 订阅

订阅专栏

数据结构

30 篇文章 0 订阅

订阅专栏

这篇博客详细介绍了如何利用树状数组（也称作二进制指数矩阵）来解决区间修改和区间查询的问题。文章通过一个POJ 3468的竞赛题目为例，展示了在C++中实现树状数组的过程，包括初始化、区间更新和查询等操作。博主特别强调了在处理大整数和区间系数时的易错点，提醒读者注意类型转换和更新细节。

摘要由CSDN通过智能技术生成

这题不讲武德，说好的Ai<=1e9，在int范围内，结果初始值竟然需要用到long long不然过不了

不讲武德

//区间修改、区间查询	http://poj.org/problem?id=3468
#include<cstdio>
#include<iostream>
using namespace std;
const int N=1e6+10;
typedef long long ll;
ll A[N],C[N],D[N],tree1[N],tree2[N];//C[]为差分数组，我们用树状数组维护差分数组 A[]为原数组 D[N]为辅助数组i*C[i] ,tree[1]维护C,tree2维护D 
int n;//n为要用到的数组规模 
inline ll read(){
	ll s=0,w=1;
	char c=getchar();
	while(c<'0' || c>'9'){
		if(c=='-')	w*=-1;
		c=getchar();
	}	
	while(c>='0' && c<='9')	s=(s<<3)+(s<<1)+c-'0',c=getchar();//易错：不能写成s+= 
	return s*w;
}
inline int lowbit(int x){
	return x&-x;
}
void update(int x,ll delta){//ll delta是个最坑的易错点，这个违反了题意！ 
	int k=x;//易错点 
	while(x<=n){
		tree1[x]+=delta;
		tree2[x]+=k*delta;//易错点：更新时是k*delta,系数不能变 
		x+=lowbit(x);
	}
}
inline void init(){
	for(int i=1;i<=n;i++){
		A[i]=read();
		C[i]=A[i]-A[i-1];
		update(i,A[i]-A[i-1]);//不要写成A[i] 
	}
}
void updateRange(int l,int r,int delta){
	update(l,delta);
	update(r+1,-delta);
}
ll query(int x){/*前缀和S=(n + 1)*(C[1] + C[2]+...+C[n]) - (1*C[1]+2*C[2]+...+n*C[n])
						=(n+1)*ΣC[n]-ΣD[i] 
				注意，用tree1和tree2进行计算！ */
	ll s=0;
	int n=x+1;
	while(x>0){
		s+=n*tree1[x]-tree2[x];
		x-=lowbit(x); //易错点 忘写该句话 
	}
	return s;
}
inline ll queryRange(int l,int r){
	return (query(r)-query(l-1));
}
int main(){
	n=read();
	int Q=read();
	init();
	while(Q--){
		char a=getchar();
		if(a=='Q'){
			int x,y;
			x=read(),y=read();
			printf("%lld\n",queryRange(x,y));
		}else{
			int l=read(),r=read(),delta=read();
			updateRange(l,r,delta);
		}
	}
	return 0;
}