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;
    }
发布了96 篇原创文章 · 获赞 2 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/reachwang/article/details/103532155