leetcode树之在受污染的二叉树中查找元素

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

给出一个满足下述规则的二叉树：

• root.val == 0
• 如果 treeNode.val == x 且 treeNode.left != null，那么 treeNode.left.val == 2 * x + 1
• 如果 treeNode.val == x 且 treeNode.right != null，那么 treeNode.right.val == 2 * x + 2
  现在这个二叉树受到「污染」，所有的 treeNode.val 都变成了 -1。

请你先还原二叉树，然后实现 FindElements 类：

• FindElements(TreeNode* root) 用受污染的二叉树初始化对象，你需要先把它还原。
• bool find(int target) 判断目标值 target 是否存在于还原后的二叉树中并返回结果。

---

思路：

通过递归来还原二叉树，然后对于判断二叉树是否存在目标值，可以使用二进制来判断查找目标值的路径，通过这样来判断是否存在。

class FindElements:
  	# 前序遍历回复二叉树的值
    def __init__(self, root: Optional[TreeNode]):
      	# 迭代是向下层传递本来应该的值，如果节点存在，进行节点赋值
        def dfs(root, val):
            if not root: return
            root.val = val
            dfs(root.left, 2 * val + 1)
            dfs(root.right, 2 * val + 2)
        dfs(root, 0)
        self.root = root
		# 获取目标值的路径，这里不停的通过对2进行求商和余数
    # 结果集是目标值倒序结果，而二叉树的路径应该是次高位到低位的一个迅速
    def getPath(self, target):
        ans = []
        d,r = target, 0
        while d:
            d, r = divmod(d, 2)
            ans.append(r)
        return ans[:-1]

    def find(self, target: int) -> bool:
        if not self.root: return False
        # 根节点从1开始，这样有利于进行节点路径的查找
        paths = self.getPath(target + 1)
        cur = self.root
        # 取路径最后一位，如果是0，取左子节点，否则取右子节点，如果没有，说明该路径不存在，返回False
        while paths:
            curPath = paths.pop()
            if curPath:
                if not cur.right: return False
                cur = cur.right
            else:
                if not cur.left: return False
                cur = cur.left
        # 如果遍历完成后，节点存在，说明二叉树中包含该目标值
        return cur != None