题目描述
给定一颗二叉搜索树,请找出其中的第k大的结点。例如, 5 / \ 3 7 /\ /\ 2 4 6 8 中,按结点数值大小顺序第三个结点的值为4。
分析:中序遍历二叉树,输出第k-1个节点。
代码:
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
class Solution {
public:
void helper(TreeNode* pRoot){
if(pRoot==NULL) return;
if(pRoot->left) helper(pRoot->left);
v.push_back(pRoot);
if(pRoot->right) helper(pRoot->right);
}
TreeNode* KthNode(TreeNode* pRoot, unsigned int k)
{
if(pRoot==NULL||k==0) return NULL;
helper(pRoot);
if(k>v.size()) return NULL;
return v[k-1];
}
private:
vector<TreeNode*> v;
};