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;
}
};
感觉算递归里比较难的。关键还是递归子函数的返回值、参数、以及一些“操作”,比较绕。
只能多思考、熟能生巧了~