leetcode_230:二叉搜索树中第K小的元素

这个题目我知道可以使用前序的非递归遍历,然后遍历的过程中改变k,直到找到为止。中序的遍历的非递归实现其实不难,我们只要缕清中序遍历的过程就很好写,前序遍历也就是左—>根—>右的顺序,那么我们可以先将根节点和其左子树进栈,然后在出栈的时候看看它的右子树是不是空,不为空则把右子树作为根节点进栈,并将此根节点的所有左子树进栈。这样就完成了先遍历左子树,然后根节点,最后右子树的过程。代码如下:

    public static int kthSmallest(TreeNode root, int k) {
        Stack<TreeNode> stack = new Stack<>();
        TreeNode p = root;
        while (p != null) {		//先把所有的左子树进栈
            stack.push(p);
            p = p.left;
        }
        //前序非递归遍历,k>1就是说我们只遍历前k-1个,然后到第k个的时候也就是栈顶的元素
        while (!stack.empty() && k > 1) {	
            TreeNode curNode = stack.pop();
            //当前根节点右子树不为null,则把右子树和其以它为根节点的左子树也放进去
            if (curNode.right != null) {	
                p = curNode.right;
                while (p != null) {
                    stack.push(p);
                    p = p.left;
                }
            }
            --k;
        }
        return stack.pop().val;
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值