c++实现---二叉树的第k个结点

题目:
给定一棵二叉搜索树,请找出其中的第k小的结点。例如, (5,3,7,2,4,6,8) 中,按结点数值大小顺序第三小结点的值为4。
思路:
按照中序遍历的顺序遍历一棵搜索二叉树,则数值是递增顺序的,接着找到第k大即可。

class Solution {
public:
    void midOrdered(vector<TreeNode*>& help,TreeNode* pRoot){
        if(pRoot->left) midOrdered(help, pRoot->left);//左子树
        help.push_back(pRoot);//根节点
        if(pRoot->right) midOrdered(help,pRoot->right);//右子树
    }
    TreeNode* KthNode(TreeNode* pRoot, int k)
    {
        if(pRoot==nullptr||k<0) return nullptr;
        vector<TreeNode *> help;
        midOrdered(help,pRoot);//对于二叉搜索树,中序遍历后即是排序数组
        int count=0;
        for(const auto val:help){//寻找第k个元素即可
            count++;
            if(count==k)
                return val;
        }
        return nullptr;
    }
};

猜你喜欢

转载自blog.csdn.net/Fizz6018/article/details/107198128