二叉搜索树的第k个节点

二叉搜索树的第k个结点

http://www.nowcoder.com/questionTerminal/ef068f602dde4d28aab2b210e859150a

一是利用大家都能想到的中序遍历递归法:

public class Solution {
    List<TreeNode> list = new ArrayList<>();
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot == null || k < 1){
            return null;
        }
        minOrder(pRoot, k);
        //当k超过节点个数时
        if(list.size() < k)
            return null;
        return list.get(k-1);
    }

    void minOrder(TreeNode pRoot, int k){
        if(pRoot != null){
            minOrder(pRoot.left, k);
            list.add(pRoot);
            k = k - 1;
            if(k == 0)
                return;
            minOrder(pRoot.right, k);
        }
    }
}

二是利用栈来弹出第k个节点:
public class Solution {
    List<TreeNode> list = new ArrayList<>();
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot == null || k < 1){
            return null;
        }
        TreeNode node = pRoot;
        int count = 0;
        Stack<TreeNode> stack = new Stack<>();

        while(!stack.isEmpty() || node != null){
            //中序遍历
            //遍历到左子树为空
            if(node != null){
                stack.push(node);
                node = node.left;
            }else{
                //再弹出去
                node = stack.pop();
                count++;
                //当弹到第k个时,直接返回
                if(count == k)
                    return node;
                //否则,遍历右子树
                node = node.right;
            }
        }
        return null;
    }

}
全部评论

相关推荐

06-18 13:28
已编辑
门头沟学院 Web前端
爱睡觉的冰箱哥:《给予你300的工资》,阴的没边了
点赞 评论 收藏
分享
Twilight_m...:表格简历有点难绷。说说个人看法: 1.个人基本情况里好多无意义信息,什么婚姻状况、健康状况、兴趣爱好、户口所在地、身份证号码、邮政编码,不知道的以为你填什么申请表呢。 2.校内实践个人认为对找工作几乎没帮助,建议换成和测开有关的项目,实在没得写留着也行。 3.工作经历完全看不出来是干什么的,起码看着和计算机没啥关系,建议加强描述,写点你在工作期间的实际产出、解决了什么问题。 4.个人简述大而空,看着像AI生成,感觉问题最大。“Python,C,C++成为我打造高效稳定服务的得力工具”、“我渴望凭借自身技术知识与创新能力,推动人工智能技术的应用发展,助力社会实现智能化转型”有种小学作文的美感。而且你确定你个人简述里写的你都会嘛?你AI这块写的什么“深入研究”,发几篇顶会的硕博生都不一定敢这么写。而且你AI这块的能力和软测也完全无关啊。个人简述建议写你对哪些技术栈、哪些语言、哪些生产工具的掌握,写的有条理些,而且最好是和测开强相关的。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务