首页 > 代码库 > 【树形DP】 HDU 2196 Computer
【树形DP】 HDU 2196 Computer
题意:求节点间的最大距离
先DFS一次 记录下 每一节点的子树下的最大距离(DP[ u ] [ 0 ])和第二大距离(DP[ u ] [ 1 ])
用DP[ v ] [ 2 ] 表示由v的父节点来的最大距离
再取DP[ u ] [ 0 ] 与 DP[ u ][ 2 ] 的最值
#include <cstdio> #include <cstring> #include <cstdlib> #include <string> #include <iostream> #include <algorithm> #include <sstream> #include <cmath> using namespace std; #include <queue> #include <stack> #include <vector> #include <deque> #include <set> #include <map> #include <time.h>; #define cler(arr, val) memset(arr, val, sizeof(arr)) #define FOR(i,a,b) for(int i=a;i<=b;i++) #define IN freopen ("in.txt" , "r" , stdin); #define OUT freopen ("out.txt" , "w" , stdout); typedef long long LL; const int MAXN = 10014; const int MAXM = 20014; const int INF = 0x3f3f3f3f; const int mod = 1000000007; struct node { int v,next; LL val; } edge[MAXM]; int head[MAXM],tol; LL dp[MAXN][3]; bool vis[MAXN]; void init() { cler(head,-1); tol=0; } void addedge(int u,int v,LL val) { edge[tol].v=v,edge[tol].val=val,edge[tol].next=head[u]; head[u]=tol++; edge[tol].v=u,edge[tol].val=val,edge[tol].next=head[v]; head[v]=tol++; } void dfs1(int u) { if(vis[u]) return ; vis[u]=true; for(int i=head[u]; ~i; i=edge[i].next) { int v=edge[i].v; if(!vis[v]) { dfs1(v); dp[u][1]=max(dp[u][1],dp[v][0]+edge[i].val); if(dp[u][1]>dp[u][0])swap(dp[u][1],dp[u][0]); } } } void dfs2(int u) { if(vis[u]) return ; vis[u]=true; for(int i=head[u]; ~i; i=edge[i].next) { int v=edge[i].v,val=edge[i].val; if(!vis[v]) { if(dp[u][0]>dp[v][0]+val)//dp[u][0]不是由dp[v][0]+val而来的 dp[v][2]=max(dp[v][2],max(dp[u][0]+val,dp[u][2]+val));//所以从非v子树来的更长 else //dp[u][0]由dp[v][0]+val而来的 dp[v][2]=max(dp[v][2],max(dp[u][1]+val,dp[u][2]+val));//判断非v子树来的哪个长 dfs2(v); } } } int main() { #ifndef ONLINE_JUDGE freopen("in.txt", "r", stdin); #endif int n; while(~scanf("%d",&n)) { init(); for(int i=2; i<=n; i++) { int a; LL b; scanf("%d %I64d",&a ,&b ); addedge(i,a,b); } cler(vis,false); cler(dp,0); dfs1(1); cler(vis,false); dfs2(1); for(int i=1;i<=n;i++) printf("%I64d\n",max(dp[i][2],dp[i][0])); } }
【树形DP】 HDU 2196 Computer
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。