树状数组与线段树模板集合

最新推荐文章于 2024-05-28 21:13:10 发布
最新推荐文章于 2024-05-28 21:13:10 发布
阅读量98 收藏 1
点赞数
分类专栏： c++进阶之路 程序 文章标签： c++ 算法
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/yyf525/article/details/125864298
版权

前言

马上就要 csp \texttt{csp} csp了， 树状数组和线段树肯定是复习不可少的一部分。今天作者来整理一下它们的模板。

线段树

I. \texttt{I.} I.单点修改，区间查询。
例题：P3374
代码：

#include <bits/stdc++.h>
using namespace std;
#define lid id*2
#define rid id*2+1
const int N=5e6+7;
int n,m;
int a[N];
struct sa{
	int l;
	int r;
	int sum;
}tree[N<<1];
void build(int id,int l,int r){
	tree[id].l=l,tree[id].r=r;
	if(l==r){
		tree[id].sum=a[l];
		return;
	}
	int mid=(l+r)>>1;
	build(lid,l,mid);
	build(rid,mid+1,r);
	tree[id].sum=tree[lid].sum+tree[rid].sum;
	return;
}
int search(int id,int l,int r){
	if(tree[id].l>r||tree[id].r<l)return 0;
	if(tree[id].l>=l&&tree[id].r<=r){
		return tree[id].sum;
	}
	int s=0;
	if(tree[lid].r>=l){
		s+=search(lid,l,r);
	}
	if(tree[rid].l<=r){
		s+=search(rid,l,r);
	}
	return s;
}
void add(int id,int x,int w){
	if(tree[id].l==tree[id].r){
		tree[id].sum+=w;
		return;
	}
	int mid=(tree[id].l+tree[id].r)>>1;
	if(x<=mid)add(lid,x,w);
	else add(rid,x,w);
	tree[id].sum=tree[lid].sum+tree[rid].sum;
	return;
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++)cin>>a[i];
	build(1,1,n);
	while(m--){
		int op,x,y;
		cin>>op>>x>>y;
		if(op==1){
			add(1,x,y);
		}
		else{
			cout<<search(1,x,y)<<'\n';
		}
	}
	return 0;
}

II. \texttt{II.} II.区间修改，单点查询。
例题：P3368
代码：

#include <bits/stdc++.h>
using namespace std;
void sleep(int p);int read();int read(int p);
#define ll long long
#define ull unsigned long long
#define db double
#define lid id*2
#define rid id*2+1
const int N=3e6+7,inf=0x3f3f3f3f,mod=1e9+7;
struct sa{
	int sum;
	int l,r;
}tree[N];
int n,ans;
int a[N];
void build(int id,int l,int r){
	tree[id].l=l,tree[id].r=r;
	if(l==r){
		tree[id].sum=a[l];
		return ;
	}
	int mid=l+r>>1;
	build(lid,l,mid);
	build(rid,mid+1,r);
	return;
}
void modify(int id,int l,int r,int k){
	if(tree[id].l>=l&&tree[id].r<=r){
		tree[id].sum+=k;
		return;
	}
	if(tree[id].l>r||tree[id].r<l)return;
	int mid=tree[id].l+tree[id].r>>1;
	if(mid>=l)modify(lid,l,r,k);
	if(mid<r)modify(rid,l,r,k);
	return;
}
void query(int id,int dis){
	ans+=tree[id].sum;
	if(tree[id].l==tree[id].r)
		return;
	int mid=tree[id].l+tree[id].r>>1;
	if(dis<=mid)query(lid,dis);
	else query(rid,dis);
	return ;
}
int main()
{
	ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++)cin>>a[i];
	build(1,1,n);
	while(m--){
		int c;
		cin>>c;
		if(c==1){
			int x,y,z;
			cin>>x>>y>>z;
			modify(1,x,y,z);
		}
		else{
			ans=0;
			int x;cin>>x;
			query(1,x);
			cout<<ans<<'\n';
		}
	}
	return 0;
}

