BST 递归解中的第 k 个最小元素

Kth Smallest Element in a BST recursive solution

本文关键字：元素 递归 BST      更新时间：2023-10-16

此代码用于返回 BST 中的第 k 个最小元素。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
int fid;
void inorder(TreeNode* node, int& fid, int& k) {
if (!node) return;
inorder(node->left, fid, k);
if (!k) return;
fid = node->val;
k--;
inorder(node->right, fid, k);
}
int kthSmallest(TreeNode* root, int k) {
inorder(root, fid, k);
return fid;
}
};

它给出了正确的答案。我想我可以通过将 inorder 函数和 kthTiny 函数组合在一起来简化代码。

class Solution {
public:
int fid;
 int kthSmallest(TreeNode* root, int k) {
 if (!root) return fid;
kthSmallest(root->left, k);
if (!k) return fid;
fid = root->val;
k--;
kthSmallest(root->right, k);
return fid;
}
};

它给出了错误的答案。如果我输入 BST{2,1} k=1，它将返回 2 而不是 1。我不知道为什么两个代码不同。感谢您的帮助！

作为参数int& k（通过引用）和int k（通过值）之间存在很大差异。

在第一种情况下，对k所做的任何更改（例如 k-- ）都会反映在作为参数传递的调用函数的变量中。

在第二种情况下，函数接收自己的k副本，对其所做的任何更改都不会影响调用函数。

你可以改变你的kthSmallest，把k作为参考（int &），但这会改变你被允许调用它的方式。您可能仍然需要一个帮助程序函数，但您可以将所有代码放入其中。

相关文章：