程序员代码面试指南刷题--第三章.二叉树节点间的最大距离问题

题目描述
从二叉树的节点 A 出发，可以向上或者向下走，但沿途的节点只能经过一次，当到达节点 B 时，路径上的节点数叫作 A 到 B 的距离。
现在给出一棵二叉树，求整棵树上每对节点之间的最大距离。
输入描述:

第一行输入两个整数 n 和 root，n 表示二叉树的总节点个数，root 表示二叉树的根节点。
以下 n 行每行三个整数 fa，lch，rch，表示 fa 的左儿子为 lch，右儿子为 rch。(如果 lch 为 0 则表示 fa 没有左儿子，rch同理)
最后一行为节点 o1 和 o2。

输出描述:

输出一个整数表示答案。

示例1

输入

7 1
1 2 3
2 4 5
4 0 0
5 0 0
3 6 7
6 0 0
7 0 0

输出

5

解法一：树形dp问题

import java.io.*;
import java.util.*;
public class Main{
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        String[] info = br.readLine().trim().split(" ");
        int len = Integer.parseInt(info[0]);
        int r = Integer.parseInt(info[1]);
        TreeNode[] nodes = new TreeNode[len+1];
        for(int i=1;i<=len;i++){
            nodes[i] = new TreeNode(i);
        }
        createTree(nodes,br);
        TreeNode root = nodes[r];
        //不能只用左高加右高因为有可能是单支树
        ResultType res = findMax(root);
        System.out.println(res.maxDis);
    }
    public static ResultType findMax(TreeNode node){
        if(node==null) return new ResultType(0,0);
        ResultType left = findMax(node.left);
        ResultType right = findMax(node.right);
        int height = Math.max(left.height,right.height)+1;
        int maxDis = Math.max(left.height+right.height+1,Math.max(left.maxDis,right.maxDis));
        return new ResultType(maxDis,height);
    }
    public static void createTree(TreeNode[] nodes,BufferedReader br) throws Exception{
        for(int i=0;i<nodes.length-1;i++){
            String[] datas = br.readLine().trim().split(" ");
            int data = Integer.parseInt(datas[0]);
            int left = Integer.parseInt(datas[1]);
            int right = Integer.parseInt(datas[2]);
            if(left!=0){
                nodes[data].left = nodes[left];
                nodes[left].parent = nodes[data];
            }
            if(right!=0){
                nodes[data].right = nodes[right];
                nodes[right].parent = nodes[data];
            }
        }
        return ;
    }
}
class ResultType{
    int maxDis;
    int height;
    public ResultType(int maxDis,int height){
        this.maxDis = maxDis;
        this.height = height;
    }
}
class TreeNode{
    TreeNode left;
    TreeNode right;
    TreeNode parent;
    int val;
    public TreeNode(int val){
        this.val = val;
    }
}