题目描述如上,思路:我们可以想到,不论我们找到一条什么样的路径,我们都可以找到这条路径上最高的那个点(离根节点最近的一个点),此时,这个点左边的最长路径即为以这个点为根节点的子树的左子树的最大深度,右边的最长路径长度即为右子树的最大深度。以题目中给的样例为例,答案中的两条路径所包含的最高点是1,也就是根节点,最长路径长度即为根节点左边的最长路径加上根节点右边的最长路径。而这个最长路径也就是分别对应了左子树和右子树的深度,左子树的深度为2,右子树的深度为1,故最终的答案为2+1=3。那么如何求出左子树深度加右子树深度最大的那个子树呢,我们需要对树的每个节点进行遍历,对于每个节点,采用递归的方法计算其深度,我们知道树的深度等于max(左孩子为顶点对应子树的深度,右孩子为顶点对应子树的深度)+1,故我们可以实现一个递归函数depth(),用一个result来存储当前已遍历节点最大的深度。代码如下:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
int result = 0;
int depth(TreeNode *root){
if(root==nullptr){
return 0;
}
int L = depth(root->left);
int R = depth(root->right);
result = max(result,L+R+1);
return max(L,R)+1;
}
int diameterOfBinaryTree(TreeNode* root) {
result = 1;
depth(root);
return result-1;
}
};