剑指offer: 二叉搜索树的第K个节点

题目描述

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

 

解题思路:

对于二叉搜索树,中序遍历即使从小到大的排序,同时用一个变量记录,当遍历到第k个的时候就返回。

class Solution {
public:
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if(pRoot != NULL) {
            TreeNode* node = KthNode(pRoot->left,k);
            if(node != NULL)
                return node;
            
            index++;
            if(index == k)        //遍历到目标节点,返回
                return pRoot;
            
            node = KthNode(pRoot->right,k);
            if(node != NULL)
                return node;
        }
        
        return NULL;
    }

private:
    int index = 0;
};

 

©️2020 CSDN 皮肤主题: 像素格子 设计师: CSDN官方博客 返回首页
实付0元
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值