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

题目描述

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

 

解题思路：使用中序遍历进行遍历，得到的就是按照顺序的，当遍历到第k个就输出即可。

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
    //实现中序遍历,对于二叉搜索树中序遍历是按节点从小到大排序
    TreeNode *KthNodeCore(TreeNode *root, int &k)//注意此处k为引用，因为在操作过程中k改变
    {
        TreeNode *target = NULL;
        if(root->left != NULL)
        {
            target = KthNodeCore(root->left,k);
        }
        if(target == NULL)//无左节点且还未找到，返回其叶子节点的父节点，k--
        {
            if(k == 1)
                target = root;
            k--;
        }
        if(target == NULL && root->right != NULL)//找右结点
        {
            target = KthNodeCore(root->right,k);
        }
        return target;
    }
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if(pRoot == NULL || k == 0)
            return NULL;
        return KthNodeCore(pRoot,k);
    }
};