题目:
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。
思路:
按照中序遍历的顺序遍历一棵搜索二叉树,则数值是递增顺序的,接着找到第k大即可。
class Solution {
public:
void midOrdered(vector<TreeNode*>& help,TreeNode* pRoot){
if(pRoot->left) midOrdered(help, pRoot->left);//左子树
help.push_back(pRoot);//根节点
if(pRoot->right) midOrdered(help,pRoot->right);//右子树
}
TreeNode* KthNode(TreeNode* pRoot, int k)
{
if(pRoot==nullptr||k<0) return nullptr;
vector<TreeNode *> help;
midOrdered(help,pRoot);//对于二叉搜索树,中序遍历后即是排序数组
int count=0;
for(const auto val:help){//寻找第k个元素即可
count++;
if(count==k)
return val;
}
return nullptr;
}
};