博客
关于我
hdu6567 Cotree (树形dp 树的重心)
阅读量:249 次
发布时间:2019-03-01

本文共 1711 字,大约阅读时间需要 5 分钟。

Problem Description

Avin has two trees which are not connected. He asks you to add an edge between them to make them connected while minimizing the function

∑ni=1∑nj=i+1dis(i,j)
, where dis(i,j) represents the number of edges of the path from i to j. He is happy with only the function value.

Input

The first line contains a number n (2<=n<=100000). In each of the following n−2 lines, there are two numbers u and v, meaning that there is an edge between u and v. The input is guaranteed to contain exactly two trees.

Output

Just print the minimum function value.

Sample Input

3

1 2

Sample Output

4

思路:

树中所有点到某个点的距离和中,到重心的距离和是最小的;

如果有两个重心,那么他们的距离和一样

两次dfs树形dp找到两颗树的重心

把两个重心连接起来
再dfs树形dp一次计算答案

计算树上任意两点的距离和,转化为计算每条边被经过的次数乘上边权

每条边被经过的次数等于这条边分隔开
两端点分别所在的连通块的大小相乘
最后一次dfs树形dp的过程中
假设u是v的父节点,
以v为根的子树大小为sz[v]
则连接他们的边被经过的次数为(n-sz[v])*sz[v]

code:

#include
#include
#include
#include
#include
#include
#include
using namespace std;#define ll long longconst int inf=1e9;const int maxm=1e5+5;int head[maxm],nt[maxm<<1],to[maxm<<1],cnt;int mark[maxm];int sz[maxm],son[maxm];int num;//第一个连通块的大小int size;//第二个连通块的大小int root1,root2;int n;ll ans;void init(){ memset(head,0,sizeof head); memset(mark,0,sizeof mark); cnt=1; num=0; ans=0;}void add(int x,int y){ cnt++;nt[cnt]=head[x];head[x]=cnt;to[cnt]=y;}void dfs(int x){ num++;//计算连通块大小 sz[x]=1; son[x]=0; mark[x]=1; for(int i=head[x];i;i=nt[i]){ int v=to[i]; if(mark[v])continue; dfs(v); sz[x]+=sz[v]; son[x]=max(son[x],sz[v]); }}void dfs2(int x){ sz[x]=1; son[x]=0; mark[x]=1; for(int i=head[x];i;i=nt[i]){ int v=to[i]; if(mark[v])continue; dfs2(v); sz[x]+=sz[v]; son[x]=max(son[x],sz[v]); } son[x]=max(son[x],size-sz[x]); if(son[x]

转载地址:http://bnzv.baihongyu.com/

你可能感兴趣的文章
MySQL
查看>>
mysql
查看>>
MTK Android 如何获取系统权限
查看>>
MySQL - 4种基本索引、聚簇索引和非聚索引、索引失效情况、SQL 优化
查看>>
MySQL - ERROR 1406
查看>>
mysql - 视图
查看>>
MySQL - 解读MySQL事务与锁机制
查看>>
MTTR、MTBF、MTTF的大白话理解
查看>>
mt_rand
查看>>
mysql -存储过程
查看>>
mysql /*! 50100 ... */ 条件编译
查看>>
mudbox卸载/完美解决安装失败/如何彻底卸载清除干净mudbox各种残留注册表和文件的方法...
查看>>
mysql 1264_关于mysql 出现 1264 Out of range value for column 错误的解决办法
查看>>
mysql 1593_Linux高可用(HA)之MySQL主从复制中出现1593错误码的低级错误
查看>>
mysql 5.6 修改端口_mysql5.6.24怎么修改端口号
查看>>
MySQL 8.0 恢复孤立文件每表ibd文件
查看>>
MySQL 8.0开始Group by不再排序
查看>>
mysql ansi nulls_SET ANSI_NULLS ON SET QUOTED_IDENTIFIER ON 什么意思
查看>>
multi swiper bug solution
查看>>
MySQL Binlog 日志监听与 Spring 集成实战
查看>>