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

二叉树的直径

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

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

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

示例 1：

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