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;
    }
};

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

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

猜你喜欢

转载自blog.csdn.net/L_bic/article/details/88948267