二叉搜索树中第k大和第k小结点

最新推荐文章于 2024-09-14 23:40:40 发布

VipPeterGee

最新推荐文章于 2024-09-14 23:40:40 发布

阅读量244

点赞数

分类专栏： LeetCode 文章标签：数据结构算法二叉树 leetcode

本文链接：https://blog.csdn.net/sinat_35241409/article/details/123393808

版权

LeetCode 专栏收录该内容

58 篇文章 1 订阅

订阅专栏

题目

查找二叉搜索中第k大和第k小结点

解法

构建二叉树

public class BSTNode<T extends Comparable<T>> {
    T key;
    BSTNode<T> left;
    BSTNode<T> right;
    BSTNode<T> parent;

    public BSTNode(T key, BSTNode<T> parent, BSTNode<T> left, BSTNode<T> right) {
        this.key = key;
        this.parent = parent;
        this.left = left;
        this.right = right;
    }
}

中序遍历

   private static void inOrder(BSTNode<Integer> tree, ArrayList<BSTNode> nodeList) {
        if (tree != null) {
            inOrder(tree.left, nodeList);
            nodeList.add(tree);
            inOrder(tree.right, nodeList);
        }

    }

查找第k大结点

 /**
     * 查找第k大结点
     *
     * @param tree
     * @param k    第几个
     * @return
     */
    public static BSTNode findMaxKNode(BSTNode<Integer> tree, int k) {
        if (tree == null || k < 1) return null;
        // 中序遍历tree
        ArrayList<BSTNode> nodeList = new ArrayList();
        inOrder(tree, nodeList);
        if (nodeList.size() < k) return null;
        return nodeList.get(k-1);
    }

查找第k小结点

 /**
     * 查找第k小结点
     *
     * @param tree
     * @param k    第几个
     * @return
     */
    public static BSTNode findMinKNode(BSTNode<Integer> tree, int k) {
        if (tree == null || k < 1) return null;
        // 中序遍历tree
        ArrayList<BSTNode> nodeList = new ArrayList();
        inOrder(tree, nodeList);
        if (nodeList.size() < k) return null;
        return nodeList.get((nodeList.size()-k));
    }