给定一颗二叉搜索树,返回该二叉搜索树第K大的节点
思路:由于二叉搜索树的中序遍历序列是从小到大的序列,模拟这一过程每访问一个节点,count++直到第K个
注意点:
- 由于递归的进入与返回过程,递归调用栈频繁进出,计数器可设置成全局的变量
- 递归的返回条件是NULL 注意特殊情况
class Solution {
public:
int count=0;//计数器 注意设置成全局,避免放入递归调用栈
TreeNode* KthNode(TreeNode* pRoot, int k)
{
if(pRoot==NULL)
return NULL;
{
TreeNode* left=KthNode(pRoot->left,k);
if(left!=NULL)
return left;
count++;
if(count==k)
return pRoot;
TreeNode* right=KthNode(pRoot->right,k);
if(right!=NULL)
return right;
}
return NULL;//没有找到,可能是K过大
}
};
变种