1. 题目描述
&esmp;&esmp;给定一颗二叉搜索树,请找出其中的第k大的结点。例如, 5 / \ 3 7 /\ /\ 2 4 6 8 中,按结点数值大小顺序第三个结点的值为4。
2. 思路分析
&esmp;&esmp;首先一点要清晰二叉搜索树的特点,也就是:左孩子<父节点<右孩子。所以,根据这一特点可知,二叉搜索树的中序遍历结果就是一个升序的序列。题目中的实例可知,是将二叉搜索树的元素升序排列后找到第k个元素。
&esmp;&esmp;思路就是从根节点一直遍历左孩子,如果左子树的遍历中有遍历结果,就返回结果,如果结果NULL则进行下一步判断 –> 首先对计数器加一(即NULL的叶子节点为零),然后判断当前节点是不是目标节点,通过比较计数器cnt和k,如果相等,当前节点极为目标节点,如果不等就对其右子树进行遍历,返回右子树的遍历结果即可。
3. 解答(c++)
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};
*/
class Solution {
public:
int cnt = 0;
TreeNode* KthNode(TreeNode* pRoot, int k)
{
if (pRoot == NULL || k <= 0)
return NULL;
//if (k == cnt)
//return pRoot;
TreeNode* answer;
if (pRoot->left != NULL) {
answer = KthNode(pRoot->left, k);
}
if (answer != NULL)
return answer;
++cnt;
if (cnt == k)
return pRoot;
answer = KthNode(pRoot->right, k);
if (answer != NULL)
return answer;
return NULL;
}
};