BST递归解决方案中的Kth最小元素

时间:2016-04-27 23:56:18

标签: c++ algorithm recursion binary-search-tree

此代码用于返回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函数和kthSmallest函数组合在一起来简化代码。

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。 我无法弄清楚为什么两个代码不同。谢谢你的帮助!

1 个答案:

答案 0 :(得分:1)

int& k(按引用)和int k(按值)作为参数之间存在很大差异。

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

在第二种情况下,函数会收到自己的k副本,对其进行的任何更改都不会影响调用函数。

您可以将kthSmallest更改为k作为参考(int &),但这会改变您允许调用它的方式。您可能仍需要辅助函数,但您可以将所有代码放入其中。