算法通关村第七关——迭代实现二叉树遍历

迭代法实现中序遍历

中序遍历先访问当前节点，再一层一层向下找左子树，找到最底部，再进行处理。

在迭代法中，使用栈来处理节点上的元素。

public List<Integer> inorderTraversal(TreeNode root){
    List<Integer> res = new ArrayList<Integer>();
    Deque<TreeNode> stack = new LinkedList<TreeNode>();
    while (root != null || !stack.isEmpty()){
        while (root != null){
            stack.push(root);
            root = root.left;
        }
        root = stack.pop();
        res.add(root.val);
        root = root.right;
    }
    return res;
}

首先，定义一个结果列表，在定义一个栈，进入循环

先将从root开始到最左下角的那个节点，按顺序全部压入栈中。因为栈是先进后出的结构。

现在的节点是整个树中最左下角的那个节点，将这个节点弹出，加入结果列表，让root=root.right，然后进入下一层循环。

如果root.right是null的话，因为现在栈中不为空，所以还是会进入循环，但是不会进入压栈的那个循环，这是往结果列表中压入的就是刚开始处理的那个节点的父节点。