树的高度

树的高度

题目描述

现在有一棵合法的二叉树，树的节点都是用数字表示，现在给定这棵树上所有的父子关系，求这棵树的高度

输入描述:

输入的第一行表示节点的个数n（1 ≤ n ≤ 1000，节点的编号为0到n-1）组成，
下面是n-1行，每行有两个整数，第一个数表示父节点的编号，第二个数表示子节点的编号

输出描述:

输出树的高度，为一个整数

示例1

输入

5
0 1
0 2
1 3
1 4

输出

3

题意：略

解题思路：类似于求图的最大直径，由于不知道树的根，所以用了两遍BFS。这题还有个坑点就是数据里面给的不是二叉树，后面的节点需要去掉。

#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
#include<vector>
#include<queue>
#define maxn 1005

vector<int> v[maxn];
int n,ans,tmp;
int flag[maxn];

struct line
{
    int r,l;
} arr[maxn];
struct node
{
    int x,step;
    node() {}
    node(int a,int b)
    {
        x=a;
        step=b;
    }
};
void bfs(int s)
{
    ans=1;
    tmp=s;
    memset(flag,0,sizeof(flag));
    queue<node> q;
    q.push(node(s,1));
    flag[s]=1;
    while(!q.empty())
    {
        node now=q.front();
        q.pop();
        if(ans<now.step)
        {
            ans=now.step;
            tmp=now.x;
        }
        for(int i=0; i<v[now.x].size(); i++)
        {
            node next;
            next.x=v[now.x][i];
            next.step=now.step+1;
            if(flag[next.x]) continue;
            flag[next.x]=1;
            q.push(next);
        }
    }
}
int main()
{
//    freopen("in.txt","r",stdin);
    scanf("%d",&n);
    for(int i=0; i<n; i++)
        v[i].clear();
    for(int i=0; i<n-1; i++)
    {
        scanf("%d%d",&arr[i].l,&arr[i].r);
    }
    for(int i=0; i<n-1; i++)
    {
        if(v[arr[i].l].size()>=2) continue;
        v[arr[i].l].push_back(arr[i].r);
    }
    bfs(0);
    for(int i=0; i<n; i++)
        v[i].clear();
    for(int i=0; i<n-1; i++)
    {
        if(v[arr[i].r].size()>=2) continue;
        v[arr[i].r].push_back(arr[i].l);
    }
    bfs(tmp);
    printf("%d\n",ans);
    return 0;
}