高级数据结构—线段树（一）

学线段树的原因是因为cf的一道题目始终想不出来怎么优化，后来知道区间查询和修改要用到线段树。。。

原题：Iva & Pav

线段树的作用

1. 区间最值查询：可以高效地找到给定区间内的最大值、最小值等。

2. 区间和查询：可以高效地计算给定区间内元素的和、积等。

3. 区间更新：可以高效地对给定区间内的元素进行更新操作，如增加一个固定值、赋值等。

4. 区间覆盖：可以将给定区间内的元素全部赋值为一个固定值。

5. 区间合并：可以将多个区间合并成一个区间，快速地进行区间合并操作。

6. 区间离散化：可以将区间内的元素进行离散化处理，方便进行查询和统计操作。

7. 区间交集：可以快速地找到多个区间之间的交集

线段树和树状数组的区别 

 刚学完树状数组来学线段树，一开始还不知道他们具体的差别在哪里，那么以下是我的理解。

1.树状数组是前缀和优化，要用到前缀和的时候较为方便。

2.树状数组用来进行单点修改，区间查询；或者区间修改，单点查询较为方便，而区间查询和区间修改较为复杂，因此可以用线段树优化。

3.线段树适用于需要频繁的区间查询和更新操作的问题，如区间最值、区间和等，能够灵活处理各种区间操作。

4.树状数组适用于一维数组的前缀和查询和更新操作，对于简单的区间操作也能够提供高效的解决方案。

例题： 

最大数

题目链接：最大数

直接看代码：

#include <bits/stdc++.h>
#define int long long
using namespace std;
//单点插入，区间查询
const int N = 2e5+5;
struct node{
	int l,r;
	int v;
}tr[N*4];

int m,p;

//子节点的信息更新父节点
void pushup(int u){
	tr[u].v=max(tr[u<<1].v,tr[u<<1|1].v);
}

//u为当前线段树节点编号
void build(int u,int l,int r){
	tr[u]={l,r};
	if(l==r)return;
	int mid=l+r>>1;
	build(u<<1,l,mid);
	build(u<<1|1,mid+1,r);
}

//查询以u为根节点，区间[l,r]中的最大值
int query(int u, int l, int r) {
    //      Tl-----Tr
    //   L-------------R   
    //1.不必分治，直接返回
    if(tr[u].l >= l && tr[u].r <= r) return tr[u].v;

    int mid = tr[u].l + tr[u].r >> 1;
    int v = 0;
    //     Tl----m----Tr
    //        L-------------R 
    //2.需要在tr的左区间[Tl, m]继续分治
    if(l <= mid) v = query(u << 1, l, r);

    //     Tl----m----Tr
    //   L---------R 
    //3.需要在tr的右区间(m, Tr]继续分治
    if(r > mid) v = max(v, query(u << 1 | 1, l, r));

    //     Tl----m----Tr
    //        L-----R 
    //2.3涵盖了这种情况
    return v;
}

//u为节点编号，x为修改位置，v为修改的值
void modify(int u,int x,int v){
	if(tr[u].l==tr[u].r)tr[u].v=v;//叶子节点，递归出口
	else{
		int mid=tr[u].l+tr[u].r>>1;
		//分治，修改位置偏左往左边遍历，偏右往右边遍历
		if(x<=mid)modify(u<<1,x,v);
		else {
			modify(u<<1|1,x,v);
		}
		pushup(u);//回溯，子节点的信息更新父节点
	}
}

signed main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	//n表示树中节点个数，last表示上一次查询的结果
	int n=0,last=0;
	cin>>m>>p;

	//初始化线段树，节点的区间最多为[1,m]
	build(1,1,m);

	while(m--){
		char op;cin>>op;
		if(op=='A'){//添加节点
			int t;cin>>t;
			//在n+1处插入
			modify(1,n+1,(t+last)%p);
			//节点个数+1
			n++;
		}
		else {
			int l;cin>>l;
			//查询[n - L + 1, n]内的最大值，u = 1，即从根节点开始查询
			last=query(1,n-l+1,n);
			cout<<last<<"\n";
		}
	}

	return 0;
}

