题解 | #二叉树之寻找第k大#

二叉树之寻找第k大

https://www.nowcoder.com/practice/8e5f73fa3f1a407eb7d0b0d7a105805e

知识点:二叉搜索树,中序遍历

二叉搜索树的性质是当前节点的左子树节点值均小于当前节点值,右子树节点值均大于当前节点值,而中序遍历刚好是所有节点顺序排列的结果。

要想获得第k大的节点值,我们可以将中序遍历的顺序反过来,先遍历右节点,然后遍历当前节点,最后遍历左节点,这样得到的就是逆序的节点排列,当遍历到第k个元素时,即为第k大的元素。

Java题解如下

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @param k int整型 
     * @return int整型
     */
    private int val;
    private int count;
    private int k;
    
    public int kthLargest (TreeNode root, int k) {
        // write code here
        this.k = k;
        count = 0;
        order(root);
        return val;
    }

    private void order(TreeNode node) {
        if(node == null) {
            return;
        }
        order(node.right);
        count++;
        if(count == k) {
            val = node.val;
            return;
        }
        order(node.left);
    }
}

全部评论

相关推荐

群星之怒:不是哥们,你就不好奇瘫痪三十年的老植物人是啥样的吗?
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务