题目
思路:
这道题一开始理解错了,想到求左右子树的最大深度相加就可以了,这是错误的。
正确思路:
- 用一个全局变量ans保存最长长度,ans初始化为1。
- 计算从每个节点出发到达左右叶节点的路径长度,更新全局变量,
ans = Math.max(ans,left + right);
- 同时将到达左右叶节点的路径长度的最大值作为到达该节点的路径长度。
return Math.max(left,right)+1;
具体代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
int ans;
public int diameterOfBinaryTree(TreeNode root) {
if(root == null||(root.left == null && root.right == null)){
return 0;
}
ans = 1;
maxForEveryRoot(root);
return ans;
}
public int maxForEveryRoot(TreeNode root){
if(root == null){
return 0;
}
int left = maxForEveryRoot(root.left);
int right = maxForEveryRoot(root.right);
ans = Math.max(left + right,ans);
return Math.max(left,right)+1;
}
}
时间复杂度:O(N),每个节点遍历一次
空间复杂度:O(Height),递归调用栈深度为二叉树高度