你能回答这些问题吗

题目链接：你能回答这些问题吗

如图：

如图，假设我们要求区间的最大子段和，有三种情况：

1.包含所有左半边，部分右半边----->左半边的区间和+右半边的前缀和

2.包含所有右半边，部分左半边----->右半边的区间和+左半边的后缀和

3.中间的一部分----->左半边的后缀和+右半边的前缀和

因此我们的结构体要记录四个信息：

struct node{
	int l,r;
	int sum;//[l,r]的区间和
	int lmax;//最大前缀和
	int rmax;//最大后缀和
	int tmax;//区间[l,r]最大连续子段和
}tr[N*4];

 同时pushup函数根据上图可以推出：（重载函数）

//u表示该节点，l表示该节点的左子树，r表示该节点的右子树
void pushup(node &u,node &l,node &r){
	u.sum=l.sum+r.sum;
	//三种最大连续子段和的情况
	u.lmax=max(l.lmax,l.sum+r.lmax);
	u.rmax=max(r.rmax,r.sum+l.rmax);
	u.tmax=max({l.tmax,r.tmax,l.rmax+r.lmax});
}

void pushup(int u){
	pushup(tr[u],tr[u<<1],tr[u<<1|1]);
}

代码附上：

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int N = 5e5+5;
int w[N];
int n,m;
struct node{
	int l,r;
	int sum;//[l,r]的区间和
	int lmax;//最大前缀和
	int rmax;//最大后缀和
	int tmax;//区间[l,r]最大连续子段和
}tr[N*4];

//u表示该节点，l表示该节点的左子树，r表示该节点的右子树
void pushup(node &u,node &l,node &r){
	u.sum=l.sum+r.sum;
	//三种最大连续子段和的情况
	u.lmax=max(l.lmax,l.sum+r.lmax);
	u.rmax=max(r.rmax,r.sum+l.rmax);
	u.tmax=max({l.tmax,r.tmax,l.rmax+r.lmax});
}

void pushup(int u){
	pushup(tr[u],tr[u<<1],tr[u<<1|1]);
}

void build(int u,int l,int r){
	if(l==r)tr[u]={l,r,w[r],w[r],w[r],w[r]};//找到叶子节点
	else{
		tr[u]={l,r};//设当前区间为[l,r]
		int mid=l+r>>1;
		build(u<<1,l,mid);//左子树
		build(u<<1|1,mid+1,r);//右子树
		pushup(u);//修改父节点
	}
}

//每次从1号节点开始找，找到位置位于x的数，并把它修改为v
void modify(int u,int x,int v){
	if(tr[u].l==x && tr[u].r==x)tr[u]={x,x,v,v,v,v};
	else{
		int mid=tr[u].l+tr[u].r>>1;
		if(x<=mid)modify(u<<1,x,v);//x位于当前区间的左半子区间
		else modify(u<<1|1,x,v);//x位于当前区间的右半子区间
		pushup(u);//修改父节点的相关信息
	}
}

node query(int u,int l,int r){
	if(tr[u].l>=l&&tr[u].r<=r)return tr[u];//被包含
	else{
		int mid=tr[u].l+tr[u].r>>1;
		if(r<=mid)return query(u<<1,l,r);//查询左半区间
		else if(l>mid)return query(u<<1|1,l,r);//查询右半区间
		else{//横跨左右区间
			auto left=query(u<<1,l,r);
			auto right=query(u<<1|1,l,r);
			node res;
			pushup(res,left,right);
			return res;
		}
	}
}

signed main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>n>>m;
	for(int i=1;i<=n;i++)cin>>w[i];
	build(1,1,n);//建树

	int k,x,y;
	while(m--){
		cin>>k>>x>>y;
		if(k==1){//查询
			if(x>y)swap(x,y);
			cout<<query(1,x,y).tmax<<"\n";
		}
		else modify(1,x,y);//修改
	}

	return 0;
}