【 Gym - 101138J 】Valentina and the Gift Tree(树链剖分)
BUPT2017 wintertraining(15) 4 D
Gym - 101138J
数据
题意
n个节点的一棵树,每个节点的权值为g,q个询问,树上的节点U-V,求U到V的路径的最大子段和。
题解
先考虑这么一个问题:求区间[L,R]的最大子段和。
q个询问,用线段树可以做到每个询问的时间是O(log n)。
线段树的节点x代表区间[L,R],我们要存这些值:
- sum: 区间[L,R]的总和
- s: 区间[L,R]的最大子段和
- suml: 最大前缀和
- sumr: 最大后缀和
叶子节点这些值都是g[L],否则先求出左右孩子的值,然后可以计算出当前节点的值。
再考虑原问题,因为是在树上,我们可以用两次dfs把每个节点放到线段树的区间上,使得每个节点和它最长的儿子在区间上是相邻的,其实就是树链剖分一下。
具体地说(后面的内容最好对着代码看),就是第一次dfs求得每个节点的父亲fa,重儿子son,子树节点总数siz,深度dep。第二次dfs,标记u所在的链的顶标top,在区间的下标rk=++w,以及该下标对应的节点id[w]=u,然后如果有son,则dfs2(son,f),f是当前的链的顶标,接着对于非重儿子v,dfs2(v,v),因为非重儿子重新起一条链,顶标是本身。
对于查询u和v,将顶标深度大的点往上翻,直到他们的顶标相同。翻的过程求出这一段路径 和 之前翻的路径 合并成的路径的sum等值,也就是用线段树节点来保存。过程和求线段树的差不多。然后顶标相同后,再将深度大的节点和之前的路径合并一下,最后我们有两个点la,ra代表(u,x)和(x,v),x是最近公共祖先,注意他们的合并是suml+suml,不明白可以画一下,答案就是max(la.s,ra.s,la.suml+ra.suml)。
代码
#include <cstdio>
#include <cstring>
#include <algorithm>
#define N 100005
#define inf (1LL<<60)
#define ll long long
using namespace std;
struct edge{
int to,next;
}e[N<<1];
int head[N],cnt;
int n,q,g[N];
int dep[N],fa[N],siz[N],son[N];
int top[N],rk[N],w,id[N];
void add(int u,int v){
e[++cnt]=(edge){v,head[u]};head[u]=cnt;
}
void dfs1(int u){
siz[u]=1;
for(int i=head[u],v;i;i=e[i].next)if(!dep[v=e[i].to]){
fa[v]=u;
dep[v]=dep[u]+1;
dfs1(v);
siz[u]+=siz[v];
if(siz[son[u]]<siz[v])son[u]=v;
}
}
void dfs2(int u,int f){
top[u]=f;
rk[u]=++w;
id[w]=u;
if(!son[u])return;
dfs2(son[u],f);
for(int i=head[u],v;i;i=e[i].next){
v=e[i].to;
if(v==son[u]||v==fa[u])continue;
dfs2(v,v);
}
}
#define lson l,m,x<<1
#define rson m+1,r,x<<1|1
struct node{
ll s,sum,suml,sumr;
node(){
sum=0,s=suml=sumr=-inf;
}
}tr[N<<2];
void pushUp(node& x,node ls,node rs){
x.s=max(max(ls.s,rs.s),ls.sumr+rs.suml);
x.sum=ls.sum+rs.sum;
x.suml=max(ls.suml,ls.sum+rs.suml);
x.sumr=max(rs.sumr,rs.sum+ls.sumr);
}
void build(int l,int r,int x){
if(l==r){
tr[x].s=tr[x].suml=tr[x].sumr=tr[x].sum=g[id[l]];
return;
}
int m=l+r>>1;
build(lson);build(rson);
pushUp(tr[x], tr[x<<1], tr[x<<1|1]);
}
node query(int L,int R,int l,int r,int x){
if(L<=l&&r<=R)return tr[x];
int m=l+r>>1;
if(R<=m)return query(L,R,lson);
if(L>m) return query(L,R,rson);
node now;
pushUp(now, query(L,R,lson), query(L,R,rson));
return now;
}
int main() {
scanf("%d",&n);
for(int i=1,u,v;i<n;i++)
scanf("%d%d",&u,&v), add(u,v), add(v,u);
for(int i=1;i<=n;i++) scanf("%d",&g[i]);
dep[1]=1, dfs1(1), dfs2(1,1), build(1,n,1);
scanf("%d",&q);
while(q--){
int a,b;node la,ra;
scanf("%d%d",&a,&b);
int ta=top[a],tb=top[b];
while(ta!=tb){
if(dep[ta]<dep[tb])
pushUp(ra, query(rk[tb],rk[b],1,n,1), ra), b=fa[tb], tb=top[b];
else pushUp(la, query(rk[ta],rk[a],1,n,1), la), a=fa[ta], ta=top[a];
}
if(dep[a]<dep[b]) pushUp(ra, query(rk[a],rk[b],1,n,1), ra);
else pushUp(la, query(rk[b],rk[a],1,n,1), la);
printf("%lld\n",max(max(la.s,ra.s),la.suml+ra.suml));
}
return 0;
}
【 Gym - 101138J 】Valentina and the Gift Tree(树链剖分)的更多相关文章
- Hdu 5274 Dylans loves tree (树链剖分模板)
Hdu 5274 Dylans loves tree (树链剖分模板) 题目传送门 #include <queue> #include <cmath> #include < ...
- POJ3237 Tree 树链剖分 边权
POJ3237 Tree 树链剖分 边权 传送门:http://poj.org/problem?id=3237 题意: n个点的,n-1条边 修改单边边权 将a->b的边权取反 查询a-> ...
- Query on a tree——树链剖分整理
树链剖分整理 树链剖分就是把树拆成一系列链,然后用数据结构对链进行维护. 通常的剖分方法是轻重链剖分,所谓轻重链就是对于节点u的所有子结点v,size[v]最大的v与u的边是重边,其它边是轻边,其中s ...
- 【BZOJ-4353】Play with tree 树链剖分
4353: Play with tree Time Limit: 20 Sec Memory Limit: 256 MBSubmit: 31 Solved: 19[Submit][Status][ ...
- SPOJ Query on a tree 树链剖分 水题
You are given a tree (an acyclic undirected connected graph) with N nodes, and edges numbered 1, 2, ...
- poj 3237 Tree 树链剖分
题目链接:http://poj.org/problem?id=3237 You are given a tree with N nodes. The tree’s nodes are numbered ...
- Codeforces Round #200 (Div. 1) D Water Tree 树链剖分 or dfs序
Water Tree 给出一棵树,有三种操作: 1 x:把以x为子树的节点全部置为1 2 x:把x以及他的所有祖先全部置为0 3 x:询问节点x的值 分析: 昨晚看完题,马上想到直接树链剖分,在记录时 ...
- poj 3237 Tree 树链剖分+线段树
Description You are given a tree with N nodes. The tree’s nodes are numbered 1 through N and its edg ...
- Aizu 2450 Do use segment tree 树链剖分+线段树
Do use segment tree Time Limit: 1 Sec Memory Limit: 256 MB 题目连接 http://www.bnuoj.com/v3/problem_show ...
随机推荐
- SQL Server中的Merge关键字(转载)
简介 Merge关键字是一个神奇的DML关键字.它在SQL Server 2008被引入,它能将Insert,Update,Delete简单的并为一句.MSDN对于Merge的解释非常的短小精悍:”根 ...
- Maven学习笔记-03-Eclipse和Maven集成
本文使用 Eclipse 集成 Maven,并创一个基于 maven的web工程 一 环境版本信息 本文使用的版本信息如下: Eclipse Version: Mars.1 Release (4.5. ...
- select 下拉选中
<body> <select name="" id=""> <option value="">张三< ...
- ES6入门之let、cont
一.前提 解决ES5中只有全局作用域和函数作用域,没有块级作用域而带来的不合理的场景. let 基本用法 用法和var 一样,只是let声明的变量只有在let命令所在的代码块有效 { let a = ...
- Oracle数据库设置为归档模式的操作方法
Oracle归档模式非常非常重要!对于有些数据库刚装好后可能是非归档模式,这是很危险的!为了安全起见,一定要谨记:对于Oracle数据库,一定要设置为归档模式,尤其是生产库,只有这样才能实现数据库的有 ...
- Python 工程管理及 virtualenv 的迁移
virtualenv 是管理 python 工程的利器,它可以很好的帮你维护项目中的依赖,使用 virtualenv,还能保持 global 库的干净.不会被不同项目中的第三方库所污染. virtua ...
- 这是一个数学题牛客训练赛E
题目描述 https://www.nowcoder.net/acm/contest/78/E 已知有一个n+1个数的数列,对于给定的A0和An ,当i满足当1<=i<=n-1时有 ...
- Some questions after Reading 《移山之道》
很少见到用故事的形式来写技术书籍的,这是我看到的第一本,书写得比较有趣,看了之后也是有一定的收获. 作者在此书中旁征博引,引用的东西虽不能一个一个查询是否正确,但是每次读到时候,感觉一种现代的软件工 ...
- windows上tomcat8的安装及配置
提示:在安装tomcat之前,确定安装好jdk. 一.下载tomcat8 http://tomcat.apache.org/download-80.cgi 点击这个网址 根据自己电脑的才做系统版本安装 ...
- linux及安全第六周总结
进程控制块pcb——task_struct 操作系统三大功能: 进程管理(核心) 内存管理 文件系统 为了管理进程,内核必须对每个进程进行清晰的描述,进程描述符提供了内核所需了解的进程信息: 进程状态 ...