【LeetCode回溯算法#12】二叉树的直径,树形dp的前置内容(使用dfs)
二叉树的直径
给你一棵二叉树的根节点,返回该树的 直径 。
二叉树的 直径 是指树中任意两个节点之间最长路径的 长度 。这条路径可能经过也可能不经过根节点 root
。
两节点之间路径的 长度 由它们之间边数表示。
示例 1:
![img](https://assets.leetcode.com/uploads/2021/03/06/diamtree.jpg)
输入:root = [1,2,3,4,5]
输出:3
解释:3 ,取路径 [4,2,1,3] 或 [5,2,1,3] 的长度。
示例 2:
输入:root = [1,2]
输出:1
代码与思路
class Solution {
int maxPath;
int dfs(TreeNode* node){
if(node == nullptr) return 0;//终止条件
int left_depth = dfs(node->left);//递归求当前节点左右子节点的深度
int right_depth = dfs(node->right);
//更新最大直径(路径),如果当前节点的左右子节点深度更大就被选取
maxPath = max(maxPath, left_depth + right_depth);
return max(left_depth, right_depth) + 1;//返回深度更大的方向
//加1是当前层的深度
}
public:
int diameterOfBinaryTree(TreeNode* root) {
maxPath = 0;
dfs(root);
return maxPath;
}
};