栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > C/C++/C#

二叉搜索树的第k个节点(C++)

C/C++/C# 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

二叉搜索树的第k个节点(C++)

二叉搜索树的第k个节点 描述

给定一棵结点数为n 二叉搜索树,请找出其中的第 k 小的TreeNode结点值。

    返回第k小的节点值即可不能查找的情况,如二叉树为空,则返回-1,或者k大于n等等,也返回-1保证n个节点的值不一样

数据范围:

进阶:空间复杂度 O(n)O(n),时间复杂度 O(n)O(n)

如输入{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的结点即可。

示例1
输入:
{5,3,7,2,4,6,8},3
返回值:
4
示例2
输入:
{},1
返回值:
-1

备注:当树为空的时候。

解法思路

  通过二叉树的中序遍历将节点进行排序,存放至线性结构(数组)中即可。

解法1
class Solution {
  public:
    
    vector arr;
    //通过递归来对二叉树进行中序遍历
    void Inorder(TreeNode* pHead) {
         if(pHead)
         {
             Inorder(pHead->left);
             arr.push_back(pHead->val);
             Inorder(pHead->right);
         }
    }
    int KthNode(TreeNode* proot, int k) {
        if (nullptr == proot ||
                (proot->left == nullptr && proot->right == nullptr && k != 1) ||
                k == 0)return -1;
        Inorder(proot);
        if (arr.size() < k)return -1;
        return arr[k - 1];
    }
};
解法2
class Solution {
  public:
    
    vector arr;
    //通过栈结构实现对二叉树的中序遍历
    void Inorder(TreeNode* pHead) {
        stack stack;
        TreeNode* tempNode, *pStart = pHead;
        while (pStart || !stack.empty()) {
            if (pStart) {
                stack.push(pStart);
                pStart = pStart->left;
            } else {
                tempNode = stack.top();
                arr.push_back(tempNode->val);
                stack.pop();
                pStart = tempNode->right;
            }
        }
    }
    int KthNode(TreeNode* proot, int k) {
        if (nullptr == proot ||
                (proot->left == nullptr && proot->right == nullptr && k != 1) ||
                k == 0)return -1;
        Inorder(proot);
        if (arr.size() < k)return -1;
        return arr[k - 1];
    }
};
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/769039.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号