剑指offer之二叉搜索树的第k个结点

题目描述

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

解题思路

二叉搜索树的一个重要性质就是它的中序遍历是排序的，因此这道题目只需要用中序遍历算法遍历一棵二叉搜索树，就很容易找出它的第K大结点。
中序遍历用递归实现比较容易，但要想清楚的是遍历到一个根结点的时候要做的是什么？
中序遍历二叉树打印的时候我们在递归完左子树之后打印根结点，本题目要求的当然不是打印，如果左子结点不是要找的结点，才会访问根结点，所以访问到根结点的时候要做的操作是将K减去1，因为左子结点已经证实不是要找的结点了，排除左子结点。
这个过程可以看成目标移位的过程，每移过一个结点，K减1，直到K等于1时，当前结点就是要求的结点。

非递归版：

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
    int count = 0;
public:
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if(pRoot == NULL || k == 0)
            return NULL;
        return KthNodeCore(pRoot, k);
    }

    TreeNode* KthNodeCore(TreeNode* pRoot, int& k){
        TreeNode* target = NULL;
        // 先成左子树中找
        if(pRoot->left != NULL)
            target = KthNodeCore(pRoot->left, k);
        // 如果在左子树中没有找到
        if(target == NULL){
            //说明当前的根结点是所要找的结点
            if(k == 1)
                target = pRoot;
            // 当前的根结点不是要找的结点，但是已经找过了，所以计数器减一
            else
                k--;
        }
        // 根结点以及根结点的左子树都没有找到，则找其右子树
        if(target == NULL && pRoot->right != NULL)
            target = KthNodeCore(pRoot->right, k);
        return target;
    }
};

递归版：

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
    int count = 0;
public:
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if(pRoot){
            TreeNode *ret = KthNode(pRoot->left, k); //先成左子树中找
            if(ret) return ret;  // 左子树中找到符合要求的节点返回
            if(++count == k) return pRoot; //从最左节点开始，count+，当++count == k时说明当前的根结点是所要找的结点
            ret = KthNode(pRoot->right,k); //根结点以及根结点的左子树都没有找到，则找其右子树
            if(ret) return ret;
        }
        return nullptr;
    }
};