leetcode-java-二叉树的直径

题目描述

给定一棵二叉树，你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。

示例 :
给定二叉树
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) 。

剖析

运用递归套路解决问题时，也要灵活化简问题。