Input
["FindElements","find","find"]
[[[-1,null,-1]],[1],[2]]
Output
[null,false,true]
Explanation
FindElements findElements = new FindElements([-1,null,-1]);
findElements.find(1); // return False
findElements.find(2); // return True
Input
["FindElements","find","find","find"]
[[[-1,-1,-1,-1,-1]],[1],[3],[5]]
Output
[null,true,true,false]
Explanation
FindElements findElements = new FindElements([-1,-1,-1,-1,-1]);
findElements.find(1); // return True
findElements.find(3); // return True
findElements.find(5); // return False
Input
["FindElements","find","find","find","find"]
[[[-1,null,-1,-1,null,-1]],[2],[3],[4],[5]]
Output
[null,true,false,false,true]
Explanation
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
/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
type FindElements struct {
tree *TreeNode
}
func Constructor(root *TreeNode) FindElements {
root.Val = 0
createTree(root)
return FindElements{ tree: root }
}
func createTree(root *TreeNode) {
if root == nil { return }
if root.Left != nil {
root.Left.Val = 2 * root.Val + 1
}
if root.Right != nil {
root.Right.Val = 2 * root.Val + 2
}
createTree(root.Left)
createTree(root.Right)
}
func (this *FindElements) Find(target int) bool {
return find(this.tree, target)
}
func find(root *TreeNode, target int) bool {
if root == nil { return false }
if root.Val == target { return true }
return find(root.Left, target) || find(root.Right, target)
}
/**
* Your FindElements object will be instantiated and called as such:
* obj := Constructor(root);
* param_1 := obj.Find(target);
*/