深度优先搜索
class Solution {
public int diameterOfBinaryTree(TreeNode root) {
if (root == null){
return 0;
}
/**
* 包含根节点的最长路径
*/
int containRoot = maxDepth(root.left) + maxDepth(root.right);
/**
* 左右子树的路径
*/
int leftDepth = diameterOfBinaryTree(root.left);
int rightDepth = diameterOfBinaryTree(root.right);
int maxChild = Math.max(leftDepth, rightDepth);
return Math.max(containRoot, maxChild);
}
public int maxDepth(TreeNode root){
if (root == null){
return 0;
}
return Math.max(maxDepth(root.left), maxDepth(root.right)) + 1;
}
}
/**
* 时间复杂度 O(n)
* 空间复杂度 O(logn)
*/
优化1——等效于求所有节点的最大深度
class Solution {
int max = 0;
public int diameterOfBinaryTree(TreeNode root) {
/**
* 在计算每个节点的最大深度时,顺便记录下其中深度最大的节点的值
*/
depth(root);
return max;
}
/**
* 计算每个节点为根节点时的最大深度
* 但是要的结果是过程中记录的max变量
*/
public int depth(TreeNode root){
if (root == null){
return 0;
}
int left = depth(root.left);
int right = depth(root.right);
max = Math.max(left + right, max);
return Math.max(left, right) + 1;
}
}
/**
* 时间复杂度 O(n)
* 空间复杂度 O(logn)
*/
https://leetcode-cn.com/problems/diameter-of-binary-tree/