首页 > 代码库 > Hdu 2196 Computer (树形dp)
Hdu 2196 Computer (树形dp)
题目大意:
求出每一个结点距离叶子节点的最大距离。
思路分析:
假设1为树的根。
通过一遍dfs 可以求出每个节点到儿子节点上的叶子节点的最远距离。
然后我们要做的就是如果更新答案。
问题就是如果我们开始求的最大值本来就是一个儿子更新的,如果再深搜的话又会造成重复更新。
所以就记录两个值,最大值和次大值。
然后不断更新这两个值。
#include <cstdio> #include <iostream> #include <cstring> #include <algorithm> #define maxn 10005 using namespace std; int head[maxn]; int to[maxn<<1]; int next[maxn<<1]; int len[maxn<<1]; int tot; void init() { tot=0; memset(head,0,sizeof head); } void addedge(int u,int v,int w) { tot++; next[tot]=head[u]; to[tot]=v; head[u]=tot; len[tot]=w; } bool vis[maxn]; int maxval[maxn],nex[maxn],maxid[maxn],nexid[maxn]; //最大值 次大值 更新最大值的id 更新次大值的id void dfs(int x) { maxval[x]=nex[x]=0; for(int p = head[x] ;p; p = next[p]) { int v = to[p]; if(!vis[v]) { vis[v]=true; dfs(v); if(nex[x]<maxval[v]+len[p]) { nex[x]=maxval[v]+len[p]; nexid[x]=v; if(nex[x]>maxval[x]) { swap(nex[x],maxval[x]); swap(nexid[x],maxid[x]); } } } } } void update(int x,int pre) { for(int p = head[x];p;p=next[p]) { int v=to[p]; if(v==pre)continue; if(v==maxid[x]) { if(nex[x]+len[p]>nex[v]) { nex[v]=nex[x]+len[p]; nexid[v]=x; if(nex[v]>maxval[v]) { swap(nex[v],maxval[v]); swap(nexid[v],maxid[v]); } } } else { if(maxval[x]+len[p]>nex[v]) { nex[v]=maxval[x]+len[p]; nexid[v]=x; if(nex[v]>maxval[v]) { swap(nex[v],maxval[v]); swap(nexid[v],maxid[v]); } } } update(v,x); } } int main() { int n; while(scanf("%d",&n)!=EOF) { init(); for(int i=2;i<=n;i++) { int l,r; scanf("%d%d",&l,&r); addedge(i,l,r); addedge(l,i,r); } memset(vis,false,sizeof vis); vis[1]=true; dfs(1); update(1,-1); for(int i=1;i<=n;i++) printf("%d\n",maxval[i]); } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。