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

更新日期: 07-Apr-2021

123 查看次数

开启你的 职业生涯

通过完成课程获得认证

开始
广告