有一棵二叉树,树上每个点标有权值,权值各不相同,请设计一个算法算出权值最大的叶节点到权值最小的叶节点的距离。二叉树每条边的距离为1,一个节点经过多少条边到达另一个节点为这两个节点之间的距离。
给定二叉树的根节点root,请返回所求距离。
import java.util.*;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}*/
public class Tree {
public int getDis(TreeNode root) {
// write code here
if(root == null||(root.left == null&&root.right == null))
return 0;
TreeNode[] tn=new TreeNode[2];
int[] max={Integer.MIN_VALUE};
int[] min={Integer.MAX_VALUE};
findNode(root,tn,max,min);
TreeNode maxNode=tn[0];
TreeNode minNode=tn[1];
if(maxNode == minNode){
return 0;
}
TreeNode lca=LCA(root,maxNode,minNode);
int[] count=new int[1];
distance(lca,maxNode,0,count);
int maxpath=count[0];
distance(lca,minNode,0,count);
int minpath=count[0];
return maxpath+minpath;
}
//找到最大最小叶节点
public void findNode(TreeNode root,TreeNode[] tn,int[] max,int[] min)
{
if(root == null){
return;
}
if(root.left == null&&root.right == null){
if(root.val > max[0]){
tn[0]=root;
max[0]=root.val;
}
if(root.val < min[0]){
tn[1]=root;
min[0]=root.val;
}
return;
}
findNode(root.left,tn,max,min);
findNode(root.right,tn,max,min);
}
//找到这两个点的LCA
public TreeNode LCA(TreeNode root,TreeNode max,TreeNode min){
if(root == null||root == max||root == min){
return root;
}
TreeNode left=LCA(root.left,max,min);
TreeNode right=LCA(root.right,max,min);
if(left!=null&&right!=null){
return root;
}
if(left!=null){
return left;
}
if(right!=null){
return right;
}
return null;
}
//分别找到lca和两个点的距离
public boolean distance(TreeNode root,TreeNode node,int cur,int[] count){
if(root == null||node == null){
return false;
}
if(root == node){
count[0]=cur;
return true;
}
cur++;
boolean flag=distance(root.left,node,cur,count);
if(flag == false){
flag=distance(root.right,node,cur,count);
}
return flag;
}
}