501 距离和

// 501 距离和.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//

/*

http://oj.daimayuan.top/course/5/problem/224

有一棵 n个节点的树,节点编号从 1到 n。
请求出每个节点到其他所有节点的距离的和。

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

输入格式
第一行一个整数 n表示节点数。

接下来 n−1行,每行两个整数 x,y表示 x号节点和 y号节点之间有一条边。


数据保证读入的是一棵树。

输出格式
输出共 n行,第 i行一个整数表示 i号节点到其他所有节点的距离的和。

样例输入
5
1 2
1 5
2 3
2 4
样例输出
6
5
8
8
9
数据规模
对于所有数据,保证 2≤n≤105,1≤x,y≤n。
*/


#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <memory.h>

using namespace std;

const int N = 100010;
long long dp[N], v[N], cnt[N];
int h[N], e[N*2], ne[N*2], idx;
int n;

void add(int a, int b) {
    e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}

void down(int u, int fa) {
    for (int i = h[u]; i != -1; i = ne[i]) {
        int j = e[i];
        if (j == fa) continue;
        down(j, u);
        dp[u] += dp[j];
        cnt[u] += cnt[j];
    }
    dp[u] += cnt[u];
    cnt[u]++;
}

void up(int u, int fa) {
    if (fa != -1) {
        v[u] = v[fa] + dp[fa] - dp[u] - cnt[u] + n - cnt[u];
    }
    for (int i = h[u]; i != -1; i = ne[i]) {
        int j = e[i];
        if (j == fa) continue;
        up(j, u);
    }
}


int main() {
    memset(h, -1, sizeof h);
    cin >> n;
    for (int i = 1; i <= n - 1; i++) {
        int a, b; cin >> a >> b;
        add(a, b); add(b, a);
    }

    down(1, -1);
    up(1, -1); 

    for (int i = 1; i <= n; i++) {
        cout << dp[i] + v[i] << endl;
    }
    
    return 0;
}

posted on   itdef  阅读(8)  评论(0编辑  收藏  举报

相关博文:
阅读排行:
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 25岁的心里话
历史上的今天:
2020-10-10 LeetCode 142. 环形链表 II 链表遍历 哈希
2017-10-10 How-to Install VMware Tools on Debian Stretch 9 32/64bit Linux+GNU

导航

< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5

统计

点击右上角即可分享
微信分享提示