leetcode 543. 二叉树的直径(python)

最新推荐文章于 2022-03-15 14:36:16 发布

circle_yy

最新推荐文章于 2022-03-15 14:36:16 发布

阅读量258

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/cy_believ/article/details/104778606

版权

leetcode 专栏收录该内容

67 篇文章 0 订阅

订阅专栏

使用深度优先搜索解决LeetCode 543题，通过遍历左子树和右子树来找到二叉树的最大直径。通过两次DFS，分别计算左子树和右子树的深度，并在回溯时更新路径长度。最后，将左右子树的最大深度相加得到二叉树的直径。

摘要由CSDN通过智能技术生成

在这里插入图片描述
题解：
本题用深度优先搜索
我们通过画图来找寻本题的思路；
容易分析得，本题需要两个dfs，一个是遍历左结点的子树，一个遍历右结点的子树；
如图所示，首先遍历左边的结点一直到根;

def dfs(node):
	if not node:
		return 0

	dfs(node.left)

此时需要进行回溯，回溯到结点2；但是路径长度需要+1；因此代码更改如下所示：

def dfs(node):
	if not node:
		return 0

	l = dfs(node.left) 
	return l + 1

同理，对右边结点的遍历也是上述的代码，即如下所示：

def dfs(node):
	if not node:
		return 0

	r = dfs(node.right) 
	return r + 1

左右子树合在一起。计算以当前结点为根的最大深度代码如下所示：

def dfs(node):
	if not node:
		return 0
	
	l = dfs(node.left)
	r = dfs(node.right) 
	return max(l,r) + 1

因此最大路径为：L+R。
代码如下所示：

class Solution:
    def diameterOfBinaryTree(self, root):

    	self.ans = 0

    	def dfs(node):

    		if not node:
    			return 0

    		L = dfs(node.left)
    		R = dfs(node.right)

    		self.ans = max(self.ans, L + R)


    		return max(L,R) + 1



    	dfs(root)

    	return self.ans