[每日一题] 148. 二叉树的直径(二叉树、易错坑点、巧妙解法)

1. 题目来源

链接：二叉树的直径

2. 题目说明

给定一棵二叉树，你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过根结点。

示例 :

给定二叉树

          1
         / \
        2   3
       / \     
      4   5    
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

注意：

• 两结点之间的路径长度是以它们之间边的数目表示。

3. 题目解析

方法一：二叉树+易错坑点+巧妙解法

这个问题乍一看还是很简单的么，就从根节点出发，统计其左子树、右子树的最大高度，加起来就行了。统计树的最大高度的话可参考博主博文：[剑指-Offer] 55 - I. 二叉树的深度及II. 平衡二叉树(二叉树、后序遍历、代码优化、巧妙解法)

结果就凉了…😦😭

102 / 106 个通过测试用例
[4,-7,-3,null,null,-9,-3,9,-7,-4,null,6,null,-6,-6,null,null,0,6,5,null,9,null,null,-1,-4,null,null,null,-2]

一开始确实没想到它能不从根节点出发啊~

那么就需要对 maxDepth 做改进了，首先函数外申请一个计数 cnt 用以记录当前节点的最大直径，若新节点的直径大于 cnt，则更新 cnt 即可。

参见代码如下：

// 执行用时 :8 ms, 在所有 C++ 提交中击败了95.78%的用户
// 内存消耗 :22.3 MB, 在所有 C++ 提交中击败了15.63%的用户

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int cnt = 0;
    int diameterOfBinaryTree(TreeNode* root) {
        if (root == nullptr) return 0;
        maxDepth(root); 
        return cnt;
    }
    int maxDepth(TreeNode* root) {
        if (root == nullptr) return 0;
        int left = maxDepth(root->left);
        int right = maxDepth(root->right);
        if ((left + right) > cnt) cnt = left + right;
        return 1 + max(left, right);
    }
};