III. \texttt{III.} III.区间修改，区间查询
例题：P3372
代码：

#include <bits/stdc++.h>
using namespace std;
#define lid id*2
#define rid id*2+1
const long long N=5e5+7;
long long n,m;
long long a[N];
struct sa{
	long long l,r,sum;
	long long lz;
}tree[N];
void build(long long id,long long l,long long r){
	tree[id].l=l,tree[id].r=r,tree[id].lz=0;
	if(l==r){
		tree[id].sum=a[l];
		return;
	}
	long long mid=l+r>>1;
	build(lid,l,mid);
	build(rid,mid+1,r);
	tree[id].sum=tree[lid].sum+tree[rid].sum;
	return;
}
void push_down(long long id){
	if(!tree[id].lz)return ;
	long long mid=tree[id].l+tree[id].r>>1;
	tree[lid].lz+=tree[id].lz;
	tree[lid].sum+=(mid-tree[id].l+1)*tree[id].lz;
	tree[rid].lz+=tree[id].lz;
	tree[rid].sum+=(tree[rid].r-mid)*tree[id].lz;
	tree[id].lz=0;
	return;
}
void update(long long id,long long l,long long r,long long k){
	if(tree[id].r<=r&&tree[id].l>=l){
		tree[id].sum+=k*(tree[id].r-tree[id].l+1);
		tree[id].lz+=k;
		return;
	}
	push_down(id);
	long long mid=tree[id].l+tree[id].r>>1;
	if(l<=mid)update(lid,l,r,k);
	if(mid<r)update(rid,l,r,k);
	tree[id].sum=tree[lid].sum+tree[rid].sum;
	return;
}
long long search(long long id,long long l,long long r){
	if(tree[id].l>=l&&tree[id].r<=r)
		return tree[id].sum;
	if(tree[id].r<l|tree[id].l>r)return 0;
	push_down(id);
	long long s=0;
	if(tree[lid].r>=l)s+=search(lid,l,r);
	if(tree[rid].l<=r)s+=search(rid,l,r);
	return s;
}
int main(){
	cin>>n>>m;
	for(long long i=1;i<=n;i++)cin>>a[i];
	build(1,1,n);
	while(m--){
		long long x,y,z,k;
		cin>>x>>y>>z;
		if(x==1){
			cin>>k;
			update(1,y,z,k);
		}
		else{
			cout<<search(1,y,z)<<'\n';
		}
	}
	return 0;
}

IV. \texttt{IV.} IV.区间加法、乘法，区间查询。
例题：P3373
代码：

