LeetCode 5258. 在受污染的二叉樹中查找元素 Find Elements in a Contaminated Binary Tree

地址 https://leetcode-cn.com/contest/weekly-contest-163/problems/find-elements-in-a-contaminated-binary-tree/

題目描述
給出一個知足下述規則的二叉樹：

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
二叉樹的高度不超過 20
節點的總數在 [1, 10^4] 之間
調用 find() 的總次數在 [1, 10^4] 之間
0 <= target <= 10^6

算法1
使用數組表示二叉樹 根爲x 則左子樹爲2x+1 右子樹爲2x+2
剛好索引就是題目中各個節點的值

那麼創建一個數組 記錄輸入二叉樹所擁有的節點 查詢就很是簡單 直接根據查詢值看該數組的索引是否已經標記

C++ 代碼

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class FindElements {
public:

    vector<int> record;

void Dfs(TreeNode* root, int k) {
    if (root == NULL)
        return;
    root->val = k;
    record[k] = 1;

    Dfs(root->left, k * 2 + 1);
    Dfs(root->right, k * 2 + 2);
}

 FindElements(TreeNode* root) {
    record.resize(10000000, -1);
    Dfs(root, 0);
}

bool find(int target) {
    return record[target] == 1;
}

};

/**
 * Your FindElements object will be instantiated and called as such:
 * FindElements* obj = new FindElements(root);
 * bool param_1 = obj->find(target);
 */