高级数据结构—线段树(一)

学线段树的原因是因为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;
}

相关推荐

  1. 数据结构——线索

    2024-04-26 18:34:05       76 阅读
  2. 【C++】线段

    2024-04-26 18:34:05       60 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-04-26 18:34:05       98 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-04-26 18:34:05       106 阅读
  3. 在Django里面运行非项目文件

    2024-04-26 18:34:05       87 阅读
  4. Python语言-面向对象

    2024-04-26 18:34:05       96 阅读

热门阅读

  1. Vue 3组合式API深度剖析:核心API使用指南

    2024-04-26 18:34:05       33 阅读
  2. UE5主视口导航快捷键汇总

    2024-04-26 18:34:05       37 阅读
  3. vue2中的文件命名规范

    2024-04-26 18:34:05       30 阅读
  4. Spring(25) 为什么使用 SpringCloud,而不是用 Dubbo?

    2024-04-26 18:34:05       29 阅读
  5. 【后端】python数组去重和过滤的使用方法

    2024-04-26 18:34:05       35 阅读