#include <bits/stdc++.h>
using namespace std;
void sleep(int p);long long read();int read(long long p);
#define lid id*2
#define rid id*2+1
typedef long long ll;
#define ull unsigned long long
#define db double
const int N=5e5+5,inf=0x3f3f3f3f,mod=1e9+7;
int n,m;
ll p;
ll a[N];
struct sa{
	ll l,r;
	ll sum,add,mul;
}tree[N];
inline void init(int id){
	tree[id].sum%=p,tree[id].mul%=p,tree[id].sum%=p;
	return;
}
void build(int id,ll l,ll r){
	tree[id].l=l,tree[id].r=r,tree[id].mul=1;
	if(l==r){
		tree[id].sum=a[l]%p;
		return;
	}
	int mid=tree[id].l+tree[id].r>>1;
	build(lid,l,mid);build(rid,mid+1,r);
	tree[id].sum=tree[lid].sum+tree[rid].sum;
	tree[id].sum%=p; 
	return;
}
void push_down(int id){
	int mul=tree[id].mul,add=tree[id].add;
	tree[lid].sum=tree[lid].sum*mul+add*(tree[lid].r-tree[lid].l+1);
	tree[rid].sum=tree[rid].sum*mul+add*(tree[rid].r-tree[rid].l+1);
	tree[lid].mul*=mul,tree[rid].mul*=mul;
	tree[lid].add=(tree[lid].add*mul+add)%p;
	tree[rid].add=(tree[rid].add*mul+add)%p;
	tree[id].mul=1,tree[id].add=0;
	init(lid),init(rid);
	return;
}
void update(int id,bool op,ll l,ll r,ll k){
	if(tree[id].l>=l&&tree[id].r<=r){
		if(op){
			tree[id].mul*=k;
			tree[id].add*=k;
			tree[id].sum*=k;
			init(id);
		}
		else{
			tree[id].sum+=(tree[id].r-tree[id].l+1)*k%p;
			tree[id].add+=k;
			init(id);
		}
		return;
	}
	push_down(id);
	ll mid=tree[id].l+tree[id].r>>1;
	if(mid>=l)update(lid,op,l,r,k);
	if(mid+1<=r)update(rid,op,l,r,k);
	tree[id].sum=(tree[lid].sum+tree[rid].sum)%p;
	return;
}
ll query(int id,int l,int r){
	if(tree[id].l>=l&&tree[id].r<=r){
		return tree[id].sum;
	}
	if(tree[id].l>r||tree[id].r<l)return 0;
	push_down(id);
	int mid=tree[id].l+tree[id].r>>1;ll s=0;
	if(mid>=l)s+=query(lid,l,r);
	if(mid+1<=r)s+=query(rid,l,r);
	return s%p;
}
int main()
{
	cin>>n>>m>>p;
	for(int i=1;i<=n;i++){
	    a[i]=read();a[i]%=p;
	}
	build(1,1,n);
	while(m--){
		ll x=read(),y=read(),z=read(),k;
		if(x==1){
		    k=read();k%=p;
			update(1,1,y,z,k);
		}else
		if(x==2){
			k=read();k%=p;
			update(1,0,y,z,k);
		}else{
		    printf("%lld\n",query(1,y,z));
		}
	}
	return 0;
}
void sleep(int p){for(int i=1;i<=p*100000;i++)int x;return;}
inline long long read(){long long x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}return x*f;}
inline int read(long long p){long long x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();x=x%p;}return x*f;}

树状数组

I. \texttt{I.} I.单点修改，区间查询。
例题：P3374
代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define ull unsigned long long
const int N=6e5+5,mod=1e9+7;
inline void sleep(int p);
inline int read();
inline int read(int mod);
int n,m;
int a[N];
int c[N];
int lbt(int x){return x & -x;}
void add(int x,int k){
	for(;x<=n;x+=lbt(x))c[x]+=k;
	return;
}
int ask(int x){
	int t=0;
	for(;x;x-=lbt(x))t+=c[x];
	return t;
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		add(i,a[i]);
	}
	while(m--){
		int op;
		cin>>op;
		if(op==1){
			int x,k;
			cin>>x>>k;
			add(x,k);
		}
		else{
			int l,r;
			cin>>l>>r;
			cout<<ask(r)-ask(l-1)<<'\n';
		}
	}
	return 0;
}
inline void sleep(int p){for(int i=1;i<=100000*p;i++)int x;return;}
inline int read(){int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}return x*f;}
inline int read(int p){int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();x=x%p;}return x*f;}

II. \texttt{II.} II.区间修改，单点查询。
例题：P3368
代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define ull unsigned long long
const int N=6e5+5,mod=1e9+7;
inline void sleep(int p);
inline int read();
inline int read(int mod);
int n,m;
int a[N];
int b[N];
int c[N];
int lbt(int x){return x & -x;}
void add(int x,int k){
	for(;x<=n;x+=lbt(x))c[x]+=k;
	return;
}
int ask(int x){
	int t=0;
	for(;x;x-=lbt(x))t+=c[x];
	return t;
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		b[i]=a[i]-a[i-1];
		add(i,b[i]);
	}
	while(m--){
		int op;
		cin>>op;
		if(op==1){
			int l,r,x;
			cin>>l>>r>>x;
			add(l,x);
			if(r<n)add(r+1,-x);
		}
		else{
			int l;
			cin>>l;
			cout<<ask(l)<<'\n';
		}
	}
	return 0;
}
inline void sleep(int p){for(int i=1;i<=100000*p;i++)int x;return;}
inline int read(){int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}return x*f;}
inline int read(int p){int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();x=x%p;}return x*f;}

