LeetCode - 543. Diameter of Binary Tree - C++

Solution里的一种解法：

先算路径的节点数，最后 -1；

以每个节点为“根节点”的最长路径，找出最大值。

最长路径是1+左子树深度+右子树深度

树深度是1+max(左子树深度，右子树深度)

递归。核心递归函数必须同时有两个作用：1.比出最长路径；2.计算深度

代码如下：

class Solution {
public:
    int diameterOfBinaryTree(TreeNode* root) {
        int maxNodeNum = 1;
        countFromNode(root, maxNodeNum);
        
        return maxNodeNum - 1;
    }
    
    int countFromNode(TreeNode* root, int& maxNodeNum) {
        if(root == nullptr) return 0;
        int left = countFromNode(root->left, maxNodeNum);
        int right = countFromNode(root->right, maxNodeNum);
        
        maxNodeNum = max(maxNodeNum, left+right+1);
        return max(left, right) + 1;
    }
};

感觉算递归里比较难的。关键还是递归子函数的返回值、参数、以及一些“操作”，比较绕。

只能多思考、熟能生巧了～