0
点赞
收藏
分享

微信扫一扫

力扣543——二叉树的直径(DFS)

嚯霍嚯 2022-01-20 阅读 58

题目(简单)

在这里插入图片描述

解题思路

单独考虑一个根节点的直径,是代表左边深度+右边深度,dfs求深度即可;
考虑孩子作为中间点可能产生更大的直径;
那一棵树的直径就是自己的直径、左孩直径、右孩直径中最大值;
套递归就行了

代码

class Solution {
public:
    int dfs(TreeNode* root, int len) {   //树的深度
        if(root == nullptr) return len;
        return max(dfs(root->left, len+1), dfs(root->right, len+1));
    }
    int diameterOfBinaryTree(TreeNode* root) {
        if(root == nullptr) return 0;
        int ans = dfs(root->left, 0) + dfs(root->right, 0);
        return max(ans, max(diameterOfBinaryTree(root->left), diameterOfBinaryTree(root->right)));
    }
};
举报

相关推荐

0 条评论