2024/3/12Leetcode每日一题（二叉树）

最新推荐文章于 2024-10-14 18:52:56 发布

为破特提灯

最新推荐文章于 2024-10-14 18:52:56 发布

阅读量1.6k

点赞数 47

分类专栏： Leetcode&Python 文章标签： leetcode python

本文链接：https://blog.csdn.net/qq_55974781/article/details/136643468

版权

Leetcode&Python 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

Python小白的LeetCode每日一题：

基础知识：

二叉树的基本概念
参考文章：atwdy博主的《【数据结构】——二叉树详解》
二叉树遍历方法之深度优先搜索（DFS）
参考文章：Turing_Sheep博主的《DFS与BFS（保姆级教学！！超级详细的图示！！）》
哈希表的基本概念
参考文章：白话机器学习博主的《算法数据结构基础——哈希表（Hash Table）》

题目：

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

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 是否存在于还原后的二叉树中并返回结果。

示例 1：
在这里插入图片描述
输入：
[“FindElements”, “find”, “find”]
[ [[-1, null, -1]], [1], [2] ]
输出：
[null, false, true]
解释：
FindElements findElements = new FindElements([-1, null, -1]);
findElements.find(1); // return False
findElements.find(2); // return True
示例 2：
在这里插入图片描述

输入：
[“FindElements”, “find”, “find”, “find”]
[ [[-1,-1,-1,-1,-1]], [1], [3], [5] ]
输出：
[null, true, true, false]
解释：
FindElements findElements = new FindElements([-1, -1, -1, -1, -1]);
findElements.find(1); // return True
findElements.find(3); // return True
findElements.find(5); // return False
示例 3：
在这里插入图片描述

输入：
[“FindElements”, “find”, “find”, “find”, “find”]
[ [[-1,null,-1,-1,null,-1]], [2], [3], [4], [5] ]
输出：
[null, true, false, false, true]
解释：
FindElements findElements = new FindElements([-1, null, -1, -1, null, -1]);
findElements.find(2); // return True
findElements.find(3); // return False
findElements.find(4); // return False
findElements.find(5); // return True

提示：
TreeNode.val == -1
TreeNode.val == -1
二叉树的高度不超过 20
节点的总数在 [1, 10^4] 之间
调用 find() 的总次数在 [1, 10^4] 之间
0 <= target <= 10^6

提交代码：

和官方题解大致相同，解题思路：

深度优先搜索根节点
创建哈希表存储二叉树的值
判断target在不在哈希表内

在这里插入图片描述

官方题解：

class FindElements:
    def __init__(self, root: Optional[TreeNode]):
        self.valSet = set()
        self.dfs(root, 0)

    def find(self, target: int) -> bool:
        return target in self.valSet

    def dfs(self, node: TreeNode, val: int):
        if node is None:
            return
        node.val = val
        self.valSet.add(val)
        self.dfs(node.left, val * 2 + 1)
        self.dfs(node.right, val * 2 + 2)

"""
作者：力扣官方题解
链接：https://leetcode.cn/problems/find-elements-in-a-contaminated-binary-tree/solutions/2674238/zai-shou-wu-ran-de-er-cha-shu-zhong-cha-5l56x/
来源：力扣（LeetCode）
"""