Acwing.提高课.一个简单的整数问题2(c++题解)

hb_zhyu

已于 2024-08-30 13:50:46 修改

阅读量469

点赞数 19

文章标签： c++ 算法数据结构

于 2024-08-30 13:50:18 首次发布

本文链接：https://blog.csdn.net/hb_zhyu/article/details/141716235

版权

给定一个长度为 N 的数列 A，以及 M 条指令，每条指令可能是以下两种之一：

C l r d，表示把 A[l],A[l+1],…,A[r]都加上 d。
Q l r，表示询问数列中第 l∼r 个数的和。

对于每个询问，输出一个整数表示答案。

输入格式

第一行两个整数 N,M。

第二行 N 个整数 A[i]。

接下来 M 行表示 M 条指令，每条指令的格式如题目描述所示。

输出格式

对于每个询问，输出一个整数表示答案。

每个答案占一行。

数据范围

1≤N,M≤10^5,
|d|≤10000
|A[i]|≤10^9

输入样例：

10 5
1 2 3 4 5 6 7 8 9 10
Q 4
Q 1
Q 2
C 1 6 3
Q 2

输出样例：

_____________________________________________________________________________

本题有两种思路，一个是树状数组，另一个是线段树，本章是树状数组。

写作不易，点个赞呗!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

_____________________________________________________________________________

#include <bits/stdc++.h>
#define LL long long 
using namespace std;
const int N=1e5+5;
int n,m;
LL a[N],b[N];
LL tr1[N],tr2[N];
LL lowbit(int x){return x&(-x);}
void update(LL x,LL v){//x为当前c数组下标，v为修改后的值
    for(int i=x;i<=n;i+=lowbit(i))tr1[i]+=v;
    for(int i=x;i<=n;i+=lowbit(i))tr2[i]+=v*x;
}
LL sum(int x){
	LL ans=0,cnt=0;
	for(int i=x;i!=0;i-=lowbit(i))ans+=tr1[i];
	for(int i=x;i!=0;i-=lowbit(i))cnt+=tr2[i];
	return ans*(x+1)-cnt;
}
signed main(){
    cin>>n>>m;
	for(int i=1;i<=n;i++)cin>>a[i];
	for(int i=1;i<=n;i++)b[i]=a[i]-a[i-1],update(i,b[i]);
	char Q;
	int l,r,v; 
	for(int i=1;i<=m;i++){
		cin>>Q>>l>>r;
		if(Q=='C'){
			cin>>v;
			update(l,v),update(r+1,-v);
		}else{
			cout<<sum(r)-sum(l-1)<<"\n";
		}
	}
}