二叉树的直径
题目描述
给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点
题目分析
示例 :
给定二叉树
1
/ \
2 3
/ \
4 5
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。
两结点之间的路径长度是以它们之间边的数目表示。
思路(一)
(1)写出一个求树高度的函数height,每次把节点当作参数传入,求出此节点向下的高度。
(2)求以某个树为顶点,左边树高+右边树高+1(算出的直径长度,必然有顶点)
(3)遍历递归整棵树,求以每一个节点为顶点的直径长度,每次总返回最大的那个路径
(4)最终返回最大直径长度
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {TreeNode} root
* @return {number}
*/
var diameterOfBinaryTree = function(root) {
if(!root) return 0;
let ans=height(root.left)+height(root.right); //将左右子树分别传到计算深度的函数中计算得到最长左子树+最长右子树+1
return Math.max(ans,diameterOfBinaryTree(root.left),diameterOfBinaryTree(root.right)); //遍历将每个节点都传入函数中,并与ans进行比较,返回最大值
};
//计算二叉树的深度
function height(node){
if(!node) return 0;
return 1+Math.max(height(node.left),height(node.right));
}
思路(二)
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {TreeNode} root
* @return {number}
*/
var diameterOfBinaryTree = function(root) {
// 默认为1是因为默认了根节点自身的路径长度
let ans = 1;
function depth(rootNode) {
if (!rootNode) {
// 如果不存在根节点,则深度为0
return 0;
}
// 递归,获取左子树的深度
let L = depth(rootNode.left);
// 递归,获取右子树的深度
let R = depth(rootNode.right);
/* 关键点1
L+R+1的公式是如何而来?
等同于:左子树深度(节点个数) + 右子树深度(节点个数) + 1个根节点
便是这株二叉树从最左侧叶子节点到最右侧叶子节点的最长路径
类似于平衡二叉树的最小值节点到最大值节点的最长路径
之所以+1是因为需要经过根节点
*/
// 获取该树的最长路径和现有最长路径中最大的那个
ans = Math.max(ans, L + R + 1);
/* 关键点2
已知根节点的左右子树的深度,
则,左右子树深度的最大值 + 1,
便是以根节点为数的最大深度*/
return Math.max(L, R) + 1;
}
depth(root);
// 由于depth函数中已经默认加上数节点的自身根节点路径了,故此处需减1
return ans - 1;
};