qduoj 树结构重逢 （两遍深搜）

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

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

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

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

样例输入1 复制
4
1 2
3 2
4 2
样例输出1
5
3
5
5
分析：
以1号节点为根建树。
第一遍深搜，记录一个点子节点到达这个节点的距离之和子节点的个数。在回溯的时候，当前节点的所有子节点和它的距离之和就是他直接子节点和值（其子节点距离之和）加上其子节点的数目。这样深搜完之后就可以得出根节点的结果。
第二遍深搜就是用根节点的答案推出子节点的答案。如果知道了父节点的答案，那么子节点的答案就是父节点的答案减去其对父节点的贡献，其实就是得出了除去以当前节点为根节点的子树之后父节点的答案，再用这个值加上除去这个子树的节点数。就是当前节点的答案了。
可以画画图，就显而易见了。

#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;

typedef long long ll;
struct node{
    ll ans, up, son;
} p[200000];
vector<int> a[200000];
int vis[200000];
int n, m;

void dfs(int k){
    vis[k] = 1;
    for(int i = 0; i < a[k].size(); i++){
        if(vis[a[k][i]] == 0){
            dfs(a[k][i]);
            p[k].son += p[a[k][i]].son + 1;
            p[k].up += p[a[k][i]].up + p[a[k][i]].son + 1;
        }
    }
}

void dfs1(int k, int fa){
    vis[k] = 1;
    if(k != 1){
        p[k].ans = p[fa].ans - (p[k].son + 1) + n - p[k].son - 1;
    }
    for(int i = 0; i < a[k].size(); i++){
        if(vis[a[k][i]] == 0){
            dfs1(a[k][i], k);
        }
    }
}

int main(){
    scanf("%d", &n);
    for(int i = 1; i < n; i++){
        int x, y;
        scanf("%d%d", &x, &y);
        a[x].push_back(y);
        a[y].push_back(x);
    }
    dfs(1);
    memset(vis, 0, sizeof(vis));
    p[1].ans = p[1].up;
    dfs1(1, -1);
    for(int i = 1; i <= n; i++){
        printf("%lld\n", p[i].ans);
    }
    return 0;
}