刷题系列 - 在二叉树中查找给出节点,并返回以该节点为根的树
很简答的一道题目,就是二叉树遍历找到某个节点的val是给出值,如果要返回的是以该节点为根节点的树,那么就是按照层级遍历,这里使用递归实现。如果找不到返回为空,如果找到返回该节点即可。
#Definitionforabinarytreenode.#classTreeNode:#def__init__(self,x):#self.val=x#self.left=None#self.right=NoneclassSolution:defsearchBSTNodeList(self,rootList,val):ifrootList==[]:returnNonenextRootList=[]fornodeinrootList:ifnode.val==val:returnnodeelse:ifnode.left!=None:nextRootList.append(node.left)ifnode.right!=None:nextRootList.append(node.right)returnself.searchBSTNodeList(nextRootList,val)defsearchBST(self,root:TreeNode,val:int)->TreeNode:returnself.searchBSTNodeList([root],val)
声明:本站所有文章资源内容,如无特殊说明或标注,均为采集网络资源。如若本站内容侵犯了原著者的合法权益,可联系本站删除。