[蓝桥杯学习]树上差分

差分

前缀和 sum_i = sum_i-1 + a_i

差分 diff_i = a_i - a_i-1

差分的好处

点的差分

问题引入
解决问题

要用到差分的思想,每次从叶子向上的回溯,让父结点+=子结点的cnt值,但是仅仅这样,还不行

回溯的过程中,LCA被加了两次,要减去一,LCA的父结点原本应该没有数值的,但是因为会加上LCA 的值,所以会多1,要减去。

此上操作,便实现了,只是把s t 两个结点的值赋一,便将s到t路径上的点经过次数赋值为1

代码实现

cnt计算的代码实现,如下

cpp 复制代码
void dfs1(int x){
    for(int i=head[x];i;i=edge[i].nex){
        int u=edge[i].to;
        if(u!=fa[x][0]){
            dfs1(u);
            cnt[x]+=cnt[u];
        }
    }
    return ;
}

cnt[x]+=cnt[u] ,在dfs1 之后,就是从子结点回到父结点后进行cnt相加。

边的差分

问题引入

将路上每条边都权值加z

解决问题

把边塞给点,边的权值放在点上,使用差分的方法,每次修改都是O(1)的时间复杂度,将最后的结果最坏下是O(n)遍历要求的区间,就可以得到区间权值和。

解释为什么要把 dlt[lca(u,v)]-=2x

因为根据方法,回溯时父结点会加上子结点的权值,所以LCA会加上左右两边的孩子的权值(即边的权值),根据我们给点权值的定义,意思是,会经过该边两次,但是该边没有经过两次,所以会减去2x

代码实现

与点上的差分一样的

时间复杂度分析

维护的时间复杂度是O(1),最后dfs一遍统计答案,最坏时间复杂度是O(n)

相关推荐
做cv的小昊7 小时前
计算机图形学:【Games101】学习笔记05——着色(插值、高级纹理映射)与几何(基本表示方法)
笔记·opencv·学习·计算机视觉·图形渲染·几何学
车载测试工程师8 小时前
CAPL学习-CAN相关函数-统计API函数
网络·网络协议·学习·capl·canoe
好奇龙猫9 小时前
【AI学习-comfyUI学习-第二十四节-open(contorlnet多重处理)+图生图openpose-各个部分学习】
人工智能·学习
wanzhong23339 小时前
CUDA学习5-矩阵乘法(共享内存版)
深度学习·学习·算法·cuda·高性能计算
PNP Robotics11 小时前
PNP机器人受邀参加英业达具身智能活动
大数据·人工智能·python·学习·机器人
iconball11 小时前
个人用云计算学习笔记 --24 虚拟化、KVM 基础使用与热迁移实验、VMware ESXi笔记
运维·笔记·学习·云计算
奕辰杰13 小时前
Netty私人学习笔记
笔记·学习·netty·网络通信·nio
Larry_Yanan14 小时前
Qt多进程(五)QUdpSocket
开发语言·c++·qt·学习·ui
De-Alf14 小时前
Megatron-LM学习笔记(6)Megatron Model Attention注意力与MLA
笔记·学习·算法·ai
Gary Studio15 小时前
MPP充电学习笔记
笔记·学习