树链剖分入门+HYSBZ - 1036树的统计Count

今天学习了树链剖分,记录一下。
【题目背景】
HYSBZ - 1036树的统计Count
在这里插入图片描述
【题目分析】
题目要求求任意结点之间路径的和以及路径上最大的结点,还有可能修改。如果正常做可能会很复杂(我也不知道正常应该怎么做,应该要用到LCA什么的,我还不太会)。
但是如果我们能够用线段树或者树状数组维护这个树,那么这种问题就会变得很简单。树链剖分就是这样一种将树映射在一个数组上变成线性结构然后用线段树进行维护的数据结构。
【基础知识】

  • 重儿子:儿子中子树结点数目最多的那个儿子(size最大)
  • 重边:父亲结点和重儿子连成的边
  • 重链:由多条重边连接而成的路径
  • 轻儿子:除了重儿子的其他儿子
  • 轻边:父亲和轻儿子连成的边
    如图所示,红圈的表示重儿子,黑边表示重边。由黑边组成的链为重链。
    在这里插入图片描述

【具体实现】
我们先进行一次遍历得到重儿子以及深度等信息储存起来

void dfs1(int u,int f)
{int i,v;siz[u]=1;	//储存该结点子树的大小(最小只有自身一个结点)son[u]=0;	//储存重儿子fa[u]=f;	//储存父节点h[u]=h[f]+1;//储存深度for(i=0;i<g[u].size();i++){v=g[u][i];if(v!=f){dfs1(v,u);	//深度优先遍历siz[u]+=siz[v];if(siz[son[u]]<siz[v]) son[u]=v;}}
}

得到以上数据后,我们可以按重链将树映射在一个数组上。从根节点开始,优先将重链映射到数组上,然后按照深度依次进行轻儿子,轻儿子又是某一个重链的开始(每一个节点都处于一个且仅有一个重链中)。记录每条每个节点所属重链的开头(从而判断两个节点是否在同一个重链上)。

void dfs2(int u,int f,int k)
{int i,v;top[u]=k;	//记录所属重链的开头pos[u]=++cnt;//映射到数组上的下标(同一个重链的下标是连续的)A[cnt]=val[u];//确定数组所对应节点的值方便进行维护if(son[u]) dfs2(son[u],u,k);//优先遍历重儿子,从而得到连续的重链for(i=0;i<g[u].size();i++){v=g[u][i];	if(v!=f&&v!=son[u]) dfs2(v,u,v);	//遍历其他轻儿子}
}

成功将树映射到数组上以后我们再用线段树对数组进行维护。对于线段树的维护是常规操作。

void update(int k,int l,int r,int x,int v)
{if(l==r){Sum[k]=Max[k]=v;return;}int mid=(l+r)/2;if(x<=mid) update(k<<1,l,mid,x,v);else update(k<<1|1,mid+1,r,x,v);Sum[k]=Sum[k<<1]+Sum[k<<1|1];Max[k]=max(Max[k<<1],Max[k<<1|1]);
}int QuerySum(int k,int l,int r,int L,int R)
{if(L<=l && r<=R) return Sum[k];int mid=(l+r)/2;int ret=0;if(L<=mid) ret+=QuerySum(k<<1,l,mid,L,R);if(R>mid) ret+=QuerySum(k<<1|1,mid+1,r,L,R);return ret;
}int QueryMax(int k,int l,int r,int L,int R)
{if(L==l && r==R) return Max[k];int mid=(l+r)/2;if(R<=mid) return QueryMax(k<<1,l,mid,L,R);else if(L>mid) return QueryMax(k<<1|1,mid+1,r,L,R);else return max(QueryMax(k<<1,l,mid,L,mid),QueryMax(k<<1|1,mid+1,r,mid+1,R));
}

重点还是对于树上两个点如何得到他们之间的一条路径以及这个路径在映射数组中的位置。我们每次从深度更深的点向上升,直到两个节点处在同一条链中(或者处于同一节点处)。在上升的过程中记录每条链的值(每条链都处于映射数组的一个连续的区间内)

