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;
}
作 者: itdef
欢迎转帖 请保持文本完整并注明出处
技术博客 http://www.cnblogs.com/itdef/
B站算法视频题解
https://space.bilibili.com/18508846
qq 151435887
gitee https://gitee.com/def/
欢迎c c++ 算法爱好者 windows驱动爱好者 服务器程序员沟通交流
如果觉得不错,欢迎点赞,你的鼓励就是我的动力
欢迎转帖 请保持文本完整并注明出处
技术博客 http://www.cnblogs.com/itdef/
B站算法视频题解
https://space.bilibili.com/18508846
qq 151435887
gitee https://gitee.com/def/
欢迎c c++ 算法爱好者 windows驱动爱好者 服务器程序员沟通交流
如果觉得不错,欢迎点赞,你的鼓励就是我的动力


【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 分享一个免费、快速、无限量使用的满血 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