线段树
Posted _ZZH
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了线段树相关的知识，希望对你有一定的参考价值。
（大概长这样）
注：本人未经过系统的学习线段树，可能理解有误，欢迎读者指出，但我相信按照我下面的解释，你一定能知道怎么写基本的线段树题，至少个人感觉读懂下文并不难。
定义：
线段树是一种二叉搜索树，与区间树相似，它将一个区间划分成一些单元区间，每个单元区间对应线段树中的一个叶结点。对于线段树中的每一个非叶子节点[a,b]，它的左儿子表示的区间为[a,(a+b)/2]，右儿子表示的区间为[(a+b)/2+1,b]。因此线段树是平衡二叉树，最后的子节点数目为N，即整个线段区间的长度。
用途：
线段树支持：插入，修改（分为单点修改和区间修改)，询问（可以作ST表来查询区间最值，只是慢了点。也可以用作询问区间和。其他骚操作，以后再讲）
实现（发现“r n”和“m”长得很像，读者注意区分）:
从上面的图中可以看出，每个节点都是一条【l,r】的线段，每棵子树有一个根节点（废话），所以关于每个节点rn以及表示的线段【l，r】我们记录m=(l+r)/2，也就是线段的中点，那么他的左儿子（lson）应为节点rn*2,来表示【,l,m】的线段，右儿子（rson）应为节点rn*2+1，表示【m+1，r】的线段。这样我们就可以通过这种方式遍历下去，来找到我们的答案。
1.建树：
void update(ll rn)// “ll”就是“long long”啦，题目需要，也忘了改了，请见谅qwq { sum[rn]=sum[rn<<1]+sum[rn<<1|1];//位运算符“<<1”相当于“*2”，“|1”相当于“+1”，“>>1”相当于“/2”； } void build(ll l,ll r,ll rn)//参数表示区间【l，r】和根节点rn { if(l==r) { sum[rn]=z[l];//我们不妨看一下图，当l=r时节点rn表示的是一个点l，所以直接赋上点值就好（至于数组名为什么叫z，忘了） return; } int m=(l+r)>>1; build(lson);//lson和rson的表示就是以上描述中提到的，下面完整代码中有它们的定义 build(rson); update(rn); //节点rn是lson和rson的父节点，恰好完全包含lson，rson表示的区间，所以sum[rn]应是两个儿子的“sum”之和，一直递归下去，树就建好了…… }
2.区间询问：细心地读者应该能从图中发现这一点：如果我们询问区间【6，10】自然很简单，因为这就是sum【3】存的东西，那么如果是【4，10】呢？【5，9】呢？
这才是难点。解决方法：我们可以将这一区间拆成几个已知的小区间，再统计各个小区间的sum。
ll find(ll l,ll r,ll rn,ll nowl,ll nowr)//不要被参数名迷惑qwq，l，r，rn都是当前找到的节点的元素，你要询问的是nowl到nowr的什么什么东西（这里当然就是区间和了……） { ll ans=0; if(nowl<=l&&nowr>=r)//这句话说明当前区间已被你要找的区间完全包含，是它的一部分，那么我们就不用往下找了，理由见上面蓝色文字描述。 { return sum[rn];//找到一块小区间，返回 } int m=(l+r)>>1; if(nowl<=m)ans+=find(lson,nowl,nowr);//询问区间跨过了m说明在左儿子部分有一块区间 if(nowr>m)ans+=find(rson,nowl,nowr);//右端点跨过m+1说明在右儿子部分也有一块，那就递归下去逐块解决 return ans;//不要忘了返回累加答案（也许只有我会忘……） }
3.单点修改: 我们更改一个数字后，会牵动所有包含它的区间，那么怎么处理呢？
void _change(int l,int r,int rn,int p,int v) { //我们在【l，r】里面找到p使这个数加上v（变量名好随意的说）
if(l==r)//可以先看看下面的程序吧，当l=r时rn刚好锁定p，想一想为什么？
{
sum[rn]+=v;
return ;
}
int m=(l+r)>>1;
if(p<=m)_change(lson,p,v);//位置在左边就找左子树，不然找右子树，类似二分查找
if(p>m)_change(rson,p,v);
update(rn);//这就是来解决蓝字描述上的问题了
}
4.区间修改：更改区间【l，r】的所有数字，我们可以发现单点修改其实就是修改区间【l,l】所以为什么不早拿出来？这里就不太好理解了qwq
每一个数字都要更改，那么一个区间要改几次？如果一个区间包含的许多小区间都变了，它要改几次？
我们肯定是希望它只更改一次了，那么我们可以给每个节点一个数组col（color的意思，名字是瞎取的），学名“懒标记”。
因为区间修改是区间的所有数字按照统一规则变化，所以我们在所有访问过这个点的时候打上标记，下次询问是再加上这个值。
void put_col(ll rn,ll m,ll l,ll r)//参数为【l,r】，中点m，根节点rn（顺序好奇怪，瞎写的），参数可以只有l r rn，m可以算出来 { if(col[rn]) { col[rn<<1]+=col[rn]; col[rn<<1|1]+=col[rn];//这是将标记传递给下一代，因为可能更改多次再询问，所以标记是累加关系 sum[rn<<1]+=(m-l+1)*col[rn]; sum[rn<<1|1]+=(r-m)*col[rn];//显而易见的公式，每个元素都有一个相同的col，这个区间就会增大长度*col col[rn]=0;//注意清零，不然再查询就加多了…… } } void change(ll l,ll r,ll rn,ll v,ll nowl,ll nowr)//区间修改 { if(nowl<=l&&nowr>=r)//包含关系 { col[rn]+=v;//叠标记 sum[rn]+=v*(r-l+1); return; } int m=(l+r)>>1; put_col(rn,m,l,r);//向下传递标记 if(nowl<=m)change(lson,v,nowl,nowr); if(nowr>m)change(rson,v,nowl,nowr); update(rn);//这很重要，下面的值修改完了，递归到上一层 }
还有，在“find”中加入一句话：
ll find(ll l,ll r,ll rn,ll nowl,ll nowr)
{
ll ans=0;
if(nowl<=l&&nowr>=r)
{
return sum[rn];
}
int m=(l+r)>>1;
put_col(rn,m,l,r);//为啥叫“懒”标记，因为它在修改时只是打上标记，欠了一屁股债，等到“find”再一并还清
if(nowl<=m)ans+=find(lson,nowl,nowr);
if(nowr>m)ans+=find(rson,nowl,nowr);
return ans;
}
接下来具体写一下代码：（https://www.luogu.org/problemnew/show/P3372）
#include<iostream> #include<cstdio> #define lson l,m,rn<<1 #define rson m+1,r,rn<<1|1 #define ll long long using namespace std; ll z[2000010],sum[4000010],n,M,question[2000010],total,col[2000010]; void update(ll rn) { sum[rn]=sum[rn<<1]+sum[rn<<1|1]; } void build(ll l,ll r,ll rn) { if(l==r) { sum[rn]=z[l]; return; } int m=(l+r)>>1; build(lson); build(rson); update(rn); } void put_col(ll rn,ll m,ll l,ll r) { if(col[rn]) { col[rn<<1]+=col[rn]; col[rn<<1|1]+=col[rn]; sum[rn<<1]+=(m-l+1)*col[rn]; sum[rn<<1|1]+=(r-m)*col[rn]; col[rn]=0; } } ll find(ll l,ll r,ll rn,ll nowl,ll nowr) { ll ans=0; if(nowl<=l&&nowr>=r) { return sum[rn]; } int m=(l+r)>>1; put_col(rn,m,l,r); if(nowl<=m)ans+=find(lson,nowl,nowr); if(nowr>m)ans+=find(rson,nowl,nowr); return ans; } void change(ll l,ll r,ll rn,ll v,ll nowl,ll nowr) { if(nowl<=l&&nowr>=r) { col[rn]+=v; sum[rn]+=v*(r-l+1); return; } int m=(l+r)>>1; put_col(rn,m,l,r); if(nowl<=m)change(lson,v,nowl,nowr); if(nowr>m)change(rson,v,nowl,nowr); update(rn); } void _change(int l,int r,int rn,int p,int v) { if(l==r) { sum[rn]+=v; return ; } int m=(l+r)>>1; if(p<=m)_change(lson,p,v); if(p>m)_change(rson,p,v); update(rn); } int main() { scanf("%d%d",&n,&M); for(int i=1;i<=n;i++) cin>>z[i]; build(1,n,1); for(int i=1;i<=M;i++) { ll x,y,z; cin>>x; if(x==1) { cin>>x>>y>>z; change(1,n,1,z,x,y); }else { cin>>y>>z; question[++total]=find(1,n,1,y,z); } } for(int i=1;i<=total;i++)printf("%lld\\n",question[i]); }//大家可以结合题意和上面的讲解自己理解一下，以下讲几个例题。
例题：
1.统计和（https://www.luogu.org/problemnew/show/P2068#sub）
这就是一道典型的单点修改和区间询问的裸题，没什么难度，直接放代码了：
#include<iostream>
#include<cstdio>
#define lson l,mid,rn<<1
#define rson mid+1,r,rn<<1|1
using namespace std;
int n,m,sum[100010],ans[100010],point;
void update(int rn)
{
sum[rn]=sum[rn<<1]+sum[rn<<1|1];
}
void change(int l,int r,int rn,int q,int v)
{
if(l==r)
{
sum[rn]+=v;
return;
}
int mid=(l+r)>>1;
if(q<=mid)change(lson,q,v);
if(q>mid)change(rson,q,v);
update(rn);
}
int find(int l,int r,int rn,int nowl,int nowr)
{
if(nowl<=l&&nowr>=r)
return sum[rn];
int ans=0;
int mid=(l+r)>>1;
if(nowl<=mid)ans+=find(lson,nowl,nowr);
if(nowr>mid)ans+=find(rson,nowl,nowr);
return ans;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++)
{
char a;
cin>>a;
if(a==\'x\')
{
int q,b;
scanf("%d%d",&q,&b);
change(1,n,1,q,b);
}
else
{
int x,y;
scanf("%d%d",&x,&y);
ans[++point]=find(1,n,1,x,y);
}
}
for(int i=1;i<=point;i++)
printf("%d\\n",ans[i]);
}
2.在你窗外闪耀的星星(https://www.luogu.org/problemnew/show/P3353）//个人感觉题目描述还是挺不错的，说不定哪位兄弟有用呢（逃）
此题说白了就是区间和（不过这里我们来练习一下线段树的区间查询），我们枚举每个起点，如果窗户在这里，区间和是多少。然后取一个最大值，即为所求答案。
不过注意一下这个题目有一个坑点：
这个不合逻辑的东西是没法跑的（我试过……），所以特判输出0就好
#include<iostream>
#include<cstdio>
#define lson l,mid,rn<<1
#define rson mid+1,r,rn<<1|1
using namespace std;
int n,m,ans1,sum[1000010],a[1000010];
void update(int rn)
{
sum[rn]=sum[rn<<1]+sum[rn<<1|1];
}
void change(int l,int r,int rn,int x,int y)
{
if(l==r)
{
sum[rn]+=y;
return;
}
int mid=(l+r)>>1;
if(x<=mid)change(lson,x,y);
if(x>mid)change(rson,x,y);
update(rn);
}
int find(int l,int r,int rn,int nowl,int nowr)
{
if(nowl<=l&&nowr>=r)
return sum[rn];
int mid=(l+r)>>1;
int ans=0;
if(nowl<=mid)ans+=find(lson,nowl,nowr);
if(nowr>mid)ans+=find(rson,nowl,nowr);
return ans;
}
int main()
{
scanf("%d%d",&n,&m);
if(m==0)
{
printf("0");
return 0;
}
for(int i=1;i<=n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
change(1,n,1,x,y);
}
for(int i=1;i<=n;i++)
if(i+m-1<=n)
ans1=max(ans1,find(1,n,1,i,i+m-1));
else
break;
printf("%d",ans1);
}
3.I Hate It(https://www.luogu.org/problemnew/show/P1531)//这老师什么操作，校内黑幕？
这题考察单点修改和区间查询，不同的是，这次不是区间和，而是区间最值，其实就改两行就行了，题不难，请读者自行理解。
#include<iostream>
#include<cstdio>
#define lson l,mid,rn<<1
#define rson mid+1,r,rn<<1|1
using namespace std;
int n,m,sum[1000010],a[1000010],ans1[200010],point;
void update(int rn)
{
sum[rn]=max(sum[rn<<1],sum[rn<<1|1]);
}
void build(int l,int r,int rn)
{
if(l==r)
{
sum[rn]=a[l];
return;
}
int mid=(l+r)>>1;
build(lson);
build(rson);
update(rn);
}
void change(int l,int r,int rn,int x,int y)
{
if(l==r)
{
sum[rn]=max(y,sum[rn]);
return;
}
int mid=(l+r)>>1;
if(x<=mid)change(lson,x,y);
if(x>mid)change(rson,x,y);
update(rn);
}
int find(int l,int r,int rn,int nowl,int nowr)
{
if(nowl<=l&&nowr>=r)
return sum[rn];
int mid=(l+r)>>1;
int ans=0;
if(nowl<=mid)ans=max(find(lson,nowl,nowr),ans);
if(nowr>mid)ans=max(find(rson,nowl,nowr),ans);
return ans;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
build(1,n,1);
for(int i=1;i<=m;i++)
{
char now;
cin>>now;
if(now==\'Q\')
{
int x,y;
scanf("%d%d",&x,&y);
ans1[++point]=find(1,n,1,x,y);
}else
{
int x,y;
scanf("%d%d",&x,&y);
change(1,n,1,x,y);
}
}
for(int i=1;i<=point;i++)printf("%d\\n",ans1[i]);
}
4.滑动窗口(https://www.luogu.org/problemnew/show/P1886)//貌似正解并不是线段树，但是我们还是拿来练一练吧。
比起上一道题我们只需要多存一个最小值就好，区间长度一定，只是起点在变，也是和上一题有许多相似点的
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<queue>
using namespace std;
#define lson l,mid,rn<<1
#define rson mid+1,r,rn<<1|1
int n,k,maxx[2000010],minn[2000010],a[2000010],ans1[2000010],ans2[2000010],point;
void update(int rn) {
maxx[rn]=max(maxx[rn<<1],maxx[rn<<1|1]);
minn[rn]=min(minn[rn<<1],minn[rn<<1|1]);
}
void build(int l,int r,int rn) {
if(l==r)
{
minn[rn]=maxx[rn]=a[l];
return;
}
int mid=(l+r)>>1; build(lson); build(rson); update(rn); } int find_max(int l,int r,int rn,int nowl,int nowr) {
if(nowl<=l&&nowr>=r)
return maxx[rn];
int mid=(l+r)>>1;
int ans=0;
if(nowl<=mid)ans=max(ans,find_max(lson,nowl,nowr));
if(nowr>mid)ans=max(ans,find_max(rson,nowl,nowr));
return ans;
}
int find_min(int l,int r,int rn,int nowl,int nowr) {
if(nowl<=l&&nowr>=r)
return minn[rn];
int mid=(l+r)>>1;
int ans=0x7fffffff;
if(nowl<=mid)ans=min(ans,find_min(lson,nowl,nowr));
if(nowr>mid)ans=min(ans,find_min(rson,nowl,nowr));
return ans;
}
int main() {
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
build(1,n,1);
for(int i=1;i<=n;i++)
{
int j=i+k-1;
if(j>n)break;
ans1[++point]=find_max(1,n,1,i,j);
ans2[point]=find_min(1,n,1,i,j);
}
for(int i=1;i<=point;i++)
printf("%d ",ans2[i]);
printf("\\n");
for(int i=1;i<=point;i++)
printf("%d ",ans1[i]);
}
希望大家没有直接拿去交（还没交的同学可以找一下BUG）
好吧，我说了，对于每个区间，都需要查找两遍，max和min，所以一定会TLE,而且数组也开小了（为避免越界通常要开4n大小的数组）
那么能不能在找到max的同时找到min呢？
答案是肯定的，用全局变量记录就好了……
见代码：
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<queue>
using namespace std;
#define lson l,mid,rn<<1
#define rson mid+1,r,rn<<1|1
int n,k,maxx[5000010],minn[5000010],a[5000010],ans1[5000010],ans2[5000010],point,find1;
void update(int rn)
{
maxx[rn]=max(maxx[rn<<1],maxx[rn<<1|1]);
minn[rn]=min(minn[rn<<1],minn[rn<<1|1]);
}
void build(int l,int r,int rn)
{
if(l==r)
{
minn[rn]=maxx[rn]=a[l];
return;
}
int mid=(l+r)>>1;
build(lson);
build(rson);
update(rn);
}
int find_max(int l,int r,int rn,int nowl,int nowr)
{
if(nowl<=l&&nowr>=r)
{
find1=min(find1,minn[rn]);
return maxx[rn];
}
int mid=(l+r)>>1;
int ans=-0x7fffffff;
if(nowl<=mid)ans=max(ans,find_max(lson,nowl,nowr));
if(nowr>mid)ans=max(ans,find_max(rson,nowl,nowr));
return ans;
}
int main()
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
build(1,n,1);
for(int i=1;i<=n;i++)
{
int j=i+k-1;
if(j>n)break;
find1=0x7fffffff;
ans1[++point]=find_max(1,n,1,i,j);
ans2[point]=find1;
}
for(int i=1;i<=point;i++)
printf("%d ",ans2[i]);
printf("\\n");
for(int i=1;i<=point;i++)
printf("%d ",ans1[i]);
}
5.gcd区间（https://www.luogu.org/problemnew/show/P1890）
我们记录每个区间的gcd，完了。
#include<iostream>
#include<cstdio>
#include<algorithm>
#define lson l,mid,rn<<1
#define rson mid+1,r,rn<<1|1
using namespace std;
int n,m,a[1010],gcd[4013];
int _gcd(int a,int b)
{
if(b==0)return a;
return _gcd(b,a%b);
}
void update(int rn)
{
gcd[rn]=_gcd(gcd[rn<<1],gcd[rn<<1|1]);
}
void build(int l,int r,int rn)
{
if(l==r)
{
gcd[rn]=a[l];
return;
}
int mid=(l+r)>>1;
build(lson);
build(rson);
update(rn);
}
int find(int l,int r,int rn,int nowl,int nowr)
{
int ans=0;
if(nowl<=l&&nowr>=r)return gcd[rn];
int mid=(l+r)>>1;
if(nowl<=mid)ans=find(lson,nowl,nowr);
if(nowr>mid)ans=_gcd(ans,find(rson,nowl,nowr));
return ans;
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
build(1,n,1);
for(int i=1;i<=m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
printf("%d\\n",find(1,n,1,x,y));
}
}
相信通过上面的描述，大家应该都会写基本的线段树了，篇幅所限，例题不再列举，熟能生巧，大家多加练习吧！
完结，撒花。
注：本人致力于用通俗的语言来解析算法和数据结构，第一次写博客，多多包涵！
以上是关于线段树的主要内容，如果未能解决你的问题，请参考以下文章