int FindSum(int u,int v)
{int ans=0;while(top[u]!=top[v]){if(h[top[u]]<h[top[v]]) swap(u,v);ans+=QuerySum(1,1,n,pos[top[u]],pos[u]);u=fa[top[u]];}if(h[u]<h[v]) swap(u,v);ans+=QuerySum(1,1,n,pos[v],pos[u]);return ans;
}int FindMax(int u,int v)
{int ans=INT_MIN;while(top[u]!=top[v]){if(h[top[u]]<h[top[v]]) swap(u,v);ans=max(ans,QueryMax(1,1,n,pos[top[u]],pos[u]));u=fa[top[u]];}if(h[u]<h[v]) swap(u,v);ans=max(ans,QueryMax(1,1,n,pos[v],pos[u]));return ans;
}

这样我们就成功做到用线段树维护树状结构的数据啦
【AC代码】

#include<iostream>
#include<cstdio>
#include<vector>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<climits>using namespace std;const int MAXN=30010;
vector<int>g[MAXN];
int fa[MAXN],A[MAXN],val[MAXN],pos[MAXN],siz[MAXN],son[MAXN],h[MAXN],top[MAXN];
int cnt=0,n,m;
int Sum[MAXN<<2],Max[MAXN<<2];void dfs1(int u,int f)
{int i,v;siz[u]=1;son[u]=0;fa[u]=f;h[u]=h[f]+1;for(i=0;i<g[u].size();i++){v=g[u][i];if(v!=f){dfs1(v,u);siz[u]+=siz[v];if(siz[son[u]]<siz[v]) son[u]=v;}}
}
void dfs2(int u,int f,int k)
{int i,v;top[u]=k;pos[u]=++cnt;A[cnt]=val[u];if(son[u]) dfs2(son[u],u,k);for(i=0;i<g[u].size();i++){v=g[u][i];if(v!=f&&v!=son[u]) dfs2(v,u,v);}
}void update(int k,int l,int r,int x,int v)
{if(l==r){Sum[k]=Max[k]=v;return;}int mid=(l+r)/2;if(x<=mid) update(k<<1,l,mid,x,v);else update(k<<1|1,mid+1,r,x,v);Sum[k]=Sum[k<<1]+Sum[k<<1|1];Max[k]=max(Max[k<<1],Max[k<<1|1]);
}int QuerySum(int k,int l,int r,int L,int R)
{if(L<=l && r<=R) return Sum[k];int mid=(l+r)/2;int ret=0;if(L<=mid) ret+=QuerySum(k<<1,l,mid,L,R);if(R>mid) ret+=QuerySum(k<<1|1,mid+1,r,L,R);return ret;
}int QueryMax(int k,int l,int r,int L,int R)
{if(L==l && r==R) return Max[k];int mid=(l+r)/2;if(R<=mid) return QueryMax(k<<1,l,mid,L,R);else if(L>mid) return QueryMax(k<<1|1,mid+1,r,L,R);else return max(QueryMax(k<<1,l,mid,L,mid),QueryMax(k<<1|1,mid+1,r,mid+1,R));
}int FindSum(int u,int v)
{int ans=0;while(top[u]!=top[v]){if(h[top[u]]<h[top[v]]) swap(u,v);ans+=QuerySum(1,1,n,pos[top[u]],pos[u]);u=fa[top[u]];}if(h[u]<h[v]) swap(u,v);ans+=QuerySum(1,1,n,pos[v],pos[u]);return ans;
}int FindMax(int u,int v)
{int ans=INT_MIN;while(top[u]!=top[v]){if(h[top[u]]<h[top[v]]) swap(u,v);ans=max(ans,QueryMax(1,1,n,pos[top[u]],pos[u]));u=fa[top[u]];}if(h[u]<h[v]) swap(u,v);ans=max(ans,QueryMax(1,1,n,pos[v],pos[u]));return ans;
}int main()
{int a,b,i;char s[10];scanf("%d",&n);for(i=1;i<n;i++){scanf("%d%d",&a,&b);g[a].push_back(b);g[b].push_back(a);}for(i=1;i<=n;i++) scanf("%d",&val[i]);dfs1(1,0);dfs2(1,0,1);for(i=1;i<=n;i++) update(1,1,n,i,A[i]);scanf("%d",&m);while(m--){scanf("%s%d%d",s,&a,&b);if(s[1]=='H') update(1,1,n,pos[a],b);else if(s[1]=='S') printf("%d\n",FindSum(a,b));else printf("%d\n",FindMax(a,b));}return 0;
}

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/384125.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

