二叉搜索树的第k个结点

二叉搜索树的第k个结点

---

题目描述
给定一颗二叉搜索树，请找出其中的第k大的结点。例如， 5 / \ 3 7 /\ /\ 2 4 6 8 中，按结点数值大小顺序第三个结点的值为4。

---

思路：

1. 二叉搜索树为关键字排序：左子树<根<右子树
2. 二叉树的中序遍历的访问顺序为：左子树<根<右子树
3. 故找出二叉搜索树中第K大的节点实质为找出中序遍历的第K个节点
4. （1）当二叉树为空树（2）当二叉树中节点个数小于K个

---

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
    int count=0;
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        int count=0;
if(pRoot==NULL)
    return NULL;
    //利用栈实现中序遍历
        stack<TreeNode *>s;
        TreeNode *target=NULL;
        while(pRoot!=NULL||s.empty()==false)
        {
            if(pRoot)
            {
                s.push(pRoot);
                pRoot=pRoot->left;
            }
            else{
                pRoot=s.top();
                s.pop();
                count++;//记录当前节点数，当等于K时，返回当前节点
                    if(count==k){
                        target=pRoot;
                        break;
                    }
                pRoot=pRoot->right;
            }
        }
        pRoot=target;
        return pRoot;
    }


};