二叉搜索树的第k个结点

最新推荐文章于 2020-06-11 13:53:01 发布

快乐的小J

最新推荐文章于 2020-06-11 13:53:01 发布

阅读量81

点赞数

分类专栏：算法基本功剑指offer（牛客）

本文链接：https://blog.csdn.net/muzhixi/article/details/99883956

版权

算法基本功同时被 2 个专栏收录

106 篇文章 1 订阅

订阅专栏

剑指offer（牛客）

63 篇文章 0 订阅

订阅专栏

思路：二叉搜索树的中序遍历是有序的。

    ArrayList<TreeNode> midTraceForKthNode(TreeNode pRoot,int k){
        ArrayList<TreeNode> sortedList=new ArrayList<>();
        if(pRoot==null||sortedList.size()==k) return sortedList;
        //将每一个中间结果的size与k比较，相等立即返回
        sortedList.addAll(midTraceForKthNode(pRoot.left,k));
        if(sortedList.size()==k) return sortedList;
        sortedList.add(pRoot);
        if(sortedList.size()==k) return sortedList;
        sortedList.addAll(midTraceForKthNode(pRoot.right,k));
        return sortedList;
    }
    
    TreeNode KthNode(TreeNode pRoot, int k)
    {
        if(pRoot==null||k<=0) return null;
        ArrayList<TreeNode> sortedList=midTraceForKthNode(pRoot,k);
        if(sortedList.size()<k) return null;
        else return sortedList.get(k-1);
    }