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

最新推荐文章于 2021-12-20 10:08:12 发布

不懂Coding的小白

最新推荐文章于 2021-12-20 10:08:12 发布

阅读量111

点赞数

文章标签：二叉树 leetcode 算法数据结构

本文链接：https://blog.csdn.net/Mr_Jiax/article/details/109860785

版权

解题思路：因为题目要求找到搜索二叉树第k大节点，搜索二叉树的特点是当前节点比左子树所有节点大，比右子树所有节点小，所以搜索二叉树中序遍历递增

本题等价于中序遍历倒序第k个节点

(1)利用非递归方法实现中序遍历，然后过程中k的值递减，当k减为0之前，返回当前节点的值。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int kthLargest(TreeNode root, int k) {
        if(root!=null){//右 中 左
            Stack<TreeNode> stack =new Stack<>();
            while(!stack.empty()||root!=null){
                if(root!=null){
                    stack.push(root);
                    root=root.right;
                }else{
                    root=stack.pop();
                    if(--k==0){break;}
                    root=root.left;
                }
            }
        }
        return root.val;
    }
}