51nod 1405 树的距离之和（树形dp）

最新推荐文章于 2021-05-19 19:59:57 发布

原创最新推荐文章于 2021-05-19 19:59:57 发布 · 224 阅读

1 ·

CC 4.0 BY-SA版权

51nod 同时被 2 个专栏收录

109 篇文章

订阅专栏

动态规划

20 篇文章

订阅专栏

本文介绍了一种解决树上两点间距离之和问题的有效算法。通过两次深度优先搜索（DFS），分别计算每个节点的子树大小及节点到树中所有其他节点的距离之和。适用于节点数量不超过10万的大规模树形结构。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1405 树的距离之和

基准时间限制：1 秒空间限制：131072 KB 分值: 40 难度：4级算法题

关注

给定一棵无根树，假设它有n个节点，节点编号从1到n, 求任意两点之间的距离（最短路径）之和。

Input

第一行包含一个正整数n (n <= 100000)，表示节点个数。
后面(n - 1)行，每行两个整数表示树的边。

Output

每行一个整数，第i(i = 1,2,...n)行表示所有节点到第i个点的距离之和。

Input示例

Output示例

#include<iostream>
#include<vector>
#include<string.h>
#include<cstdio>
#define ll long long
#define maxn 100005
using namespace std;
vector<int>v[maxn];
int vist[maxn];
ll dp[maxn];
int num[maxn];
int n;
int dfs(int u,int sum)
{
    dp[1]+=sum;
    num[u]=1;
    vist[u]=1;
    for(int i=0;i<v[u].size();i++)
    {
        int q=v[u][i];
        if(vist[q]==0)
        {
            dfs(q,sum+1);
            num[u]+=num[q];
        }
    }
}
void dfs_dp(int u)
{
    vist[u]=1;
    for(int i=0;i<v[u].size();i++)
    {
        int q=v[u][i];
        if(vist[q]==0)
        {
            dp[q]=dp[u]-num[q]+(n-num[q]);//如果向子树移去，所有的u子节点距离都会减一，而v的子节点的数会加1;
            //q为对应的u节点所以减去num【q】而加上总数减去num【q】
            dfs_dp(q);
        }
    }
}
int main()
{
    scanf("%d",&n);
    memset(dp,0,sizeof(dp));
    for(int i=1;i<=n-1;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        v[a].push_back(b);
        v[b].push_back(a);
    }
    memset(vist,0,sizeof(vist));
    dfs(1,0);
    memset(vist,0,sizeof(vist));
    dfs_dp(1);
    for(int i=1;i<=n;i++)
        cout<<dp[i]<<endl;
    return 0;
}