剑指offer 面试题. 二叉搜索树的第k个结点

题目描述

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

 

 

解：

由于二叉搜索树的中序遍历是升序，所以在中序基础上添加计数器即可。

 class Solution {
 public:
     TreeNode* KthNode(TreeNode* pRoot, int k)
     {
         int i=INT32_MIN;
         return func(pRoot,k,i);
     }
     TreeNode* func(TreeNode* node,int k,int& i){
         if(node==nullptr){
             if(i==INT32_MIN){
                 i=;
             }
             return nullptr;
         }
         decltype(node) t;
         if(t=func(node->left,k,i)){
             return t;
         }
         ++i;
         if(i==k){
             return node;
         }
         if(t=func(node->right,k,i)){
             return t;
         }
         return nullptr;
     }
 };