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

Posted 2023-02-17 Debroon

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

• • 题目
  • 算法设计：深度优先搜索

 

---

题目

传送门：https://leetcode.cn/problems/find-elements-in-a-contaminated-binary-tree/

 

---

算法设计：深度优先搜索

还原二叉树的时候只需要遍历所有节点，通过函数参数传递每个节点的值。

由于节点的个数规模不算大，所以可以直接用一个 HashSet 缓存所有节点值，帮助 find 函数快速判断。

class FindElements 
    unordered_set<int> cache;
    void recover(TreeNode *root, int prev) 
        if(!root) return;      
        root->val = prev;
        cache.insert(root->val);      
        recover(root->left, 2*prev + 1);
        recover(root->right, 2*prev + 2);
     
public:
    FindElements(TreeNode* root)  recover(root, 0); 
    bool find(int target)   return cache.count(target) > 0; 
;