题目描述
给你一棵二叉树的根节点,返回该树的 直径 。
二叉树的 直径 是指树中任意两个节点之间最长路径的 长度 。这条路径可能经过也可能不经过根节点 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 ans = 1;
public int dfs(TreeNode node){
if(node==null) return 0;
int ldis = dfs(node.left); //左孩子能到达的最远叶子节点
int rdis = dfs(node.right); //右孩子能到达的最远叶子节点
ans = Math.max(ans, ldis+rdis+1);
return Math.max(ldis, rdis)+1;
}
public int diameterOfBinaryTree(TreeNode root) {
dfs(root);
return ans-1;
}
}