题目链接
第二题(卡码网周赛第二十六期(23年阿里淘天笔试真题))
题目描述
小红有一棵传送树,树上有 n (1 <= n <= 10^5) 个节点,编号为 1 到 n,其中 1 号节点为根节点。
每个节点都有一个传送门,传送门只可以将小红传送到子树中除了自己的其他节点中编号最小的节点。
小红想知道,经过若干次传送门直到叶子节点为止,可以到达的节点数量是多少。
输入
一行一个整数 n,表示树上的节点数量接下来 n - 1 行,每行两个整数 u, v,表示 u 号节点和 v 号节点之间有一条边(u 指向 v)。
输出
输出一行,n 个整数,分别对树上 n 个节点都需要计算答案,第 i 数字表示小红从第 i 个节点出发,经过若干次传送门到达叶子节点为止,可以到达的节点数量是多少。
样例输入
5
1 4
4 2
4 3
3 5
样例输出
2 1 2 2 1
样例解释
题解1(C++版本)
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
const int INF = 1e9;
int n, ans[N];
vector<int> edge[N];
// 树并一定是二叉树
int dfs(int u){ // 表示以u为根节点,传送门可以到达的编号最小的节点编号为dfs(u)
int sz = edge[u].size();
if(sz == 0) {
ans[u] = 1; // 顶点u为叶子节点
return u;
}
int minChildVal = INF;
for(int i = 0, v; i < sz; i++){ // 遍历当前节点的所有孩子节点
v = edge[u][i];
minChildVal = min(minChildVal, dfs(v));
}
ans[u] = ans[minChildVal] + 1;
return min(u, minChildVal);
}
int main(){
scanf("%d", &n);
for(int i = 1, u, v; i < n; i++) {
scanf("%d%d", &u, &v);
edge[u].push_back(v);
}
dfs(1);
for(int i = 1; i <= n; i++){
if(i > 1) printf(" ");
printf("%d", ans[i]);
}
return 0;
}