码迷,mamicode.com
首页 > 其他好文 > 详细

leetcode 230. Kth Smallest Element in a BST

时间:2019-08-18 11:35:11      阅读:65      评论:0      收藏:0      [点我收藏+]

标签:tac   public   节点   tree   中序   null   计数   ++   height   

技术图片

 

中序遍历,计数器数到k时,输出当前节点值

class Solution {
    private int count;
    public int kthSmallest(TreeNode root, int k) {
        count = 0;
        Stack<TreeNode> stack = new Stack<>();
        
        while(root != null){
            stack.push(root);
            root = root.left;
        }
        
        while(!stack.isEmpty()){
            TreeNode cur = stack.pop();
            count++;
            if(count == k)
                return cur.val;
            cur = cur.right;
            while(cur != null){
                stack.push(cur);
                cur = cur.left;
            }
        }
        
        return 0;
    }
}

 

leetcode 230. Kth Smallest Element in a BST

标签:tac   public   节点   tree   中序   null   计数   ++   height   

原文地址:https://www.cnblogs.com/hwd9654/p/11371552.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!