一、简单的线段树(区间最值查询,单点修改)
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#define N 100002
using namespace std;
int a[N];
struct Node{
int l,r;
int data;
}t[N*4];//注意将数组至少开至最大n的4倍,1+2+4+......2^n+……
void bulid(int p,int l,int r){
t[p].l=l,t[p].r=r;
if(l==r){
t[p].data=a[l];
return ;
}
int mid=(l+r)>>1;
bulid(p*2,l,mid);
bulid(p*2+1,mid+1,r);
t[p].data=max(t[p*2].data,t[p*2+1].data);
}
int query(int p,int l,int r){
if(l<=t[p].l&&r>=t[p].r){
return t[p].data;
}
int mid=(t[p].l+t[p].r)>>1;
int val=-(1<<30);
if(mid>=l) val=max(val,query(p*2,l,r));
if(mid<r) val=max(val,query(p*2+1,l,r));
return val;
}
void change(int p,int x,int val){
if(t[p].l==t[p].r) {
t[p].data=val;
return;
}
int mid=(t[p].l+t[p].r)>>1;
if(x<=mid) change(p*2,x,val);
else change(p*2+1,x,val);
t[p].data=max(t[p<<1].data,t[p<<1+1].data);
}
int main()
{
int n;
cin>>n;
for(int i=1;i<=n;i++) cin>>a[i];
bulid(1,1,n);
int tmp,m,x,val,l,r;
for(int i=1;i<=m;i++){
cin>>tmp;
if(tmp==1){
cin>>x>>val;
change(1,x,val);
}
else {
cin>>l>>r;
cout<<query(1,l,r)<<endl;
}
}
}
二、线段树+延迟操作
功能:
1、区间同时加上某一个数
2、查询区间和
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100001
#define ll long long
using namespace std;
struct Node
{
int l,r;
ll sum,add;//add是sum的延迟操作变量
}t[N*4];
int a[N];
void bulid(int p,int l,int r){
t[p].l=l,t[p].r=r;
if(l==r){
t[p].sum=a[l];
return ;
}
int mid=(l+r)>>1//分成两个部分;
bulid(p*2,l,mid);
bulid(p*2+1,mid+1,r);
t[p].sum=t[p*2].sum+t[p*2+1].sum;
}
void spread(int p){
if(t[p].add){//将延迟操作向下传递,注意本身设置延迟时,sum没有延迟
t[p*2].sum +=t[p].add*(t[2*p].r-t[2*p].l+1);
t[p*2+1].sum+=t[p].add*(t[2*p+1].r-t[2*p+1].l+1);
t[2*p+1].add+=t[p].add;
t[2*p].add+=t[p].add;
t[p].add=0;//清除延迟
}
}
void change(int p,int l,int r,int d){
if(l<=t[p].l&&t[p].r<=r){
t[p].sum+=(ll)d*(t[p].r-t[p].l+1);
t[p].add+=d;//当可包含当前区间被修改区间包含时,时无需操作该区间
return;
}
spread(p);//判断是否有延迟,有则下传
int mid=(t[p].l+t[p].r)/2;
if(l<=mid) change(p*2,l,r,d);
if(r>mid) change(p*2+1,l,r,d);
t[p].sum=t[p*2].sum+t[p*2+1].sum;
}
ll ask(int p,int l,int r){
if(l<=t[p].l&&t[p].r<=r){
return t[p].sum;
}
spread(p);//如果需要用到的区间存在延迟,下传
int mid=(t[p].l+t[p].r)>>1;
ll val=0;
if(l<=mid) val+=ask(p*2,l,r);
if(r>mid) val+=ask(p*2+1,l,r);
return val;
}
int main()
{
int n,m,tmp,l,r,x,val;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",a+i);
bulid(1,1,n);
for(int i=1;i<=m;i++){
scanf("%d",&tmp);
if(tmp==2){
scanf("%d%d",&l,&r);
printf("%lld\n",ask(1,l,r));
}
if(tmp==1){
scanf("%d%d%d",&l,&r,&val);
change(1,l,r,val);
}
}
}