Socket网络编程--小小网盘程序(5)

http://www.cnblogs.com/wunaozai/p/3893469.html 各位好呀&#xff01;这一小节应该就是这个小小网盘程序的最后一小节了&#xff0c;这一节将实现最后的三个功能&#xff0c;即列出用户在服务器中的文件列表&#xff0c;还有删除用户在服务器中的文件&#xff0c;最后的可以共…

进程相关概念

1.进程相关概念 进程是代码的一次动态执行&#xff0c;担当分配系统资源的角色&#xff0c;进程信息是被放在一个一个数据结构中&#xff0c;是一个结构体task_struct 2.进程控制块内容 //linux下的进程控制块 struct task_struct {volatile long state;// 说明了该进程是否可以…

SPOJ - QTREE3Query on a tree again!——树链剖分

【题目描述】 SPOJ - QTREE3Query on a tree again! 【题目分析】 题目要求是输出从111到xxx的路径上遇到的第一个黑色的点。我们可以用树链剖分&#xff08;不了解的同学请出门左拐&#xff0c;详见树链剖分入门&#xff09; 我们用线段树维护每个区间第一次遇到黑点的位置&a…

C++中的函数指针和函数对象总结

http://www.cnblogs.com/lvpengms/archive/2011/02/21/1960078.html 篇一、函数指针 函数指针&#xff1a;是指向函数的指针变量&#xff0c;在C编译时&#xff0c;每一个函数都有一个入口地址&#xff0c;那么这个指向这个函数的函数指针便指向这个地址。 函数指针的用途是很…

开发工具

1.编辑器 &#xff08;1&#xff09;vim     vim是从vi发展出来的一个文本编辑器。代码补完、编译错误跳转等方便编程的功能特别丰富&#xff0c;在程序员中被广泛使用。 &#xff08;2&#xff09;sed     sed是一种流编辑器&#xff0c;它一次处理一行内容。处理时…

c++ 智能指针用法详解

http://www.cnblogs.com/TenosDoIt/p/3456704.html 本文介绍c里面的四个智能指针: auto_ptr, shared_ptr, weak_ptr, unique_ptr 其中后三个是c11支持&#xff0c;并且第一个已经被c11弃用。 为什么要使用智能指针&#xff1a;我们知道c的内存管理是让很多人头疼的事&#xff0…

CodeForces - 786BLegacy——线段树建图+最短路

【题目描述】 CodeForces - 786BLegacy 【题目分析】 题目大概意思就是有三种操作&#xff1a; 从某个点到另一个点从某个点到另一个区间从某个区间到另一个点 然后询问从其中一个点到其他所有点的距离——这很显然是一个求单源最短路径的。我们简单的想法显然是建一个图&a…

自主编写shell

1.替换原理 用fork创建子进程后执行的是和父进程相同的程序&#xff08;但有可能执行不同的代码分支&#xff09;&#xff0c;子进程往往要调用一种exec函数以执行例外一个程序。当进程调用一种exec函数时&#xff0c;该进程的用户空间代码和数据完全被新程序替换&#xff0c;从…

HYSBZ - 2243染色——树链剖分+线段树建树技巧

【题目描述】 HYSBZ - 2243染色 【题目分析】 我一直没有看清楚题&#xff0c;以为求的是路径上出现颜色的种类&#xff0c;然后就写了一个区间染色的线段树进行维护&#xff0c;过样例的时候才发现题读错了&#xff0c;人家要求的是路径上出现的颜色段&#xff0c;所以颜色的…

打动态库和静态库

