xmuoj.宁宁与古榕树

描述

在厦门大学思明校区的新闻学院旁边，屹立着一棵巨大的古榕树，它是校园里树龄最长的一棵。这棵古树见证了校园的变迁，也承载了无数师生的思念和回忆。有一天，宁宁同学路过这颗古榕树，被它的庄严气息所吸引。站在树下，宁宁仰望着这棵树冠覆盖的天空，思绪万千。突然，她脑海中浮现出一个与树有关的问题：

对于一颗有 N 个节点的无根树，节点编号依次为 1, 2, 3 ... N，然后给出它的 N-1 条边，并定义所有的边长均为 1。对于树上每一个节点，求出它到其余节点的距离的和。（例如当 N=3 时，你需要计算 dis(1,2)+dis(1,3)，dis(2, 1)+dis(2,3) 和 dis(3,1)+dis(3,2)，其中 dis(x,y) 表示编号为 x 与编号为 y 的点之间的最短距离）

宁宁左思右想，仍找不出好的解决方法，于是她带着这个问题向你求助。

出题者为刘之流同学。

输入

第 1 行一个整数 N (2<=N<=2e5)，表示树的大小。

第 2 行到第 n 行每行都有两个整数，u 和 v (1 <= u < v <= N)，表示编号为 u 和编号为 v 的两个点之间有一条边。

输入的数据一定可以构建出一棵合法的树。

输出

输出

共 N 行，每行一个整数 Si，表示编号为 i 的点到其余所有点的距离的和。（因答案较大，在 C/C++ 中你可能需要使用 long long 计算和存储答案）

#include <iostream>
#include <vector>
using namespace std;

const int MAXN = 200005;  // 处理最多200000个节点
vector<int> adj[MAXN];     // 邻接表存储树
long long subtree_size[MAXN];  // 存储每个节点的子树大小
long long dist_sum[MAXN];  // 存储每个节点到所有其他节点的距离和

// 第一次DFS计算子树大小和从根到所有其他节点的距离和
void dfs1(int node, int parent) {
    subtree_size[node] = 1;  // 包括自身
    for (int neighbor : adj[node]) {
        if (neighbor == parent) continue;  // 跳过父节点
        dfs1(neighbor, node);
        subtree_size[node] += subtree_size[neighbor];
        dist_sum[node] += dist_sum[neighbor] + subtree_size[neighbor];
    }
}

// 第二次DFS用于计算所有节点的距离和
void dfs2(int node, int parent, int N) {
    for (int neighbor : adj[node]) {
        if (neighbor == parent) continue;  // 跳过父节点
        dist_sum[neighbor] = dist_sum[node] - subtree_size[neighbor] + (N - subtree_size[neighbor]);
        dfs2(neighbor, node, N);
    }
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);  // 加快C++的cin、cout流

    int N;  // 节点总数
    cin >> N;

    // 读入所有的边
    for (int i = 1; i < N; i++) {
        int u, v;
        cin >> u >> v;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }

    // 从节点1作为根开始第一次DFS
    dfs1(1, 0);
    // 第二次DFS计算所有节点的距离和
    dfs2(1, 0, N);

    // 输出所有节点到其他节点的距离和
    for (int i = 1; i <= N; i++) {
        cout << dist_sum[i] << "\n";
    }

    return 0;
}