【LeetCode回溯算法#12】二叉树的直径,树形dp的前置内容(使用dfs)

发布时间 2023-08-26 21:11:49作者: dayceng

二叉树的直径

给你一棵二叉树的根节点,返回该树的 直径

二叉树的 直径 是指树中任意两个节点之间最长路径的 长度 。这条路径可能经过也可能不经过根节点 root

两节点之间路径的 长度 由它们之间边数表示。

示例 1:

img
输入: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;
    }
};