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

一世执手 提交于 2020-03-10 22:12:40

在这里插入图片描述
题解:
本题用深度优先搜索
我们通过画图来找寻本题的思路;
容易分析得,本题需要两个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
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!