这个题目我知道可以使用前序的非递归遍历,然后遍历的过程中改变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;
}