qduoj 80 树结构重逢(树形DP)

题意：

给定一颗包含n个节点的无根树，并且把节点从1-n编号。

现在让你求对于每个点，求出所有点到这个点的距离的和(距离当然是最短距离了..)。

思路：跟HDU3899一毛一样的思路，注意sum要开long long。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn = 2e5+5;
vector<int> g[maxn];
ll cnt[maxn], sum[maxn];

void dfs(int x, int fa)
{
    cnt[x] = 1; sum[x] = 0;
    for(int i = 0; i < g[x].size(); i++)
    {
        int to = g[x][i];
        if(to == fa) continue;
        dfs(to, x);
        cnt[x] += cnt[to];
        sum[x] += (sum[to]+cnt[to]*1);
    }
}

void dfs1(int x, int fa)
{
    for(int i = 0; i < g[x].size(); i++)
    {
        int to = g[x][i];
        if(to == fa) continue;
        sum[to] = sum[to]+(sum[x]-sum[to]-cnt[to]*1)+(cnt[x]-cnt[to])*1;
        cnt[to] += (cnt[x]-cnt[to]);
        dfs1(to, x);
    }
}

int main(void)
{
    int n;
    while(cin >> n)
    {
        for(int i = 0; i < maxn; i++)
            g[i].clear();
        for(int i = 1; i < n; i++)
        {
            int u, v;
            scanf("%d%d", &u, &v);
            g[u].push_back(v);
            g[v].push_back(u);
        }
        dfs(1, 0);
        dfs1(1, 0);
        for(int i = 1; i <= n; i++)
            printf("%lld\n", sum[i]);
    }

    return 0;
}