剑指offer——二叉搜索树的第k个节点

1. 题目描述

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

2. 思路分析

&esmp;&esmp;首先一点要清晰二叉搜索树的特点，也就是：左孩子<父节点<右孩子。所以，根据这一特点可知，二叉搜索树的中序遍历结果就是一个升序的序列。题目中的实例可知，是将二叉搜索树的元素升序排列后找到第k个元素。
&esmp;&esmp;思路就是从根节点一直遍历左孩子，如果左子树的遍历中有遍历结果，就返回结果，如果结果NULL则进行下一步判断 –> 首先对计数器加一（即NULL的叶子节点为零），然后判断当前节点是不是目标节点，通过比较计数器cnt和k，如果相等，当前节点极为目标节点，如果不等就对其右子树进行遍历，返回右子树的遍历结果即可。

3. 解答（c++）

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
    int cnt = 0;
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if (pRoot == NULL || k <= 0)
            return NULL;
        //if (k == cnt)
            //return pRoot;
        TreeNode* answer;
        if (pRoot->left != NULL) {
            answer = KthNode(pRoot->left, k);
        }
        if (answer != NULL)
            return answer;
        ++cnt;
        if (cnt == k)
            return pRoot;
        answer = KthNode(pRoot->right, k);
        if (answer != NULL)
            return answer;
        return NULL;
    }
};