当某些节点为 null 时，二叉树搜索不返回匹配值

Question

TL;DR - 如何更改此算法以返回给定具有 null 节点的 BST 的匹配 val？

树节点：

/**
 * 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
 * @param {number} val
 * @return {TreeNode}
 */

给定一棵二叉树，找到匹配的 val。 我的算法如下

var searchBST = function(root, val) {
    const matchingNode = visitNode(root, val);
    // if no match, return null
    return matchingNode ? matchingNode : null;
};

const visitNode = (node, val) => {
    if (node.left !== null) {
        return visitNode(node.left, val)
    }
    if (node.right !== null) {
        return visitNode(node.right, val)
    }
    if (node.val === val) {
        console.log('MATCH!')
        return node;
    }
}

这适用于没有null值的标准树，例如 [4,2,7,1,3] 和空树 []。

I'm having an issue with trees that have null nodes, such as [18, 2, 22, null, null, null, 63, null, 84, null, null].

function 似乎过早停止。 如果我删除前两个 if 块中的返回，我可以在匹配的 val 处停止递归，但无法获得要返回的值。

如何更改此算法以返回给定具有 null 节点的 BST 的匹配 val？

Answer 1

您可以考虑下面的visitNode()重构版本。 基本情况发生在传入节点是null时，在这种情况下 function 只返回null 。 否则，它会检查当前节点是否与寻求的值匹配。 如果不是，则它递归地遍历左或右。

const visitNode = (node, val) => {
    if (node == null) return null;

    if (node.val === val) {
        console.log('MATCH!')
        return node;
    }

    if (val < node.val) {
        return visitNode(node.left, val);
    }
    else {
        return visitNode(node.right, val)
    }
}

Answer 2

如果你正在寻找二叉搜索树，试试这个

if (node == null || node.val === val) 
   return node;
 
// val is greater than node's val
if (node->val < val) 
   return visitNode(node.right, val); 

// val is smaller than node's val
return visitNode(node.left, val); 

或者你必须通过DFS或BFS遍历树