题目
试题编号 | 洛谷3372 |
---|---|
试题名称 | 求和线段树 |
时间限制 | 1s |
内存限制 | 125MB |
问题描述 | # 【模板】线段树 1 |
题目描述
如题,已知一个数列,你需要进行下面两种操作:
- 将某区间每一个数加上 k k k。
- 求出某区间每一个数的和。
输入格式
第一行包含两个整数 n , m n, m n,m,分别表示该数列数字的个数和操作的总个数。
第二行包含 n n n 个用空格分隔的整数,其中第 i i i 个数字表示数列第 i i i 项的初始值。
接下来 m m m 行每行包含 3 3 3 或 4 4 4 个整数,表示一个操作,具体如下:
1 x y k
:将区间 [ x , y ] [x, y] [x,y] 内每个数加上 k k k。2 x y
:输出区间 [ x , y ] [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
%
30\%
30% 的数据:
n
≤
8
n \le 8
n≤8,
m
≤
10
m \le 10
m≤10。
对于
70
%
70\%
70% 的数据:
n
≤
10
3
n \le {10}^3
n≤103,
m
≤
10
4
m \le {10}^4
m≤104。
对于
100
%
100\%
100% 的数据:
1
≤
n
,
m
≤
10
5
1 \le n, m \le {10}^5
1≤n,m≤105。
保证任意时刻数列中所有元素的绝对值之和 ≤ 10 18 \le {10}^{18} ≤1018。
【样例解释】
解答:
- BIT求解:
D [ i ] = A [ i ] − A [ i − 1 ] D[i]=A[i]-A[i-1] D[i]=A[i]−A[i−1]
∑ i = 1 n A [ i ] = ∑ i = 1 n ∑ j = 1 i D [ j ] = D [ 1 ] + ( D [ 1 ] + D [ 2 ] ) + . . . + ( D [ 1 ] + D [ 2 ] + . . . + D [ n ] ) = n D [ 1 ] + ( n − 1 ) D [ 2 ] + . . . + 1 ∗ D [ n ] = n ∗ ∑ i = 1 n D [ i ] − ∑ i = 1 n ( i − 1 ) D [ i ] \begin{aligned} \sum_{i=1}^nA[i] &= \sum_{i=1}^n\sum_{j=1}^iD[j] \\&= D[1]+(D[1]+D[2])+...+(D[1]+D[2]+...+D[n])\\&=nD[1]+(n-1)D[2]+...+1*D[n]\\&=n*\sum_{i=1}^nD[i]-\sum_{i=1}^n(i-1)D[i] \end{aligned} i=1∑nA[i]=i=1∑nj=1∑iD[j]=D[1]+(D[1]+D[2])+...+(D[1]+D[2]+...+D[n])=nD[1]+(n−1)D[2]+...+1∗D[n]=n∗i=1∑nD[i]−i=1∑n(i−1)D[i]
所以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;
}
- 线段树求解
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;
}