剑指offer | 训练题61：二叉树搜索的第K个结点

最新推荐文章于 2020-02-22 01:47:48 发布

rical730

最新推荐文章于 2020-02-22 01:47:48 发布

阅读量258

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/rical730/article/details/74612909

版权

算法专栏收录该内容

69 篇文章 0 订阅

订阅专栏

题目描述

给定一颗二叉搜索树，请找出其中的第k大的结点。例如，

5
/ \
3 7
/\ /\
2 4 6 8

中，按结点数值大小顺序第三个结点的值为4。

思路

/**
 * 思路：按照中序遍历的方法遍历，同时设置一个全局变量，每遍历一个节点就递增一次，直到到达需要的结点
 */
/*
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) return NULL;
        TreeNode* tre;
        tre = KthNode(pRoot->left, k);
        if(tre != NULL) return tre;
        cnt++;
        if(cnt == k) return pRoot;
        tre = KthNode(pRoot->right, k);
        if(tre != NULL) return tre;
        else return NULL;
    }
};