leetcode 543. 二叉树的直径

大耗子 2021年02月21日 17次浏览

题目: 543. 二叉树的直径

543. 二叉树的直径

难度简单613收藏分享切换为英文接收动态反馈

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

示例 :
给定二叉树

          1
         / \
        2   3
       / \     
      4   5    

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

解题思路

一个树中最长的结点链接是左右子树深度最长的连接起来。

那么我们只需要找到树的左右子树深度即可。

递归的方式解题

  1. 如果该节点存在,那么认为自己的深度起码有1。
  2. 获取左右子结点的深度,找到最大的与自己的深度加起来,得到自己最终的深度。例如子结点深度是3,那么自己的深度就是4。
  3. 最长的结点长度是自己的左右子树深度相加。
  4. 返回自己的深度。

代码

class Solution {
private:
	int max;
public:
	int diameterOfBinaryTree(TreeNode* root) {
		deepCore(root);
		return max;
	}
	int deepCore(TreeNode *root) {
		if (!root) return 0;
		int deepth = 1;
		int left = deepCore(root->left);
		int right = deepCore(root->right);
		deepth += left > right ? left : right;
		max = max > (left + right) ? max : (left + right);
		return (left > right ? left : right) + 1 ;
	}
};