Leetcode 543. 二叉树的直径

小灰灰 2023-02-14 03:10 8阅读 0赞

### Leetcode 543. 二叉树的直径 ###

*  1、问题分析
 *  2、问题解决
 *  3、总结

# 1、问题分析 #

题目链接：[https://leetcode-cn.com/problems/diameter-of-binary-tree/][https_leetcode-cn.com_problems_diameter-of-binary-tree]  
  本质上就是一个二叉树的遍历问题。代码我已经进行了详细的注释，理解应该没有问题，读者可以作为参考，如果看不懂(可以多看几遍)，欢迎留言哦！我看到会解答一下。

# 2、问题解决 #

笔者以`C++`方式解决。

#include "iostream"
    
    using namespace std;
    
    #include "algorithm"
    #include "vector"
    #include "queue"
    #include "set"
    #include "map"
    #include "string"
    #include "stack"
    
    /** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */
    
    struct TreeNode { 
        int val;
        TreeNode *left;
        TreeNode *right;
    
        TreeNode(int x) : val(x), left(NULL), right(NULL) { }
    };
    
    class Solution { 
    private:
        // 定义最大直径 初始值为 0
        int result = 0;
    public:
        int diameterOfBinaryTree(TreeNode *root) { 
            // 如果树为空 直接返回 0
            if (root == NULL) { 
                return 0;
            }
            // 寻找最大直径
            dfs(root);
    
            // 返回处理后的结果
            return result;
        }
    
        /** * 计算 以 root 节点为根节点的树的高度 * @param root * @return */
        int dfs(TreeNode *root) { 
            // 如果节点为空 返回 -1
            if (root == NULL) { 
                return -1;
            }
    
            // 如果是叶子节点 返回高度为 0
            if (root->left == nullptr && root->right == nullptr) { 
                return 0;
            }
    
            TreeNode *rootLeft = root->left;
            TreeNode *rootRight = root->right;
    
            // 分别计算左右子树的高度
            int a = dfs(rootLeft) + 1;
            int b = dfs(rootRight) + 1;
    
            // 左右子树的高度相加，就是二叉树的直径
            // 选取较大的保存
            if (a + b > result) { 
                result = a + b;
            }
    
            // 返回左右子树中高度较大的返回作为树的高度
            return max(a, b);
        }
    };
    
    int main() { 
        TreeNode *pNode1 = new TreeNode(1);
        TreeNode *pNode2 = new TreeNode(2);
        TreeNode *pNode3 = new TreeNode(3);
        TreeNode *pNode4 = new TreeNode(4);
        TreeNode *pNode5 = new TreeNode(5);
    // TreeNode *pNode6 = new TreeNode(6);
    // TreeNode *pNode7 = new TreeNode(7);
    
        pNode1->left = pNode2;
        pNode1->right = pNode3;
        pNode2->left = pNode4;
        pNode2->right = pNode5;
    // pNode4->left = pNode6;
    // pNode5->right = pNode7;
    
        Solution *pSolution = new Solution;
        int i = pSolution->diameterOfBinaryTree(pNode1);
        cout << i << endl;
        system("pause");
        return 0;
    }

运行结果

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMyNTEwNTk3_size_16_color_FFFFFF_t_70]  
有点菜，有时间再优化一下。

# 3、总结 #

难得有时间刷一波`LeetCode`, 这次做一个系统的记录，等以后复习的时候可以有章可循，同时也期待各位读者给出的建议。算法真的是一个照妖镜，原来感觉自己也还行吧，但是算法分分钟教你做人。前人栽树，后人乘凉。在学习算法的过程中，看了前辈的成果，受益匪浅。  
感谢各位前辈的辛勤付出，让我们少走了很多的弯路！  
哪怕只有一个人从我的博客受益，我也知足了。  
点个赞再走呗！欢迎留言哦！

[https_leetcode-cn.com_problems_diameter-of-binary-tree]: https://leetcode-cn.com/problems/diameter-of-binary-tree/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMyNTEwNTk3_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20200531222223491.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMyNTEwNTk3,size_16,color_FFFFFF,t_70