二叉树遍历几种常见方式

来源:互联网 发布:mac的usb配件已停用 编辑:程序博客网 时间:2024/06/03 16:40

  • 前序遍历
    • 递归版本
    • 非递归版本
  • 中序遍历
    • 递归版本
    • 非递归版本
  • 后序遍历
    • 递归版本
    • 非递归版本

前序遍历

递归版本

先序遍历的递归过程为:若二叉树为空,遍历结束。否则,

  • 访问根结点;
  • 先序遍历根结点的左子树;
  • 先序遍历根结点的右子树。

代码实现如下:

public List<Integer> preorderTraversal(TreeNode root) {        List<Integer> res = new LinkedList<Integer>();        preHelper(root,res);        return res;    }public void preHelper(TreeNode root, List<Integer> res) {    if (root == null) return;    res.add(root.val);    preHelper(root.left, res);    preHelper(root.right, res);}

非递归版本

 public List<Integer> preorderTraversal(TreeNode root) {    List<Integer> result = new ArrayList<>();    Deque<TreeNode> stack = new ArrayDeque<>();    TreeNode p = root;    while (!stack.isEmpty() || p != null) {        if (p != null) {            stack.push(p);            result.add(p.val);  // 在遍历到左孩子之前添加            p = p.left;        } else {            TreeNode node = stack.pop();            p = node.right;        }    }    return result;}

中序遍历

递归版本

中序遍历的递归过程为:若二叉树为空,遍历结束。否则,

  • 中序遍历根结点的左子树;
  • 访问根结点;
  • 中序遍历根结点的右子树。
public List<Integer> postorderTraversal(TreeNode root) {    List<Integer> res = new ArrayList<>();    if (root == null) return res;    help(root, res);    return res;}private void help(TreeNode root, List<Integer> res) {    if (root == null) return;    help(root.left, res);    res.add(root.val);    help(root.right, res);}

非递归版本

public List<Integer> inorderTraversal(TreeNode root) {    List<Integer> result = new ArrayList<>();    Deque<TreeNode> stack = new ArrayDeque<>();    TreeNode p = root;    while(!stack.isEmpty() || p != null) {        if(p != null) {            stack.push(p);            p = p.left;        } else {            TreeNode node = stack.pop();            result.add(node.val);              p = node.right;           }    }    return result;}

后序遍历

递归版本

后序遍历的递归过程为:若二叉树为空,遍历结束。否则,

  • 后序遍历根结点的左子树;
  • 后序遍历根结点的右子树。
  • 访问根结点;
public List<Integer> postorderTraversal(TreeNode root) {    List<Integer> res = new ArrayList<>();    if (root == null) return res;    help(root, res);    return res;}private void help(TreeNode root, List<Integer> res) {    if (root == null) return;    help(root.left, res);    help(root.right, res);    res.add(root.val);}

非递归版本

public List<Integer> postorderTraversal(TreeNode root) {    LinkedList<Integer> result = new LinkedList<>();    Deque<TreeNode> stack = new ArrayDeque<>();    TreeNode p = root;    while(!stack.isEmpty() || p != null) {        if(p != null) {            stack.push(p);            result.addFirst(p.val);              p = p.right;                     } else {            TreeNode node = stack.pop();            p = node.left;                   }    }    return result;}