【剑指 Offer 54. 二叉搜索树的第k大节点】

二叉树中第k个最大元素的查找

原创于 2022-09-05 19:43:06 发布 · 141 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度优先 #算法

JAVA 专栏收录该内容

126 篇文章

订阅专栏

该博客介绍了如何通过先序遍历和排序来找到二叉树中第k个最大的元素。提供的Java代码实现首先将二叉树的节点值存入ArrayList，然后排序，最后返回排序后列表中第k个元素，即为所求。

【解题思路】

先将二叉树中的数字放在数组中，对数组进行排序。对二叉树使用先序遍历。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    ArrayList<Integer> list = new ArrayList<Integer>();

    public int kthLargest(TreeNode root, int k) {
        if(root == null) return 0;
        DFS(root);

        Collections.sort(list);
        return list.get(list.size()-k);
    }

    public void DFS(TreeNode p)
    {
        list.add(p.val);
        if(p.left != null) DFS(p.left);
        if(p.right != null) DFS(p.right);
    }
}