给出一个满足下述规则的二叉树:
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 是否存在于还原后的二叉树中并返回结果。
输入:
["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
【代码】:
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root
*/
var FindElements = function(root) {
this.root = root;
this.recover(this.root, 0);
};
//所有在FindElements .prototype上的扩展都得放在FindElements 下面
FindElements.prototype.bianli = function(node, target){
if(node){
if(node.val == target){
return true;
}
return this.bianli(node.left, target) || this.bianli(node.right, target);
}else{
return false;
}
}
FindElements.prototype.recover = function(node, val){
if(node == null) return ;
node.val = val;
this.recover(node.left, 2*val + 1);
this.recover(node.right, 2*val + 2);
}
/**
* @param {number} target
* @return {boolean}
*/
FindElements.prototype.find = function(target) {
return this.bianli(this.root, target)
};
/**
* Your FindElements object will be instantiated and called as such:
* var obj = new FindElements(root)
* var param_1 = obj.find(target)
*/