一.动态库和静态库的定义 1.静态库     程序在编译链接时把库的代码链接到可执行文件中。程序运行时就不再需要静态库 2.动态库     程序在运行的时候才去链接动态库的代码&#xff0c;多个程序 共享使用代码 3.动态链接     在执行文件之前&#xff0c;外部…

HYSBZ - 2157树链剖分

【题目描述】 HYSBZ - 2157树链剖分 【题目分析】 这道题给出的是边权而不是点权&#xff0c;但是我们分析这个树就会发现每个节点都只有一个父亲&#xff0c;也就是每条边的边权都可以存放在儿子节点上&#xff0c;然后在遍历路径的时候我们在从前往后遍历&#xff0c;但是注…

C++11中的右值引用

http://www.cnblogs.com/yanqi0124/p/4723698.html 在C98中有左值和右值的概念&#xff0c;不过这两个概念对于很多程序员并不关心&#xff0c;因为不知道这两个概念照样可以写出好程序。在C11中对右值的概念进行了增强&#xff0c;我个人理解这部分内容是C11引入的特性中最难以…

BZOJ2115XOR——线性基

【题目描述】 BZOJ2115XOR——线性基 【题目分析】 这道题看完以后很懵逼&#xff0c;人家要是走的很复杂呢&#xff1f;各种绕来绕去怎么办&#xff1f; 首先我们应该注意到一个很明显的道理&#xff1a;重复的路径会和自身抵消&#xff0c;所以我们大可以随便跑&#xff0c;…

单链表的相关操作

1.冒泡排序对单链表进行排序 void LinkListBubbleSort(LinkNode* head) {if(head NULL){ return;//空链表} if(head -> next NULL){ return;//只有一个结点} LinkNode* cur head;//趟数LinkNode* tail NULL;//尾指针LinkNode* tmp head;//次数for(; cur -…

socket网络编程--epoll小结

http://www.cnblogs.com/wunaozai/p/3895860.html 以前使用的用于I/O多路复用为了方便就使用select函数&#xff0c;但select这个函数是有缺陷的。因为它所支持的并发连接数是有限的(一般小于1024)&#xff0c;因为用户处理的数组是使用硬编码的。这个最大值为FD_SETSIZE&#…

进程间通信(匿名管道)

1.进程通信的目的 (1) 数据传输: 一个进程需要将它的数据传输给另一个进程     (2) 资源共享: 多个进程之间共享同样的资源     (3) 通知事件: 一个进程需要向另一个或一组进程发送消息, 通知它们发生了什么事情 2.管道 管道是一种进程之间通信的一种方式, 我们把从…

单例模式及C++实现代码

http://www.cnblogs.com/cxjchen/p/3148582.html 单例模式 单例模式&#xff0c;可以说设计模式中最常应用的一种模式了&#xff0c;据说也是面试官最喜欢的题目。但是如果没有学过设计模式的人&#xff0c;可能不会想到要去应用单例模式&#xff0c;面对单例模式适用的情况&am…

命名管道

1.命名管道的创建 (1) 通过命令创建 mkfifo filename (2)在程序中创建 int mkfifo(const char* filename, mode_t mode); 2. 命名管道和匿名管道的区别 (1)匿名管道由pipe函数创建并且打开     (2)命名管道有mkfifo函数创建由open函数打开     (3) fifo 之间的两…

HYSBZ - 1101——莫比乌斯反演

【题目描述】 HYSBZ - 1101 【题目分析】 昨天测试出了一道差不多的题目&#xff0c;我只能想到暴力&#xff0c;各种优化&#xff0c;最后都是运行了好久TLE&#xff0c;最后才知道要用到莫比乌斯反演&#xff0c;就想着今天研究一下&#xff0c;得出的结论就是&#xff0c;我…

Linux下I/O多路转接之select --fd_set

http://blog.csdn.net/li_ning_/article/details/52165993 fd_set 你终于还是来了&#xff0c;能看到这个标题进来的&#xff0c;我想&#xff0c;你一定是和我遇到了一样的问题&#xff0c;一样的疑惑&#xff0c;接下来几个小时&#xff0c;我一定竭尽全力&#xff0c;写出我…