洛谷 B2170：树上结点深度 ← DFS + 链式前向星

​【题目来源】
https://www.luogu.com.cn/problem/B2170

【题目描述】
给定一棵包含 n 个结点的树，结点编号为 1∼n。我们将 1 号结点指定为这棵树的根。
定义根结点的深度为 1。对于任意非根结点 u，其深度定义为其父结点的深度加 1。
请你计算并输出这棵树中每个结点的深度。

【输入格式】
输入的第一行包含一个整数 n，表示树的结点个数。
接下来 n−1 行，每行包含两个整数 u,v，表示结点 u 和结点 v 之间存在一条无向边。保证输入的数据构成一棵树。​​​​​​​

【输出格式】
输出一行，包含 n 个整数。第 i 个整数表示编号为 i 的结点的深度。两个整数之间请用一个空格隔开。

【输入样例】
5
1 2
1 3
2 4
2 5​​​​​​​

【输出样例】
1 2 2 3 3

【数据范围】
对于 30% 的数据，满足 n≤100。
对于 60% 的数据，满足 n≤1000。
对于 100% 的数据，满足 1≤n≤500,000。保证输入的各条边能组成一棵含有 n 个结点的树。​​​​​​​

【算法分析】
● 链式前向星：https://blog.csdn.net/hnjzsyjyj/article/details/139369904
● 提升 cin/cout 速度：https://blog.csdn.net/hnjzsyjyj/article/details/143176072

【算法代码】

#include <bits/stdc++.h>
using namespace std;

const int N=5e5+5;
int e[N<<1],ne[N<<1],h[N],idx;
int dep[N];

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

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

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    memset(h,-1,sizeof h);
    int n;
    cin>>n;
    for(int i=1; i<n; i++) {
        int x,y;
        cin>>x>>y;
        add(x,y),add(y,x);
    }

    dfs(1,0);

    for(int i=1; i<=n; i++) {
        cout<<dep[i]<<" ";
    }

    return 0;
}

/*
in:
5
1 2
1 3
2 4
2 5

out:
1 2 2 3 3
*/

【参考文献】
https://blog.csdn.net/hnjzsyjyj/article/details/160746347
https://blog.csdn.net/hnjzsyjyj/article/details/152726091
https://blog.csdn.net/hnjzsyjyj/article/details/152729089

 

​