题目描述:
现在有一棵合法的二叉树,树的节点都是用数字表示,现在给定这棵树上所有的父子关系,求这棵树的高度
输入:输入的第一行表示节点的个数n(1<=n<=1000,节点的编号为0到n-1)组成,
下面是n-1行,每行有两个整数,第一个数表示父节点的编号,第二个数表示子节点的编号
输出树的高度,为一个整数
样例输入:
5
0 1
0 2
1 3
1 4
现在有一棵合法的二叉树,树的节点都是用数字表示,现在给定这棵树上所有的父子关系,求这棵树的高度
输入:输入的第一行表示节点的个数n(1<=n<=1000,节点的编号为0到n-1)组成,
下面是n-1行,每行有两个整数,第一个数表示父节点的编号,第二个数表示子节点的编号
输出树的高度,为一个整数
样例输入:
5
0 1
0 2
1 3
1 4
样例输出
3
思路:先找到根节点,再递归寻找其左子树和又子树的长度,高度为左右子树中最大高度+1。
代码如下:
#include <iostream>
#include <vector>
#include <set>
#include <cstdio>
using namespace std;
int height(int root, vector<int> &lch, vector<int> &rch) {
int n = lch.size();//n为左子树的长度
if (root < 0 || root >= n) { return -1; }//若根节点root<0或root>n不合法返回-1
if (lch[root] < 0 && rch[root] < 0) { return 1; }//如果无左右子树则返回1
if (lch[root] < 0 || rch[root] < 0) {//如果只有左或右子树
if (lch[root] >= 0) { return 1+height(lch[root], lch, rch); }//有左子树 返回长度
if (rch[root] >= 0) { return 1+height(rch[root], lch, rch); }//有右子树 返回长度
}
return max(height(lch[root], lch, rch), height(rch[root], lch, rch)) + 1;//都有则返回长的
}
int main() {
int n;
while (1 == scanf("%d", &n)) {
vector<int> lch(n, -1), rch(n, -1);
set<int> ps, cs;
for (int i = 0; i < n - 1; ++i) {
int p, c;
scanf("%d %d", &p, &c);
ps.insert(p); cs.insert(c);//ps存父节点,cs存子节点
if (lch[p] < 0) { lch[p] = c; continue; }//给父节点的左子树赋值
if (rch[p] < 0) { rch[p] = c; continue; }//存在左节点,则给又子树赋值
}
int root = -1;
for (set<int>::iterator it = ps.begin(); it!= ps.end(); it++) {//遍历ps
if (!cs.count(*it)) {//根节点为父节点在子节点中未出现的元素
root = *it;
break;
}
}
cout << height(root, lch, rch) << "\n";
}
}