二叉树的高度

题目描述：
现在有一棵合法的二叉树，树的节点都是用数字表示，现在给定这棵树上所有的父子关系，求这棵树的高度
输入：输入的第一行表示节点的个数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";
    }
}