III. \texttt{III.} III.区间修改，区间查询。
例题：P3372
代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define int long long
const int N=7e5+5,mod=1e9+7;
inline void sleep(int p);
inline int read();
inline int read(int mod);
int n,m;
int a[N];
int b[N];
int c[2][N];
void add(int op,int x,int k){
	for(;x<=n;x+=x&-x)c[op][x]+=k;
	return;
}
int ask(int op,int x){
	int t=0;
	for(;x;x-=x&-x)t+=c[op][x];
	return t;
}
signed main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		b[i]=b[i-1]+a[i]; 
	}
	while(m--){
		int op;
		cin>>op;
		if(op==1){
			int l,r,d;
			cin>>l>>r>>d;
			add(0,l,d);
			add(0,r+1,-d);
			add(1,l,d*l);
			add(1,r+1,-d*(r+1));
		}
		else{
			int l,r;
			cin>>l>>r;
			int ans=b[r]+(r+1)*ask(0,r)-ask(1,r);
			ans-=b[l-1]+l*ask(0,l-1)-ask(1,l-1); 
			cout<<ans<<'\n';
		}
	}
	return 0;
}
inline void sleep(int p){for(int i=1;i<=100000*p;i++)int x;return;}
inline int read(){int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();}return x*f;}
inline int read(int p){int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=(x<<1)+(x<<3)+(ch^48);ch=getchar();x=x%p;}return x*f;}

结语

希望大家看了这篇博文后，能有更多的进步。如果想进一步提升 OI \texttt{OI} OI 水平，可以做一下这道题，本人原创。

