剑指offer--62.二叉搜索树的第k个结点

题目描述

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

时间限制：1秒 空间限制：32768K 热度指数：157331

思路

中序遍历，递归

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if(pRoot!=NULL)//当结点非空
        {
            TreeNode* ans =KthNode(pRoot->left, k);//访问左子树
            if(ans!=NULL)//若不为NULL,则说明已经找到
                return ans;
            ++count;        //计数
            if(count==k)
                return pRoot;
            ans =KthNode(pRoot->right, k);//访问右子树
             if(ans!=NULL)//若不为NULL,则说明已经找到
                return ans;
        }
        //左右子树不包括第K个结点时，返回NULL
            return NULL;
        
    }
private:
    int count=0;//计数
    
    
};

非递归

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
//中序遍历非递归
class Solution {
public:
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if(k==0||pRoot==NULL)
            return NULL;
        stack<TreeNode*> p;
        int cnt=0;//计数器
        TreeNode* t=pRoot;
        while(t||!p.empty())
        {
            if(t)
            //当有左子树的时候，不断将左子树入栈
            {
                p.push(t);
                t=t->left;
            }
            else //没有左子树时或当前结点为空时，返回父母结点，访问其右子树
            {
                //自增和判断是否到达k，即中序遍历中的访问结点内容
                ++cnt;
                t=p.top();
                p.pop();
                if(cnt==k)
                    return t;
                
                //中序访问右子树
               t=t->right;
            }
        }
        return NULL;
    }

};