Daimayuan #224. 距离和——树形DP

题意

有一棵 nn 个点的树，请求出每个点到其他所有点的距离的和。

定义两个点的距离为它们的简单路径上经过了多少条边。

样例输入

5
1 2
1 5
2 3
2 4

样例输出

6
5
8
8
9

数据规模

对于所有数据，保证 2≤n≤105,1≤x,y≤n2≤n≤105,1≤x,y≤n。

题意：一个树，求每个点到其他点的距离和

题目链接：  ​​​​​​​#224. 距离和

思路

第一次写csdn上没有题解的题，实际上知乎有题解的，但是大佬写的实在是太长了，能看懂的话可以试试换根树形动态规划问题 - 知乎

首先要知道一个点的子树的结点数量（包括自己）dp[i]，以及结点1的答案，通过结点1的答案和dp[i]来推出剩余点的答案

void dfs(int d,int pre)
{
    dp[d]=1;            //所有点初始为1
    h[d]=h[pre]+1;
    ans[1]+=h[d];           //结点1的答案
    int l=v[d].size();
    fo(0,l-1){
        int now=v[d][i];
        if(pre!=now){
            dfs(now,d);
            dp[d]+=dp[now];         //加上子的子树数
        }
    }
}

这是因为统计点从父节点到子节点，变得近的要减，变得远要加，变近的就是dp[子节点]个，离远的就是n-dp[子节点]个。

所以 子节点的答案 = 父结点答案 + (n-dp[now]) - dp[now] = ans[d]+  (n - 2*dp[now])

注意的是该代码要放dfs上面，因为要保证ans[d]是已经得出的，回溯的话ans[d]还没有处理

void qdfs(int d,int pre)
{
    int l=v[d].size();
    fo(0,l-1){
        int now=v[d][i];
        if(pre!=now){
            ans[now]=ans[d]+(n-2*dp[now]);          //(n-dp[now])-dp[now]
            qdfs(now,d);
        }
    }
}

 代码

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<stack>
#include<cmath>
#include<vector>
#include<map>
using namespace std;
#define fo(a,b) for(int i=a;i<=b;i++)
#define inf 0x3f3f3f3f
#define ll long long
#define M 100005
int n,x,y;
vector<int>v[M];
int dp[M],h[M];
ll ans[M];
void dfs(int d,int pre)
{
    dp[d]=1;            //所有点初始为1
    h[d]=h[pre]+1;
    ans[1]+=h[d];           //结点1的答案
    int l=v[d].size();
    fo(0,l-1){
        int now=v[d][i];
        if(pre!=now){
            dfs(now,d);
            dp[d]+=dp[now];         //加上子的子树数
        }
    }
}
void qdfs(int d,int pre)
{
    int l=v[d].size();
    fo(0,l-1){
        int now=v[d][i];
        if(pre!=now){
            ans[now]=ans[d]+(n-2*dp[now]);          //(n-dp[now])-dp[now]
            qdfs(now,d);
        }
    }
}
int main()
{
    scanf("%d",&n);
    fo(1,n-1){
        scanf("%d%d",&x,&y);
        v[x].push_back(y);
        v[y].push_back(x);
    }
    h[0]=-1;
    dfs(1,0);
    qdfs(1,0);
    fo(1,n) printf("%lld\n",ans[i]);
	return 0;
}

两次dfs应该没有办法只跑一遍。