【BZOJ】 [Coci2015]Kamp-树形DP

博客介绍了如何使用树形动态规划解决Kamp聚会问题,即在树形结构中找到最小时间将K个人从不同点送到同一地点再返回原点。文章通过四遍深度优先搜索(DFS)或广度优先搜索(BFS)计算节点的最优路径,涉及实树、虚树的概念,并给出了详细的解题思路和代码实现。
摘要由CSDN通过智能技术生成

传送门:BZOJ3743Kamp


题意

     一颗树n个点,n-1条边,经过每条边都要花费一定的时间,任意两个点都是联通的。
     有K个人(分布在K个不同的点)要集中到一个点举行聚会。
     聚会结束后需要一辆车从举行聚会的这点出发,把这K个人分别送回去。
     请你回答,对于i=1~n,如果在第i个点举行聚会,司机最少需要多少时间把K个人都送回家。


输入

    第一行两个数,n,K。
    接下来n-1行,每行三个数,x,y,z表示x到y之间有一条需要花费z时间的边。
    接下来K行,每行一个数,表示K个人的分布。


输出

输出n个数,第i行的数表示:如果在第i个点举行聚会,司机需要的最少时间。


数据规模

K <= N <= 500000
1 <= x,y <= N, 1 <= z <= 1000000


题解

    我们任意以某个人的位置结点设为树根dfs构造一颗树。
    将所有人的位置结点之间连接,构造出实树(就这样叫),在实树中的结点的答案就是两倍的总链的长度减去以它为根在实树范围内的最长链的长度
    同样对于不在实树范围内的结点,我们可以肯定它是某一个实树内结点为根的子树中的一个结点。所以对于实数范围外的点,答案就是它到最近实树的位置结点的距离加上这个位置结点的答案
    剩下的就是代码实现的问题了。
    第一遍dfs求出这颗实树,F[x]指的是在实树内的以结点x为根的子树的边权之和的两倍(由上述结论,我们预先处理成两倍),就是向下走完这颗实树。
    第二遍dfs求出dis[x]在整个树范围内的结点x向上走(对于实树内的x,用它的父节点y的dis加上它们之间的边权和(F[y]-F[x]),对于实树外的,则加上x与y之间的边权),就是向上走完这颗实树。
    第三遍dfs求出ch[x][0],ch[x][1]在实树范围内的x向下走完(实树外的结点是不可能往下走的)实树的边权之和,ch[x][0]存的是最大值,ch[x][1]存的是次大值(当x在实树中,我们要比较它往上还是往下走答案更小,但有可能它在父节点往下走的最大路径上,这样就无法求出父节点不经过x结点的最大路径来更新答案了,所以我们要存一个次大值,而且要用每个子结点来更新这个次大值,以保证结点x不同时在最大值和次大值的链上(好像不是很有道理))。
    第四遍dfs就算整棵树范围内的结点往上走完实树的边权之和,d[x]。
    这样处理完之后,我们就可以得到想要的值啦。
    对于实树内:
     ans[x]=F[x]+dis[x]max(d[x],ch[x][0]) a

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
题目描述 有一个 $n$ 个点的棋盘,每个点上有一个数字 $a_i$,你需要从 $(1,1)$ 走到 $(n,n)$,每次只能往右或往下走,每个格子只能经过一次,路径上的数字和为 $S$。定义一个点 $(x,y)$ 的权值为 $a_x+a_y$,求所有满足条件的路径中,所有点的权值和的最小值。 输入格式 第一行一个整数 $n$。 接下来 $n$ 行,每行 $n$ 个整数,表示棋盘上每个点的数字。 输出格式 输出一个整数,表示所有满足条件的路径中,所有点的权值和的最小值。 数据范围 $1\leq n\leq 300$ 输入样例 3 1 2 3 4 5 6 7 8 9 输出样例 25 算法1 (树形dp) $O(n^3)$ 我们可以先将所有点的权值求出来,然后将其看作是一个有权值的图,问题就转化为了在这个图中求从 $(1,1)$ 到 $(n,n)$ 的所有路径中,所有点的权值和的最小值。 我们可以使用树形dp来解决这个问题,具体来说,我们可以将这个图看作是一棵树,每个点的父节点是它的前驱或者后继,然后我们从根节点开始,依次向下遍历,对于每个节点,我们可以考虑它的两个儿子,如果它的两个儿子都被遍历过了,那么我们就可以计算出从它的左儿子到它的右儿子的路径中,所有点的权值和的最小值,然后再将这个值加上当前节点的权值,就可以得到从根节点到当前节点的路径中,所有点的权值和的最小值。 时间复杂度 树形dp的时间复杂度是 $O(n^3)$。 C++ 代码 算法2 (动态规划) $O(n^3)$ 我们可以使用动态规划来解决这个问题,具体来说,我们可以定义 $f(i,j,s)$ 表示从 $(1,1)$ 到 $(i,j)$ 的所有路径中,所有点的权值和为 $s$ 的最小值,那么我们就可以得到如下的状态转移方程: $$ f(i,j,s)=\min\{f(i-1,j,s-a_{i,j}),f(i,j-1,s-a_{i,j})\} $$ 其中 $a_{i,j}$ 表示点 $(i,j)$ 的权值。 时间复杂度 动态规划的时间复杂度是 $O(n^3)$。 C++ 代码
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值