剑指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=0;
            }
            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;
    }
};