543.diameter-of-binary-tree

一个努力的码农

于 2021-07-26 17:13:06 发布

阅读量193

点赞数 1

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签：二叉树算法 leetcode java 数据结构

本文链接：https://blog.csdn.net/qq_26000415/article/details/119113887

leetcode 专栏收录该内容

33 篇文章

订阅专栏

本文介绍了一种计算二叉树直径长度的方法。通过递归计算每个节点的左右子树深度，并利用备忘录减少重复计算，最终得到任意两点间路径的最大长度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

示例 :
给定二叉树

在这里插入图片描述

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

注意：两结点之间的路径长度是以它们之间边的数目表示。

解法

首先要计算每个节点的路径长度,路径长度 = 左节点的深度+右节点的深度.

由于可能存在重复计算,因此要使用备忘录.

最后,在递归的时候,更新结果就行

class Solution {
        private int max = 0;
		
		private Map<TreeNode, Integer> r = new HashMap<>();
		
		public int diameterOfBinaryTree(TreeNode root) {

			if (null == root) {
				return 0;
			}

			
			int l = depth(root.left);
			int r = depth(root.right);
			max = Math.max(max, l + r);
			
			diameterOfBinaryTree(root.left);
			diameterOfBinaryTree(root.right);

			return max ;

		}

		private int depth(TreeNode root) {
			
			if(r.containsKey(root)) {
				return r.get(root);
			}
			
			if (null == root) {
				return 0;
			}
			
			r.put(root, Math.max(depth(root.left), depth(root.right)) + 1);
			return r.get(root);
		}
}