力扣题目地址:https://leetcode-cn.com/problems/diameter-of-binary-tree/
首先看题目:
给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过根结点。
示例:
示例 :
给定二叉树
1
/ \
2 3
/ \
4 5
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。
注意:两结点之间的路径长度是以它们之间边的数目表示。
解决思路:
二叉树的题目,第一时间想到的就是递归算法。首先,根据题目要求,我们要取二叉树中的最长路径,其实也就是该二叉树最长的节点链。既然是最长节点链,中间肯定有一个节点是在转折点上的,也就是这条最长节点链maxNodes的根节点。而我们的最长节点链maxNodes是由这个根节点的最长左子链和最长右子链拼接而成的,我们的路径长度就是maxNodes-1。我们假设最长左子链节点数为left,最长右子链节点数为right;那么我们的最长路劲节点数就是left+right+1。
代码实现:
/**
* 求二叉树路径最大长度
* https://leetcode-cn.com/problems/diameter-of-binary-tree/
* @param root
* @author Geyuxuan 2020-03-10 21:52:18
* @return int
*/
//设置全局变量,记录最达节点数
//设置为0的话,放入一个空树,返回为-1,所以设置为1
int maxNodes= 1;
public int diameterOfBinaryTree(TreeNode root) {
//递归得出最大节点数(算出最长路径)
depth(root);
//最大节点数减一
return maxNodes-1;
}
/**
* 递归算法求当前节点的最大树深度
* @param treeNode
* @author Geyuxuan 2020-03-10 23:33:27
* @return int
*/
private int depth(TreeNode treeNode){
//递归结束条件 空节点
if(treeNode == null){
return 0;
}
//计算左最长子链
int left = depth(treeNode.left);
//计算右最长子链
int right = depth(treeNode.right);
//计算当前节点的最大节点数
maxNodes= Math.max(maxNodes,left+right+1);
//返回当前节点的最大树深度
return Math.max(left,right)+1;
}
static class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
不忘初心,砥砺前行。