AQY:查找二叉树中值为k的节点

class TreeNode():
    def __init__(self, x: int):
        self.val = x;
        self.left = None
        self.right = None


class Tree():
    def __init__(self, k: int):
        self.k = k
        self.searchedNode = None

    # DFS 深度遍历 - 先序遍历
    def searchValueKTreeNode(self, node: TreeNode):
        if node is None:
            return

        if node.val == self.k:
            self.searchedNode = node
            return

        self.searchValueKTreeNode(node.left)
        self.searchValueKTreeNode(node.right)

    # BFS 层序遍历
    def searchBFSValueKTreeNode(self, root: TreeNode):
        if root is None:
            return

        queue = [root]
        while queue:
            tmp = []
            for node in queue:
                if node.val == self.k:
                    self.searchedNode = node
                    return
                if node.left: tmp.append(node.left)
                if node.right: tmp.append(node.right)

            queue = tmp


root = TreeNode(9)
secondNode = TreeNode(8)
threeNode = TreeNode(7)
fourNode = TreeNode(10)

root.left = secondNode
secondNode.left = threeNode
root.right = fourNode

tree = Tree(8)
tree.searchValueKTreeNode(root)
# tree.searchBFSValueKTreeNode(root)
print(tree)

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值