/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param proot TreeNode类
* @param k int整型
* @return int整型
*/
int KthNode(TreeNode* proot, int k) {
int count = 0;
return KthNode2(proot, k, count);
}
int KthNode2(TreeNode *proot, int k, int &count) {
if (!proot) {
return -1;
}
int l = KthNode2(proot->left, k, count);
if (l != -1) {
return l;
}
++count;
if (count == k) {
return proot->val;
}
return KthNode2(proot->right, k, count);
}
};
思路:二叉搜索树中序遍历即为顺序遍历,所以中序遍历并计数即可。

京公网安备 11010502036488号