yyf525
关注
  • 0
    点赞
  • 踩
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
并查集 树状数组 线段树 主席树 树链剖分（网址）
qq_43507388的博客
07-06 141
这只是一篇网址集合因为好像如果放在浏览器文件夹里便永远不会再打开 并查集 基础模板+HDU1232+POJ1611两道例题： https://blog.csdn.net/oliver233/article/details/70162173 树状数组 最详细的两篇博客： https://blog.csdn.net/bestsort/article/details/80796531 https://b...
HDOJ-1156 Brownie Points II 线段树/树状数组（模板）
dong52130的博客
07-18 93
http://acm.hdu.edu.cn/showproblem.php?pid=1156 在一张二位坐标系中，给定n个点的坐标，玩一个划线游戏（线必须穿过点），Stan先手画一条垂直的线，然后Ollie画一条水平的线（要求要穿过Stan那条线所穿过的某个点）。划分后，左上和右下点数是Ollie 的得分，左下和右上是Stan的得分。求Stan在保证最低得分（即不论Ollie后手怎么划...
线段树模板讲解
dengtun2502的博客
12-30 71
洛谷题目链接:线段树 线段树是一种用于区间修改查询的数据结构,可以支持的操作有单点修改区间查询,区间修改单点查询,区间修改区间查询等. 线段树有递归版和结构体版，递归版在处理一开始没有赋初始值的问题时可以不用建树，而结构体版的则显得比较条理清晰． 线段树比树状数组的代码复杂的多,但是树状数组多支持一个区间修改区间查询的操作,并且可以与其他的一些数据结构相适应(像是树链剖分等),也有...
单调栈、单调队列、线段树、LCA、二维树状数组、Bitset讲解
jmsyzsfq的博客
09-22 513
转自lzj209一、单调栈1.问题引入考虑这样一个问题，给出一个数字序列，一段连续的子序列的权值定义为这个子序列中最小的权值乘以子序列的长度，求最大的子序列权值，数据范围O(n)可过。2.问题转化稍加思索我们可以得到一个O(n^2)的算法，但是显然是太慢，我们想办法换一种枚举方式，假如我们枚举每一个数作为最小值，计算这个序列最大能扩张多长，那么这样就优化成了O(n)个数进行比较，于是只要能快速算出一
python 树状数组_nlogn求LIS（树状数组）
weixin_39843698的博客
02-19 177
之前一直是用二分但是因为比较难理解，写的时候也容易忘记怎么写。今天比赛讲评的时候讲了一种用树状数组求LIS的方法(1)好理解，自然也好写(但代码量比二分的大)(2)扩展性强。这个解法顺带求出以i为结尾的LIS，而很多题要用到这个数组来做而二分的做法求得是当前长度下的最小值，不容易拓展。#include#define REP(i, a, b) for(register int i = (a); i ...
线段树集训
weixin_45360119的博客
11-12 248
线段树集训 P3372 【模板】线段树 1 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn) 区间修改 + 区间查询（板子题） #include <iostream> #include <cstdio> #include <cstring> #include <algorithm> using namespace std; typedef long long ll; const int N = 1e5 + 50; int n, m; ll a
线段树合集整理
zjy_code的博客
08-11 203
作为oier刷题是碰见的大数据结构的集合 作者水平有限，如有不足欢迎批评指正 会随时更新其中的题目
线。段。树--树状数组-主席树
weixin_54010759的博客
04-11 2569
简单了解一下线段树 以前写过的内容，搬运过来 线段树的应用场景：满足区间加法性质且多次查询，什么是区间加法性质，比如最大值，求和，树状数组、线段树、主席树依次。 线段树框架：建树--查询--更新。。。懒标记 代码关于下标，原始数据下标，节点下标，节点表示的区间[l,r] 原始数组下标从0/1开始，节点表示区间的意义，节点下标与原始数组下标的关系 板子题，线段树模板 给你一个长为n的序列，有m个操作，全局加，或者查询区间最大子段和。 输入描述: 第一行两个数n,m 之后一行n个数表示.
权值线段树模板+练习题整理（刷题中）
蒲公英殇的博客
03-09 738
权值线段树自己的一些练习
树状数组&线段树.pptx
07-28
技术文档分享，免费获取请私信博主。
线段树树状数组课件（ppt）
02-16
线段树&树状数组课件 树状数组&线段树是最基本的高级数据结构之二 一般出现于较难题中 应用广泛，可用于直接写正解/把暴力改进成正解/拿大量部分分
算法NOIP树状数组校门外的树.pdf
05-02
本文详细介绍了树状数组的原理，并用树状数组解校门外的树这个问题.
线段树与树状数组 ACM课件
04-23
线段树与树状数组 ACM课件，通过这个课件你可以学到很多，里面也有代码示例
poj 2352 stars（树状数组，线段树）
07-18
这是一道很不错的题目，即可以用线段树做也可以用树状数组，可谓经典。不过当然了线段树是比较难搞，而树状数组是极其简洁的，构造很简单，下面就分别来介绍一下两种方法...
C++：round函数用法
Solititude的博客
05-28 341
C++：round函数用法
精通C++ STL（二）：string类的模拟实现
Russ_Leo的博客
05-27 818
string类各函数接口。
【如何在Qt C++中使用SSL和TLS加密传输数据？】
最新发布
m0_45463480的博客
05-28 200
在这个示例中，我们创建了一个简单的Qt应用程序，向指定的HTTPS地址发送请求，并显示响应。信号来处理响应数据。您可以从响应中读取数据并进行进一步处理。来配置SSL/TLS连接，包括对证书进行验证和加密要求等。为您要访问的实际URL，并根据需要进一步配置SSL策略。在发送请求之前，可设置SSL配置。确保您的Qt项目已经链接了网络模块（对象发送网络请求，并接收响应。对象指定要访问的URL。
关于C++的特殊类定制
m0_62126248的博客
05-28 363
C++的一些特殊性质的类如何进行设计
树状数组和线段树的区别
03-27
1. 存储方式：树状数组使用数组存储数据，而线段树使用树形结构存储。因此，树状数组的空间复杂度较低，而线段树的空间复杂度较高。 2. 操作复杂度：树状数组和线段树都支持查询和更新操作，但是它们的复杂度不同。...

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值