洛谷3372线段树模板1求和

题目

试题编号	洛谷3372
试题名称	求和线段树
时间限制	1s
内存限制	125MB
问题描述	# 【模板】线段树 1

题目描述

如题，已知一个数列，你需要进行下面两种操作：

1. 将某区间每一个数加上 $k$。
2. 求出某区间每一个数的和。

输入格式

第一行包含两个整数 $n,m$，分别表示该数列数字的个数和操作的总个数。

第二行包含 $n$ 个用空格分隔的整数，其中第 $i$ 个数字表示数列第 $i$ 项的初始值。

接下来 $m$ 行每行包含 $3$ 或 $4$ 个整数，表示一个操作，具体如下：

1. 1 x y k：将区间 $[x,y]$ 内每个数加上 $k$。
2. 2 x y：输出区间 $[x,y]$ 内每个数的和。

输出格式

输出包含若干行整数，即为所有操作 2 的结果。

样例 #1

样例输入 #1

5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4

样例输出 #1

11
8
20

提示

对于 $30\%$ 的数据：$n\le 8$，$m\le 10$。
对于 $70\%$ 的数据：$n\le {10}^3$，$m\le {10}^4$。
对于 $100\%$ 的数据：$1\le n,m\le {10}^5$。

保证任意时刻数列中所有元素的绝对值之和 $\le {10}^{18}$。

【样例解释】

解答：

1. BIT求解：
   $$D[i]=A[i]-A[i-1]$$
   $$\begin{array}{rl}\sum _{i=1}^{n}A[i]& =\sum _{i=1}^n\sum _{j=1}^{i}D[j]\\ & =D[1]+(D[1]+D[2])+...+(D[1]+D[2]+...+D[n])\\ & =nD[1]+(n-1)D[2]+...+1\ast D[n]\\ & =n\ast \sum _{i=1}^{n}D[i]-\sum _{i=1}^n(i-1)D[i]\end{array}$$
   所以BIT求解这种区间修改区间查询也非常高效！

AC代码

#include <bits/stdc++.h>
#define ll long long
#define in(a) a=qcin()
#define out(a) qcout(a)
#define rep(i,a,n) for(int i=a;i<n;i++)
int a[100001]={0};
ll sum1[100001]={0};//di
ll sum2[100001]={0};//(i-1)*di
inline int qcin(){
	int f=0,x=0;char c=getchar();
	for(;!isdigit(c);c=getchar()){ f |= c == '-';}
	for(;isdigit(c);c=getchar()){x=(x<<3)+(x<<1)+(c^48);}
	return f?-x:x;
}
inline void qcout(ll x){
	if(x<0)putchar('-'),x=-x;
	if(x>9)qcout(x/10);
	putchar(x%10+'0');
} 
int n,m,l,r,k,p; 
void update(int i,int x){
	int d=i;
	while(i<=n){
		sum1[i]+=x;
		sum2[i]+=(d-1)*x;
		i+=i&-i;
	}
}
ll sum(int i){
	ll res=0;
	int d=i;
	while(i>0){
		res+=sum1[i]*d-sum2[i];
		i-=i&-i;
	}
	return res;
}
int main(){
	in(n);in(m);
	rep(i,1,n+1){
		in(a[i]);
		update(i,a[i]-a[i-1]);
	}
	while(m--){
		in(p);in(l);in(r);
		if(p==1){
			in(k);
			update(l,k);
			update(r+1,-k);
		}
		else{
			out(sum(r)-sum(l-1));
			printf("\n");
		}
	}
	return 0;
} 

2. 线段树求解

AC代码(pushdown)

#include <bits/stdc++.h>
#define ll long long
#define rep(i,a,n) for(int i=a;i<n;i++)
#define ls(i) (i<<1)
#define rs(i) (i<<1|1)
int n,m,l,r,k,p; 
const int N = 100001;
int a[N];
inline int qcin(){
	int f=0,x=0;char c=getchar();
	for(;!isdigit(c);c=getchar()){ f |= c == '-';}
	for(;isdigit(c);c=getchar()){x=(x<<3)+(x<<1)+(c^48);}
	return f?-x:x;
}
inline void qcout(ll x){
	if(x<0)putchar('-'),x=-x;
	if(x>9)qcout(x/10);
	putchar(x%10+'0');
} 
struct node{
	ll sum;
	ll lz;
	int l;
	int r;
	node():lz(0),l(0),r(0),sum(0){}
}tree[N<<2];
inline void pushdown(int i){
	if(tree[i].lz){
		tree[ls(i)].lz+=tree[i].lz;
		tree[rs(i)].lz+=tree[i].lz;
		int mid=(tree[i].l+tree[i].r)>>1;
		tree[ls(i)].sum+=(mid-tree[ls(i)].l+1)*tree[i].lz;
		tree[rs(i)].sum+=(tree[rs(i)].r-mid)*tree[i].lz;
		tree[i].lz=0;
	}
}
inline void build(int i,int l,int r){
	tree[i].l=l;tree[i].r=r;
	if(l==r){tree[i].sum=a[l];return;}
	int mid=(l+r)>>1;
	build(ls(i),l,mid);
	build(rs(i),mid+1,r);
	tree[i].sum=tree[ls(i)].sum+tree[rs(i)].sum;
}
inline void update(int i,int L,int R,int k){
	if(tree[i].l>=L&&tree[i].r<=R){
		tree[i].sum+=k*(tree[i].r-tree[i].l+1);
		tree[i].lz+=k;
		return;
	}
	pushdown(i);
	if(tree[ls(i)].r>=L)update(ls(i),L,R,k);
	if(tree[rs(i)].l<=R)update(rs(i),L,R,k);
	tree[i].sum=tree[ls(i)].sum+tree[rs(i)].sum;
}
ll query(int i,int L,int R){
	if(tree[i].l>=L&&tree[i].r<=R)return tree[i].sum;
	if(tree[i].l>R||tree[i].r<L)return 0;
	pushdown(i);
	ll res = 0;
	if(tree[ls(i)].r>=L)res+=query(ls(i),L,R);
	if(tree[rs(i)].l<=R)res+=query(rs(i),L,R);
	return res;
}  
int main(){
	n=qcin();m=qcin();
	rep(i,1,n+1){
		a[i]=qcin();
	}
	build(1,1,n);
	while(m--){
		p=qcin();l=qcin();r=qcin();
		if(p==1){
			k=qcin();
			update(1,l,r,k);
		}
		else{
			qcout(query(1,l,r));
			printf("\n");
		}
	}
	return 0;
}