《算法笔记》9.3小节——数据结构专题(2)-＞树的遍历 问题 B: 树的高度

题目描述

一棵树有n个节点，其中1号节点为根节点。

输入

第一行是整数n，表示节点数

后面若干行，每行两个整数a b，表示b是a的子节点。

输出

求这棵树的高度（根节点为第1层）

样例输入

5
1 2
1 3
3 4
3 5

样例输出

3

分析： 从根节点开始，向当前节点的孩子设置高度，直到所有的孩子都有高度之后，输出高度的最大值。

#include    <algorithm>
#include     <iostream>
#include      <cstdlib>
#include      <cstring>
#include       <string>
#include       <vector>
#include       <cstdio>
#include        <queue>
#include        <stack>
#include        <ctime>
#include        <cmath>
#include          <map>
#include          <set>
#include<unordered_map>
#define INF 0x3f3f3f3f
#define db1(x) cout<<#x<<"="<<(x)<<endl
#define db2(x,y) cout<<#x<<"="<<(x)<<", "<<#y<<"="<<(y)<<endl
#define db3(x,y,z) cout<<#x<<"="<<(x)<<", "<<#y<<"="<<(y)<<", "<<#z<<"="<<(z)<<endl
#define db4(x,y,z,a) cout<<#x<<"="<<(x)<<", "<<#y<<"="<<(y)<<", "<<#z<<"="<<(z)<<", "<<#a<<"="<<(a)<<endl
#define db5(x,y,z,a,r) cout<<#x<<"="<<(x)<<", "<<#y<<"="<<(y)<<", "<<#z<<"="<<(z)<<", "<<#a<<"="<<(a)<<", "<<#r<<"="<<(r)<<endl
using namespace std;

int main(void)
{
    #ifdef test
    freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    clock_t start=clock();
    #endif //test

    int n,a,b,ans=1;scanf("%d",&n);
    vector<int>num[n+1];
    int depth[n+5]={0};depth[1]=1;
    while(~scanf("%d%d",&a,&b))
    {
        num[a].push_back(b);
    }
    queue<int>que;que.push(1);
    while(!que.empty())
    {
        int index=que.front();que.pop();
        int l=num[index].size();
        for(int i=0;i<l;++i)
        {
            depth[num[index][i]]=depth[index]+1;
            ans=max(ans,depth[num[index][i]]);
            que.push(num[index][i]);
        }
    }
    printf("%d\n",ans);


    #ifdef test
    clockid_t end=clock();
    double endtime=(double)(end-start)/CLOCKS_PER_SEC;
    printf("\n\n\n\n\n");
    cout<<"Total time:"<<endtime<<"s"<<endl;        //s为单位
    cout<<"Total time:"<<endtime*1000<<"ms"<<endl;    //ms为单位
    #endif //test
    return 0;
}