树的高度

题目描述

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

输入描述:

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

输出描述:

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

示例1

输入

5
0 1
0 2
1 3
1 4

输出

3

 

var readline = require('readline');

rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout
});

var inputs = [];
var num = 0;
rl.on('line', function(data) {
    if(num == 0){
        num = Number(data.trim());
        if (num == 1) {
            console.log(1);
        }
        num = num - 1;
    } else {
        inputs.push(data.trim());
        if (num == inputs.length) {
            // 处理
            var result = deal(inputs);

            // 输出结果
            console.log(result);

            // 清0
            inputs.length = 0;  //不可改动
            num = 0;    //不可改动
        }
    }
});


function deal(inputs) {
    var result = new Array(1000).fill(0);
    var cnt = new Array(1000).fill(0);
    var cut = [];
    for (var i = 0; i < num; i++) {    
        var arr = inputs[i].split(' ');
        var p = arr[0];
        var ch = arr[1];
        cnt[p]++;
        if (cnt[p] > 2) {
            result[ch] = -1000;
            continue;
        }
        result[ch] = result[p] + 1;
    }
    result = Math.max(...result);
    return result + 1;
}

前面一坨主要是node的输入，功能实现在deal函数中。

思路： 子节点的高度等于父节点高度 + 1；

坑： 这道题题目说的是二叉树，但是测试用例里面有多叉，要手动过滤多叉（我这里用的纯暴力，没有过滤，只是把高度赋值为一个很小的值 -1000）。提交多次之后发现一组数据过不了，然后翻看大佬们的评论，才发现测试里面有多叉。

emmm.....深刻体会到node.js的输入在A题的时候是多么的麻烦！