刷題系列 - 在二叉樹(shù)中查找給出節(jié)點(diǎn)，并返回以該節(jié)點(diǎn)為根的樹(shù)

發(fā)布時(shí)間：2020-08-07 14:18:22 來(lái)源：ITPUB博客閱讀：177 作者：張國(guó)平欄目：編程語(yǔ)言

很簡(jiǎn)答的一道題目，就是二叉樹(shù)遍歷找到某個(gè)節(jié)點(diǎn)的val是給出值，如果要返回的是以該節(jié)點(diǎn)為根節(jié)點(diǎn)的樹(shù)，那么就是按照層級(jí)遍歷，這里使用遞歸實(shí)現(xiàn)。如果找不到返回為空，如果找到返回該節(jié)點(diǎn)即可。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def searchBSTNodeList(self,rootList,val):
        if rootList == []:
            return None
        nextRootList = []
        for node in rootList:
            if node.val == val:
                return node
            else:
                if node.left != None:
                    nextRootList.append(node.left)
                if node.right != None:
                    nextRootList.append(node.right)
        return self.searchBSTNodeList(nextRootList,val)
                    
            
    def searchBST(self, root: TreeNode, val: int) -> TreeNode:
         return self.searchBSTNodeList([root],val)

向AI問(wèn)一下細(xì)節(jié)

刷題系列 - 在二叉樹(shù)中查找給出節(jié)點(diǎn)，并返回以該節(jié)點(diǎn)為根的樹(shù)

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標(biāo)簽

刷題系列 - 在二叉樹(shù)中查找給出節(jié)點(diǎn)，并返回以該節(jié)點(diǎn)為根的樹(shù)