首页 > 代码库 > poj1849(求树的直径)
poj1849(求树的直径)
题目链接:http://poj.org/problem?id=1849
题意:有一颗n个结点的带权的无向树, 在s结点放两个机器人, 这两个机器人会把树的每条边都走一遍, 但是最后机器人不要求回到出发点. 问你两个机器人走的路总长之和的最小值是多少?
分析:如果从某点出发遍历完一棵树再回来,那么所有边都会走两遍,而从某点有两个机器人出发去遍历,因为不用回来,所以最后那两个人距离越远越好,可以从树的直径上某个点背道而驰,那么这段距离(树的直径)只走了一遍,其他的要走两遍,所以ans=sum*2-len(总边长的和*2-树的直径)。
#include <cstdio>#include <cstring>#include <cmath>#include <iostream>#include <algorithm>#include <queue>#include <cstdlib>#include <stack>#include <vector>#include <set>#include <map>#define LL long long#define mod 1000000007#define inf 0x3f3f3f3f#define N 100010#define clr(a) (memset(a,0,sizeof(a)))using namespace std;struct edge{ int v,w,next; edge(){} edge(int v,int w,int next):v(v),w(w),next(next){}}e[2*N];int head[N],tot,n,m,mx,s;void addedge(int u,int v,int w){ e[tot]=edge(v,w,head[u]); head[u]=tot++;}void dfs(int u,int fa,int len){ if(len>mx)mx=len,s=u; for(int i=head[u];~i;i=e[i].next) { int v=e[i].v,w=e[i].w; if(v==fa)continue; dfs(v,u,len+w); }}int main(){ int u,v,w,sum; while(scanf("%d%d",&n,&m)>0) { memset(head,-1,sizeof(head)); tot=0;sum=0; for(int i=1;i<n;i++) { scanf("%d%d%d",&u,&v,&w); addedge(u,v,w); addedge(v,u,w); sum+=w*2; } mx=0; dfs(m,-1,0); dfs(s,-1,0); printf("%d\n",sum-mx); }}
poj1849(求树的直径)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。