根据父子关系建立二叉树

题目：

• 给定这棵树上所有的父子关系，求这棵树的高度。

样例：

5
0 1
0 2
1 3
1 4

思路：

• 将所有的节点保存在 HashMap 结构中，并根据节点之间的关系，建立二叉树。

代码：

import java.util.*;//树的节点class Node{    int val;    Node parent;    Node left;    Node right;    Node(int val){        this.val = val;    }}public class Main{    public static void main(String[] args){        Scanner in = new Scanner(System.in);        while(in.hasNextInt()){            int n = in.nextInt();            if(n==0 || n==1){                System.out.println(n);                continue;            }            int[][] nums = new int[n-1][2];            for(int i=0; i<n-1; i++){                Integer i1 = in.nextInt();                Integer i2 = in.nextInt();                nums[i][0] = i1;                nums[i][1] = i2;                            }            //=========================建立树的过程=========================            //将所有的节点保存在 Map 结构里面            HashMap<Integer, Node> map = new HashMap<Integer, Node>();            //遍历关系对，建立节点之间的联系            for(int i=0; i<n-1; i++){                //如果原来没有该节点，则建立新节点                if(!map.containsKey(nums[i][0])){                    Node node = new Node(nums[i][0]);                    map.put(nums[i][0], node);                }                if(!map.containsKey(nums[i][1])){                    Node node = new Node(nums[i][1]);                    map.put(nums[i][1], node);                }                //建立节点之间的联系                Node node0 = map.get(nums[i][0]);                Node node1 = map.get(nums[i][1]);                node1.parent = node0;   //注意此处要保存父节点                if(node0.left==null)                    node0.left = node1;                else                    node0.right = node1;            }                       //找出根节点：没有父节点的，即为根节点，得到建立后的树            Node head = null;            for(int i=0; i<n; i++){                Node tempnode = map.get(i);                if(tempnode.parent == null){                    head = tempnode;                    break;                }            }            //=========================树建立完成=========================            //根据树，采用递归算法，得到树的高度            int result = MyFunction(head);            System.out.println(result);         }           }    //计算树的高度    public static int MyFunction(Node head){                if(head==null)            return 0;        if(head.left == null)            return MyFunction(head.right)+1;        else if(head.right == null)            return MyFunction(head.left)+1;             else            return Math.max(MyFunction(head.left), MyFunction(head.right))+1;           }}

参考：
题目来源： http://exercise.acmcoder.com/online/online_judge_ques?ques_id=3329&konwledgeId=155