NC81 二叉搜索树的第K个节点(dfs)

最新推荐文章于 2022-05-11 17:03:23 发布

weixin_46590256

最新推荐文章于 2022-05-11 17:03:23 发布

阅读量247

点赞数

分类专栏：数据结构与算法文章标签： dfs

本文链接：https://blog.csdn.net/weixin_46590256/article/details/121919244

版权

数据结构与算法专栏收录该内容

78 篇文章 0 订阅

订阅专栏

本文介绍了如何在二叉搜索树中找到第k小的节点。通过中序遍历二叉搜索树，可以得到升序排列的节点值。在给定的二叉树示例中，利用中序遍历找到第3小的节点值为4。解题思路涉及递归的中序遍历和数组存储节点值，最终返回第k个节点的值。

摘要由CSDN通过智能技术生成

描述

给定一棵结点数为n 二叉搜索树，请找出其中的第 k 小的TreeNode结点值。
如输入{5,3,7,2,4,6,8},3时，二叉树{5,3,7,2,4,6,8}如下图所示：
在这里插入图片描述
该二叉树所有节点按结点值升序排列后可得[2,3,4,5,6,7,8]，所以第3个结点的结点值为4，故返回对应结点值为4的结点即可。

解题思路

由于是二叉搜索树，采用中序遍历得到的第k个节点即第k小的值，判断输出是否是第k个就可以返回值了

class Solution {
public:
    vector<int>res;
    void InOrder(TreeNode* root){
        if(root){
            InOrder(root->left);
            res.push_back(root->val);
            InOrder(root->right);
        }
    }
    int KthNode(TreeNode* proot, int k) {
       if(!proot) return -1;
        InOrder(proot);
        if(k==0||k>res.size()) return -1;
        else return res[k-1];
    }
};