如何在二叉搜索树中找到最小的 n 个元素？

Question

我试过了，但它正在打印整个树：

function inOrder(root, i, n) {
    if (!root) return;
    inOrder(root.left, i, n);
    if(i++ < n) console.log(root.key);
    inOrder(root.right, i, n);
}

Answer 1

发生这种情况是因为i++只更改了局部变量。 它不会改变调用者的i变量，因此在左子树的遍历中完成的增量不会对传递给第二个递归调用的i的值产生任何影响。

您可以通过多种方式解决此问题，但一种是让每个递归调用将其i的值返回给调用者，以便调用者可以使用它来更新自己的i并继续使用该值。

其他一些评论：

• 一旦i达到n的值，仍然进行递归调用是没有用的，所以当这种情况发生时立即退出函数

• 与其让i递增，不如让n递减。 这样你只需要一个额外的参数而不是两个。

 function inOrder(root, n) { // No more `i` if (!root) return n; n = inOrder(root.left, n); if (n-- <= 0) return n; // decrement instead of increment now & exit console.log(root.key); return inOrder(root.right, n); } class Node { // Definition of class to make the demo work constructor(key) { this.key = key; this.left = this.right = null; } static from(...keys) { if (!keys.length) return; const root = new Node(keys.shift()); for (const key of keys) root.add(key); return root; } add(key) { if (key < this.key) { if (this.left) this.left.add(key); else this.left = new Node(key); } else { if (this.right) this.right.add(key); else this.right = new Node(key); } } } // demo const root = Node.from(5, 13, 4, 7, 11, 1, 8, 12, 9, 3, 2, 6, 10); inOrder(root, 6);