数据结构综合粗讲
线段树
基本思想
将
少量特定的子区间,通过对这些少量子区间的修改或者统计,来实现快速对
可以解决的问题
满足加法性质的运算的区间问题,如区间和,区间异或,区间
不能解决的问题
区间众数等不符合区间加法性质的问题。
原理与实现
线段树通过递归将
如果根的高度为
通过一张图来解释线段树对于
单点查询
单点查询实际上就是定位到线段树的叶子结点。
我们现在假设我们需要定位到
单点修改
进行单点修改,首先也需要定位到这个结点(同单点查询)。然后修改完成后,我们需要一路往上更新,这样才能保证线段树的正确性。
区间查询
假设查询区间为
-
当前区间被目标区间完全包含。此时直接返回当前区间的值即可。
-
当前区间与目标区间无交集。此时返回
。 -
当前区间没有被目标区间包含且有交。此时递归处理左子树与右子树。
区间修改
需要进行区间修改的时候,我们需要引入一个新东西:懒标记。
对于一个区间
这个复杂度甚至比暴力还高。所以我们引入了懒标记。
懒标记的主要原理是区间修改操作时先对这个区间打上标记,暂时不进行更新,若之后需要用到该节点的信息时再进行更新。
单打标记的复杂度为一个常数。
时空复杂度
单次操作时间复杂度为
空间复杂度为
模板
这里以区间加法为例。
struct node{
int l,r;
ll v,tag;
}a[400001];
int n,m;
ll t[100001],sum[100001];
int ls(int u){
return u<<1;
}
int rs(int u){
return (u<<1)|1;
}
bool inrange(int L,int R,int l,int r){
return (L<=l)&&(r<=R);
}
bool outofrange(int L,int R,int l,int r){
return (R<l)||(r<L);
}
void build(int u,int L,int R){
a[u]=(node){L,R,sum[R]-sum[L-1],0};
if(L!=R){
int M=L+R>>1;
build(ls(u),L,M);
build(rs(u),M+1,R);
}
}
void pushup(int u){
a[u].v=a[ls(u)].v+a[rs(u)].v;
}
void pushdown(int u){
int L=a[u].l,R=a[u].r,M=L+R>>1,K=a[u].tag;
if(L==R) return ;
a[u].tag=0;
a[ls(u)].tag+=K;
a[rs(u)].tag+=K;
a[ls(u)].v+=K*(M-L+1);
a[rs(u)].v+=K*(R-M);
}
void update(int u,int L,int R,ll k){
if(a[u].tag) pushdown(u);
if(inrange(L,R,a[u].l,a[u].r)){
a[u].tag+=k;
a[u].v+=k*(a[u].r-a[u].l+1);
pushdown(u);
}
else if(!outofrange(L,R,a[u].l,a[u].r)){
update(ls(u),L,R,k);
update(rs(u),L,R,k);
pushup(u);
}
}
ll search(int u,int L,int R){
if(a[u].tag) pushdown(u);
if(inrange(L,R,a[u].l,a[u].r)){
return a[u].v;
}
else if(!outofrange(L,R,a[u].l,a[u].r)){
return search(ls(u),L,R)+search(rs(u),L,R);
}
else return 0ll;
}
题目
P3372、P3373、P1253、P4145、P1198、P1531、P1471、P1972、P1438
树状数组
基本思想
树状数组的核心思想是将一个需要操作的区间分解成若干小区间,在维护的时候,直接对这些小区间进行处理,在查询的时候再将这些区间组装成我们想要的区间。
可以解决的问题
满足减法性质的运算的区间问题,如区间和,区间异或等。
不能解决的问题
区间众数,区间
原理与实现
这个玩意大概长成这个样子:
(这里用了百度的图片)
它就是一个特殊的前缀和数组。
仔细观察红色框内与灰色框的关系:
于是可以发现以下规律:
那么找出
那么问题来了,怎么获取最低位的
这时候就要引入
先假设该数最低位的
知道了以上知识以后,便可以写出修改函数:
void add(int x,ll y){//在位置x的数加上y
for(int i=x;i<=n;i+=lowbit(i)){
c[i]+=y;
}
}
那么上面那个公式可以这么写:
那么在跑代码的过程中,数据结构内部发生了啥?这里用
可以看到,我们要想单点修改
区间查询
利用前缀和思想,我们可以知道求
那么把问题拆开来看,如何求
我们可以先将
那么我们接下来可以将
不断重复以上操作,直到
代码如下:
ll search(int x,int y){//查询x到y的和
int sum1=0,sum2=0;
for(int i=x-1;i;i-=lowbit(i)){
sum1+=c[i];
}
for(int i=y;i;i-=lowbit(i)){
sum2+=c[i];
}
return sum2-sum1;
}
我们还是来看看树状数组内部发生的事情,这里拿查询区间
可以看到每一步中,都把
模板
#include <bits/stdc++.h>
#define ll long long
#define lowbit(x) ((x)&(-x))
using namespace std;
int n,m;
ll a[500001],c[500001];
void add(int x,ll k){
for(int i=x;i<=n;i+=lowbit(i)){
c[i]+=k;
}
}
ll search(int x,int y){
int sum1=0,sum2=0;
for(int i=x-1;i;i-=lowbit(i)){
sum1+=c[i];
}
for(int i=y;i;i-=lowbit(i)){
sum2+=c[i];
}
return sum2-sum1;
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%lld",&a[i]);
add(i,a[i]);
}
for(int i=1;i<=m;i++){
int op;
scanf("%d",&op);
if(op==1){
int x;
ll k;
scanf("%d%lld",&x,&k);
add(x,k);
}
else{
int x,y;
scanf("%d%d",&x,&y);
printf("%lld\n",search(x,y));
}
}
return 0;
}
题目
P3374、P3368、P3372、P4939、P5057、CF652D、CF61E、SP3267、P3531
ST表
基本思想
预处理+倍增。
可以解决的问题
满足不重复贡献的问题,如 RMQ 问题与区间
不能解决的问题
不满足不重复贡献性质,如区间和。
因为 ST 表需要预处理,所以也不能处理动态区间问题。
动态RMQ还是写带修莫队罢
原理与实现
预处理
使用一个二维数组存储一定范围信息,例如
for(int i=1;(1<<i)<=n;i++){
for(int j=1;j<=n;j++){
if(j+(1<<i)-1<=n){
st[j][i]=max(st[j][i-1],st[j+(1<<(i-1))][i-1]);
}
}
}
查询
对于区间
for(int i=1;i<=m;i++){
int l,r,k;
scanf("%d%d",&l,&r);
k=log2(r-l+1);
printf("%d\n",max(st[l][k],st[r-(1<<k)+1][k]));
}
模板
#include<bits/stdc++.h>
using namespace std;
int n,m,a[100001],st[100001][31];
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
st[i][0]=a[i];
}
for(int i=1;(1<<i)<=n;i++){
for(int j=1;j<=n;j++){
if(j+(1<<i)-1<=n){
st[j][i]=max(st[j][i-1],st[j+(1<<(i-1))][i-1]);
}
}
}
for(int i=1;i<=m;i++){
int l,r,k;
scanf("%d%d",&l,&r);
k=log2(r-l+1);
printf("%d\n",max(st[l][k],st[r-(1<<k)+1][k]));
}
return 0;
}
题目
并查集
基本思想
通过合并操作来快速查询一个元素是否在一个集合中。
可以解决的问题
原理与实现
初始化
我们设
一开始显然每个节点都只会有这一个元素,所以每个节点的父节点初始化为自己。
for(int i=1;i<=n;i++){
f[i]=i;
}
合并
如果要合并
查询
查询
路径压缩
为什么要路径压缩?
考虑并查集的这种情况:
这样的话,每次查询操作的时间复杂度就会退化为线性。
那么怎么进行路径压缩?
我们每次查询的时候直接把查询一路上的所有点的
我们对上图进行路径压缩:
这样我们每次查询的时间复杂度就会变为
模板
int find(int x){
if(f[x]!=x){
f[x]=find(f[x]);
}//路径压缩
return f[x];
}
void uunion(int x,int y){
f[find(x)]=find(y);
}
void init(){
for(int i=1;i<=n;i++){
f[i]=i;
}
}
题目
P1111、P1197、P1455、P1536、ABC293D、P1892、P2078、P2256、P2294、P2814
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列01:轻松3步本地部署deepseek,普通电脑可用
· 25岁的心里话
· 按钮权限的设计及实现