【优先队列-最小堆法，beat 97%】 | #二叉搜索树的第k个结点#

思路

核心就是得熟悉优先队列比较器的改写方法
① 建立最小堆存储节点：根据节点val升序排列；
② 输出结果。

代码

import java.util.*;
public class Solution {
    // 建立最小堆存储节点：根据节点val升序排列
    PriorityQueue<TreeNode> heap = new PriorityQueue<>(new Comparator<TreeNode>(){
        public int compare(TreeNode n1,TreeNode n2){
            return n1.val-n2.val; // 根据节点val升序排列
        }
    });
    // 主函数
    TreeNode KthNode(TreeNode pRoot, int k) {
        if(pRoot == null || k < 1) return null;
        // 1、保存所有节点
        saveNode(pRoot);
        TreeNode res = pRoot;
        // 2、输出结果
        while(k-- > 0) res = heap.poll();
        return res;
    }
    // 保存节点
    private void saveNode(TreeNode pRoot){
        if(pRoot != null){
            heap.add(pRoot);
            saveNode(pRoot.left);
            saveNode(pRoot.right);
        }
    }
}