JavaScript 中 BST 中的两数之和
问题
我们需要编写一个 JavaScript 函数,该函数将两个二叉查找树的根节点分别作为第一个和第二个参数接收。该函数的第三个参数是一个数字 target。
当且仅当第一个树中有一个节点和第二个树中有一个节点的值相加等于给定整数 target 时,我们的函数应返回 True,否则返回 false。
例如,如果输入函数的参数是 −
const target = 23;
BST
则输出应为 −
const output = true;
输出说明
因为第一个树中有 6,第二个树中有 17,其和为 23。
示例
代码如下 −
class Node{ constructor(data) { this.data = data; this.left = null; this.right = null; }; }; class BinarySearchTree{ constructor(){ // root of a binary seach tree this.root = null; } insert(data){ var newNode = new Node(data); if(this.root === null){ this.root = newNode; }else{ this.insertNode(this.root, newNode); }; }; insertNode(node, newNode){ if(newNode.data < node.data){ if(node.left === null){ node.left = newNode; }else{ this.insertNode(node.left, newNode); }; } else { if(node.right === null){ node.right = newNode; }else{ this.insertNode(node.right,newNode); }; }; }; }; const tree1 = new BinarySearchTree(); tree1.insert(1); tree1.insert(3); tree1.insert(4); tree1.insert(2); tree1.insert(4); tree1.insert(5); tree1.insert(6); const tree2 = new BinarySearchTree(); tree2.insert(9); tree2.insert(6); tree2.insert(11); tree2.insert(3); tree2.insert(8); tree2.insert(10); tree2.insert(17); const twoSumBSTs = (root1, root2, target) => { if(root1 == null || root2 == null) return false; if(root1.data + root2.data == target) return true; if(root1.data + root2.data < target) { return twoSumBSTs(root1.right, root2, target) || twoSumBSTs(root1, root2.right, target); }else{ return twoSumBSTs(root1.left, root2, target) || twoSumBSTs(root1, root2.left, target); }; }; const target = 23; console.log(twoSumBSTs(tree1.root, tree2.root, target));
输出
控制台中的输出将是 −
true
广告