剑指offer61-二叉搜索树的第k个节点

给定一棵二叉搜索树，请找出其中的第k小的结点。例如， （5，3，7，2，4，6，8）    中，按结点数值大小顺序第三小结点的值为4。

思路：中序遍历第k个节点

    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        //中序遍历
        int count=0;
        TreeNode*p=pRoot;
        stack<TreeNode*>s;
        if(pRoot==NULL) return p;

        do{
            while(p!=NULL)
            {
                s.push(p);
                p=p->left;
            }
            p=s.top();
            s.pop();
            count++;
            if(count==k) return p;
                p=p->right;
        }while(p!=NULL||!s.empty());
        return p;
    }