传送门;Computer Network
题目大意
这个题目主要是输入理解了好长时间,不安正常的顺序出牌呀!!
第一行输入的是点的数量,下面的N-1行(从第二行开始),第一个数和第几行这个数相连,也就是和i相连。第二个数是权值!
解题思路
解题思路,这个和求树的直径是异曲同工的!第一个DFS中的dp[i][0]表示第i个节点到子树的最大距离,dp[i][1]表示第i个节点到子树的次长距离!然后第二个DFS中的dp[i][0]表示第i个节点到整个树的最长距离,dp[i][1]表示第i个节点到整个树的次长距离!
AC代码
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN = 10000 + 5;
const int INF = 0x3f3f3f3f;
int dp[MAXN][3],head[MAXN],e;
struct Edge{
int v,nxt,w;
}E[MAXN<<2];
void init()
{
e = 0;
memset(head,-1,sizeof head);
}
void add(int u,int v,int w)
{
E[e].v = v;
E[e].w = w;
E[e].nxt = head[u];
head[u] = e++;
}
void DFS(int u,int pre)
{
for(int i=head[u];~i;i=E[i].nxt)
{
int v = E[i].v;
if(v==pre)continue;
DFS(v,u);
int tmp = dp[v][0]+E[i].w;
if(tmp>dp[u][0]) swap(tmp,dp[u][0]);
if(tmp>dp[u][1]) dp[u][1] = tmp;
}
}
void _DFS(int u,int pre)
{
for(int i=head[u];~i;i=E[i].nxt)
{
int v = E[i].v,w=E[i].w;
if(v==pre) continue;
int tmp;
if(dp[v][0]+w==dp[u][0]) tmp = w+dp[u][1];
else tmp = w+dp[u][0];
if(tmp>dp[v][0]) swap(tmp,dp[v][0]);
if(tmp>dp[v][1]) dp[v][1] = tmp;
_DFS(v,u);
}
}
int main()
{
int N;
int u,w;
while(~scanf("%d",&N))
{
init();
for(int i=2;i<=N;i++){
scanf("%d%d",&u,&w);
add(i,u,w);add(u,i,w);
}
DFS(1,-1);
_DFS(1,-1);
for(int i=1;i<=N;i++) printf("%d\n",dp[i][0]);
}
return 0;
}