二叉树的遍历

面试中经常会考的一道题目,就是二叉树的遍历,很简单,你会说“使用递归,根据要求(前序、中序、后序,以中序为例)依次对节点和其左右子节点调用递归方法。

public void printNode(Node node) {
    if (null != node.left)
        printNode(node.lef);
    System.out.print(node.toString());
    if(null != node.right)
        printNode(node.right);
    }

一般还会继续追问"能否使用非递归的方法?",在美团点评面试的时候就被问到了(同LeetCode 144)。这时候,我们可以使用栈来完成,分为下面三个步骤:

  1. 深度遍历找到这棵树最左边的节点,这个过程中将遍历的节点压入栈中。
  2. 将栈中的节点pop出来,如果该节点有右子节点,对右子节点进行步骤一的操作,如果没有,则可以打印出来,继续pop栈。
  3. 当栈空了或者进行操作的根节点为空时推出循环。
    下面就用代码来实现一下LeetCode 144
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
       ArrayList<Integer> result = new ArrayList<>();
        if (null == root) {
            return null;
        }
        Stack<TreeNode> nodeStack = new Stack<>();

        TreeNode temp = root;

        while (null != temp || !nodeStack.empty()) {
            nodeStack.push(temp);
            result.add(temp.val);
            while (null != temp.left) {
                temp = temp.left;
                nodeStack.push(temp);
                result.add(temp.val);
            }
            while (!nodeStack.empty() && null == nodeStack.peek().right) {
                nodeStack.pop();
            }
            if (nodeStack.empty())
                break;
            temp = nodeStack.pop();
            if (null != temp.right) {
                temp = temp.right;
                continue;
            }
        }
        return result; 
    }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 二叉树的三种常用遍历方式 学习过数据结构的同学都清楚,除了层序遍历外,二叉树主要有三种遍历方式: 1. 先序遍历...
    SherlockBlaze阅读 1,244评论 0 4
  • http://www.jianshu.com/p/49c8cfd07410 解决二叉树的很多问题的方案都是基于对二...
    MSG猿阅读 809评论 0 0
  • -先序遍历: 访问根结点,先序遍历其左子树,先序遍历其右子树;运用到递归void PreOrderTraversa...
    Spicy_Crayfish阅读 2,048评论 0 0
  • 二叉树的遍历想必大家都不陌生,主要有三种遍历方式:前序遍历(pre-order traversal),中序遍历(i...
    akak18183阅读 1,135评论 0 1
  • 构造 二叉树的构造。先要有一棵树,才能遍历一棵树。 首先构造一颗简单的完全二叉树 删去一些节点: 生成了一棵树,开...
    IAmWhoAmI阅读 221评论 0 2