Java 给定一棵二叉搜索树,请找出其中的第k小的结点

import java.util.*;
public class Solution {
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot==null||k==0)
            return null;
        LinkedList<Integer> queue = new LinkedList<>();
        queue = midOrder(pRoot);
        if(k>queue.size())
            return null;
        int result = 0;
        while(k>0){
            k--;
            result = queue.poll();
        }
        TreeNode nodeK = new TreeNode(result);
        return nodeK;
    }
    LinkedList<Integer> midOrder(TreeNode root){
        LinkedList<Integer> queue = new LinkedList<>();
        if(root==null)
            return queue;
        if(root.left!=null){
            midOrder(root.left);
        }
        queue.offer(root.val);
        if(root.right!=null){
            midOrder(root.right);
        }
        return queue;
    }
}
我的想法是用一个queue将二叉搜索树的中序遍历结果存起来,那queue里面存的就是从小到大的值。但是利用poll()函数知道第k个。
但是跑的结果显示,当k=1时,返回的是根节点的值。很迷惑,还望牛友指点。
不通过
您的代码已保存
答案错误:您提交的程序没有通过所有的测试用例
case通过率为3.57%
用例:
{8,6,10,5,7,9,11},1
对应输出应该为:
5
你的输出为:
8

#Java#
全部评论
为啥不维护一个count变量呢,走到一个节点加一,到了k就返回
点赞
送花
回复
分享
发布于 2019-08-29 16:12
你这中序遍历存入queue的逻辑是不是有点问题?
点赞
送花
回复
分享
发布于 2019-08-29 16:20
滴滴
校招火热招聘中
官网直投
楼主我觉得错误 是你中序遍历的queue那,把queue写在方法参数那就行,或者将queue外提出来
点赞
送花
回复
分享
发布于 2019-08-29 16:31
你的树节点能保存树的大小就好了,直接递归就行
点赞
送花
回复
分享
发布于 2019-08-29 16:43
class Solution {     LinkedList<TreeNode> queue = new LinkedList<>();     public TreeNode KthNode(TreeNode pRoot, int k) {         if (pRoot == null || k == 0)             return null;         midOrder(pRoot);         return queue.get(k - 1);     }     private void midOrder(TreeNode root) {         if (root == null) return;         midOrder(root.left);         queue.add(root);         midOrder(root.right);     } } 老哥,我根据你的代码改了改,觉得有这几个问题: 1.递归时使用的List应该是全局变量,不能递归一次new一个 2.因为要求返回TreeNode,所以List中可以直接储存TreeNode,不用保存节点值,省去了最后返回时构建节点 3.这里其实没有必要使用List储存节点,直接设定一个计数器和全局变量,当计数器为k时使用全局变量保存遍历的节点即可 4.中序递归时设定了root == null返回的边界条件,就不需要重复判断左右节点是否为null了哈
点赞
送花
回复
分享
发布于 2019-08-29 16:44

相关推荐

点赞 2 评论
分享
牛客网
牛客企业服务