题目
给你一棵二叉树的根节点,返回该树的 直径 。
二叉树的 直径 是指树中任意两个节点之间最长路径的 长度 。这条路径可能经过也可能不经过根节点 root 。
两节点之间路径的 长度 由它们之间边数表示。
代码
/**
- Definition for a binary tree node.
- public class TreeNode {
-
int val;
-
TreeNode left;
-
TreeNode right;
-
TreeNode() {}
-
TreeNode(int val) { this.val = val; }
-
TreeNode(int val, TreeNode left, TreeNode right) {
-
this.val = val;
-
this.left = left;
-
this.right = right;
-
}
- }
*/
class Solution {
int res = 0;
public int diameterOfBinaryTree(TreeNode root) {
solve(root);
return res;
}
int solve(TreeNode root){
if(root == null){
return 0;
}
int left = solve(root.left);
int right = solve(root.right);
res = Math.max(left + right, res);
return Math.max(left, right) + 1;
}
}
时间:O(n)
空间:O(n)