从二叉树的节点A出发,可以向上或者向下走,但沿途的节点只能经过一次,当到达节点B时,路径上的节点数叫作A到B的距离。对于给定的一棵二叉树,求整棵树上节点间的最大距离。
给定一个二叉树的头结点root,请返回最大距离。保证点数大于等于2小于等于500.
解析:
一个以h为头的树上,最大距离只可能来自以下三种情况:
1.h的左子树上的最大距离
2.h的右子树上的最大距离
3.h左子树上离h.left最远距离+h右子树上离h.right最远距离+1
这三个值中最大的距离就是最远距离,使用后序遍历较为容易。
代码如下:
import java.util.*;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}*/
public class LongestDistance {
int longest=0;
public int findLongest(TreeNode root) {
// write code here
getMaxDis(root);
return longest;
}
public int getMaxDis(TreeNode root){
if(root==null){
return 0;
}
int left=getMaxDis(root.left);
int right=getMaxDis(root.right);
longest=Math.max(longest,left+right+1);
return Math.max(left,right)+1; }
}