HihoCoder 1742 树的权重 【DFS】

题目来源：https://hihocoder.com/problemset/problem/1742

★这题写的时候没找到树上的关系，后来清楚了些~ DFS不难，就难在关系上

思路：

我一开始的思路是 用带权并查集来写 ，但是貌似这个一次只能求两点之间的距离，那么复杂度肯定是o(n*n)的，可能是我太菜了还不会高级的并查集吧
后来翻博客发现是 DFS 通过树上的特殊关系来解决这个问题的，如下：
首先 找任意一个点作为根（这里是 1）然后遍历一遍树 求所有子节点到父节点的距离和 以及父节点拥有的子节点的个数，分别用tmp 、num数组表示 。易知，此时的 tmp[ 1 ] 就是 ans[ 1 ]，ans记录答案 关键是现在只能求出一个点，接下来我们来找关系（看下图）现在tmp[ 1 ] 表示的是所有点到1的距离 ,我们要求ans[ 2 ] 即 所有点到点2的距离，这个转移过程中你会发现 点2到 点2的子节点（包括2）的距离 和点1到 点2的子节点的距离 相比都少了1 这是因为点2的子节点到点1的路径中 必定经过蓝色那条线 而点2到 非点2子节点的距离 和点1到 非点2的子节点的距离 相比多了1 同上原因 那么关系就很明显了 ans[ 2 ] = tmp[ 1 ] - num[ 2 ] + (n - num[ 2 ] )

代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<map>
#include<stack>
#include<queue>
#include<set>
#include<string>
using namespace std;
const int maxn=1e5+5;
const int sz=1<<20;
const int mod=1e9+7;
typedef long long LL;
int n,m;
vector<int> v[maxn];
LL ans[maxn],tmp[maxn],num[maxn];
bool vis[maxn];
inline void read(int &x)
{
    char c; x=1;
    while((c=getchar())<'0'||c>'9') if(c=='-') x=-1;
    int res=c-'0';
    while((c=getchar())>='0'&&c<='9') res=res*10+c-'0';
    x*=res;
}
void dfs1(int k)
{
    vis[k]=1;
    num[k]=0;
    for(int i=0;i<v[k].size();i++){
        if(!vis[v[k][i]]){
            dfs1(v[k][i]);
            num[k]+=num[v[k][i]];
            tmp[k]+=tmp[v[k][i]]+num[v[k][i]];
        }
    }
    num[k]++;
}
void dfs2(int k)
{
    if(k==1) ans[k]=tmp[k];
    vis[k]=1;
    for(int i=0;i<v[k].size();i++){
        if(!vis[v[k][i]]){
            ans[v[k][i]]=ans[k]-2*num[v[k][i]]+n;
            dfs2(v[k][i]);
        }
    }
}
int main()
{
    read(n);
    for(int i=1;i<=n;i++){
        v[i].clear();
        vis[i]=0;
    }
    for(int i=1;i<n;i++){
        int a,b;
        read(a);read(b);
        v[a].push_back(b);
        v[b].push_back(a);
    }
    dfs1(1);
    memset(vis,0,sizeof vis);
    dfs2(1);
    for(int i=1;i<=n;i++)  printf("%lld\n", ans[i]);
    return 0;
}