题目描述
给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。
示例 :
给定二叉树
1
/
2 3
/ \
4 5
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。
注意:两结点之间的路径长度是以它们之间边的数目表示。
我的解答
思路:递归
树的高度:Math.max(L,R)+1;
树的最大长度:Math.max(ans,L+R+1);
public static class ReturnType{
public int maxDistance;
public int h;
public ReturnType(int m, int h) {
this.maxDistance = m;;
this.h = h;
}
}
public static ReturnType process(Node head) {
if(head == null) {
return new ReturnType(0,0);
}
ReturnType leftReturnType = process(head.left);
ReturnType rightReturnType = process(head.right);
int includeHeadDistance = leftReturnType.h + 1 + rightReturnType.h;
int p1 = leftReturnType.maxDistance;
int p2 = rightReturnType.maxDistance;
int resultDistance=Math.max(Math.max(p1,p2),includeHeadDistance); //记录最大长度
int hitself = Math.max(leftReturnType.h, leftReturnType.h) + 1; //记录深度
return new ReturnType(resultDistance, hitself);
}
public static int maxDistance(Node head) {
return process(head).maxDistance;
}
官方解答
没有在递归中返回最大长度,而是用一个变量记录最大值,不保留中间值。
class Solution {
int res = 1;
public int diameterOfBinaryTree(TreeNode root) {
depth(root);
return res-1; //注意减一,因为题目问的是边数。
}
public int depth(TreeNode root){
if(root==null){
return 0; //叶子结点深度为1,注意base case是null时返回0
}
int L = depth(root.left);
int R = depth(root.right);
res = Math.max(res,L+R+1);
return Math.max(L,R)+1;
}
}
复杂度分析
时间复杂度:O(N),其中 N 为二叉树的节点数,即遍历一棵二叉树的时间复杂度,每个结点只被访问一次。
空间复杂度:O(Height),其中 Height为二叉树的高度。由于递归函数在递归过程中需要为每一层递归函数分配栈空间,所以这里需要额外的空间且该空间取决于递归的深度,而递归的深度显然为二叉树的高度,并且每次递归调用的函数里又只用了常数个变量,所以所需空间复杂度为 O(Height) 。
剖析
运用递归套路解决问题时,也要灵活化简问题。