1261.在受污染的二叉树中查找元素

链接：1261.在受污染的二叉树中查找元素
难度：Medium
标签：树、深度优先搜索、广度优先搜索、设计、哈希表、二叉树
简介：请你先还原二叉树，然后实现 FindElements 类。

题解 1 - python

• 编辑时间：2024-03-12
• 执行用时：66ms
• 内存消耗：20.3MB
• 编程语言：python
• 解法介绍：dfs。

class FindElements:
    def __init__(self, root: Optional[TreeNode]):
        self.set = set()
        def dfs(node: Optional[TreeNode], cur: int):
            if not node: return
            self.set.add(cur)
            dfs(node.left, cur * 2 + 1)
            dfs(node.right, cur * 2 + 2)
        dfs(root, 0)
    def find(self, target: int) -> bool:
        return target in self.set