二叉搜索树的第k小个结点

题目描述
给定一棵二叉搜索树，请找出其中的第k小的TreeNode结点。
示例1
输入 {5,3,7,2,4,6,8},3
返回值 {4}
说明 按结点数值大小顺序第三小结点的值为4

题目中并未给出 k 的取值范围，k 有可能比给定的二叉树节点要多，也有可能是0个。如果是0个的话，则返回nullptr，如果k比给定的二叉树节点要多的话，那么也返回nullptr。在dfs 这个函数中，k 使用的是引用，ans 保存的是 k ==0 时的节点，dfs 递归结束 k 有可能小于0，因此当 k <= 0 时，返回ans。如果遍历结束 k 仍然是 大于0，那么说明此时 k 的节点数大于二叉树的节点树，因此需要返回nullptr。
dfs 使用的是中序遍历，二叉搜索树中序遍历也就是按递增序列。如果题目 转化为 求第 k 大个节点，则需要使用中序遍历的倒序。如LeetCode 54

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
    TreeNode* ans;
   
    TreeNode* KthNode(TreeNode* pRoot, int k) { 
      if(k==0) return nullptr;
      dfs(pRoot,k);
      if(k<=0) return ans;
      else return nullptr;  
    }
    void dfs(TreeNode* root,int &k)
    {
        if(!root) return;
        dfs(root->left,k);
        k--;
        if(!k) ans=root; 
        dfs(root->right,k);
        
    }
};