线段树最基本的所有操作模板

给定一个长度为N的数列A,以及M条指令,每条指令可能是以下两种之一:
1、“C l r d”,表示把 A[l],A[l+1],…,A[r] 都加上 d。
2、“Q l r”,表示询问 数列中第 l~r 个数的和。
对于每个询问,输出一个整数表示答案。
输入格式
第一行两个整数N,M。
第二行N个整数A[i]。
接下来M行表示M条指令,每条指令的格式如题目描述所示。
输出格式
对于每个询问,输出一个整数表示答案。
每个答案占一行。
数据范围
1≤N,M≤1051≤N,M≤105,

|d|≤10000|d|≤10000,

|A[i]|≤1000000000|A[i]|≤1000000000
输入样例:
10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4

输出样例:
4
55
9
15
这道题没啥好说的,就是线段树最基本的一道题,只要在理解的基础上背下模板即可
ac代码如下:

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long LL;
const int N=100010;
int n,m;
int w[N];
struct Node{
 int l,r;
 LL sum,add;
}tr[N*4];
void pushup(int u){
 tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}
void pushdown(int u){
 Node &root=tr[u],&left=tr[u<<1],&right=tr[u<<1|1];
 if(root.add){
  left.add+=root.add,left.sum+=(LL)(left.r-left.l+1)*root.add;
  right.add+=root.add,right.sum+=(LL)(right.r-right.l+1)*root.add;
  root.add=0;
 }
}
void build(int u,int l,int r){
 if(l==r)   tr[u]={l,r,w[r],0};
 else{
  tr[u]={l,r};
  int mid=l+r>>1;
  build(u<<1,l,mid),build(u<<1 |1 ,mid+1,r);
  pushup(u);
 }
}
void modify(int u,int l,int r,int d){
 if(tr[u].l>=l && tr[u].r<=r){
  tr[u].sum+=(LL)(tr[u].r-tr[u].l+1)*d;
  tr[u].add+=d;
 }
 else{
  pushdown(u);
  int mid=tr[u].l+tr[u].r>>1;
  if(l<=mid)    modify(u<<1,l,r,d);
  if(r>mid)     modify(u<<1|1,l,r,d);
  pushup(u);
 }
}
LL query(int u,int l,int r){
 if(tr[u].l>=l&& tr[u].r<=r)   return tr[u].sum;
 pushdown(u);
 int mid=tr[u].l+tr[u].r>>1;
 LL sum=0;
 if(l<=mid)   sum=query(u<<1,l,r);
    if(r>mid)    sum+=query(u<<1|1,l,r);
 return sum; 
}
int main(){
 scanf("%d%d",&n,&m);
 for(int i=1;i<=n;i++)   scanf("%d",&w[i]);
 build(1,1,n);
 char op[2];
 int l,r,d;
 while(m--){
  scanf("%s%d%d",op,&l,&r);
  if(*op=='C'){
   scanf("%d",&d);
   modify(1,l,r,d);
  }
  else printf("%lld\n",query(1,l,r));
 }
 return 0;
}
发布了29 篇原创文章 · 获赞 24 · 访问量 3667

猜你喜欢

转载自blog.csdn.net/qq_45772483/article/details/104322139