来自LeetCode上非常经典的二叉树遍历问题,leetcode94
描述:给你二叉树的根节点 root
,返回它节点值的 前序 遍历。
这个问题,首先我们可能是先想到递归法,理解起来也比较容易。如果不让我们使用递归呢,那就不得不介绍下第二种方法:迭代法。
理论上,递归能做的迭代一定能做,但可能会比较复杂。
递归就是每次执行方法调用都会先把当前的局部变量、参数值和返回地址等压入栈中,后面在递归返回的时候,从栈顶弹出上一层的各项参数继续执行,这就是递归为什么可以自动返回并执行上一层方法的原因。
前序遍历是中左右,如果还有左子树就一直向下找。完了之后再返回从最底层逐步向上向右找。于是思路就有啦,我们定义一个栈用来存储遍历过的二叉树元素。从根节点开始,边遍历,边添加到集合中,然后继续找它的左节点。直到左节点为空,就将它出栈然后找它的右节点。直到二叉树全部遍历完,并且全部出栈。
代码实现:
public List<Integer> preOrderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
if (root == null) {
return res;
}
Stack<TreeNode> stack = new Stack<TreeNode>();
TreeNode node = root;
//第一次栈肯定为空,为了继续while循环,因此需要加入条件node!=null
while (!stack.isEmpty() || node != null) {
while (node != null) {
res.add(node.val);
stack.push(node);
node = node.left;
}
node = stack.pop();
node = node.right;
}
return res;
}