leetcode hot100刷题日记——31.二叉树的直径

news2025/6/2 8:15:50

二叉树直径详解

- 题目描述
- 对直径的理解
- 解答：dfs
- - 小TIPS

题目描述

在这里插入图片描述

对直径的理解

在这里插入图片描述
实际上，二叉树的任意一条路径均可以被看作由某个节点为起点，从其左儿子和右儿子向下遍历的路径拼接得到。

那我们找二叉树的直径（最大路径），就是需要：以左儿子为根节点的二叉树的深度，以右儿子为根节点的二叉树的深度，所有左深度加右深度的和中的最大值就是直径

而我们在计算二叉树的深度的时候，可以利用要比较左右子树深度这一点来计算路径。

（有了左深度，有了右深度，一加起来不就是路径长度嘛！全部记录下来找最大值就是路径啦~
在题解中表现为每次都和当前最大值比较，更新最大值）

解答：dfs

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int max_depth;//记录直径
    int depth(TreeNode* cur){
        if(cur==nullptr){//如果当前节点是空，说明遍历到头了，返回0
            return 0;
        }        
        int depth_L=depth(cur->left);//当前节点左子树深度
        int depth_R=depth(cur->right);//当前节点右子树深度
        max_depth=max(depth_L+depth_R,max_depth);//计算经过当前节点的路径，更新直径潜在值
        return max(depth_L,depth_R)+1;//每次返回左右子树中的更深子树的深度+1，因为自己也算一层
    }
    int diameterOfBinaryTree(TreeNode* root) {
      depth(root);//depth函数计算以节点root为根的深度
      return max_depth;
    }
};