leetcode hot100刷题日记——31.二叉树的直径
二叉树直径详解
- 题目描述
- 对直径的理解
- 解答: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){//如果当前节点是空,说明遍历到头了,返回0return 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;}
};
时间复杂度:O(N),N为二叉树节点数
空间复杂度:O(height),height为二叉树的高度
小TIPS
不要这样理解直径啊!这不是又走了一遍2节点吗!