Java实现二叉树的遍历(递归与非递归)

12 篇文章 0 订阅
11 篇文章 0 订阅

二叉树的遍历分深度递归与广度递归,深度递归主要分为三种:

1. 前序遍历
2. 中序遍历
3. 后序遍历

下面是java实现的三种遍历的方式,包括递归实现与栈方式实现(非递归),参考代码如下:

import java.util.Arrays;
import java.util.LinkedList;
import java.util.Stack;

public class BinaryTreeDemo {

    static class TreeNode {
        public Object data;
        public TreeNode leftNode;
        public TreeNode rightNode;

        public TreeNode(Object data) {
            this.data = data;
        }
    }

    /**
     * 创建二叉树
     * @param list
     * @return
     */
    public static TreeNode createBinaryTree(LinkedList<Integer> list) {
        TreeNode node = null;
        if (list == null || list.isEmpty()) {
            return null;
        }
        Integer data = list.removeFirst();
        if (data != null) {
            node = new TreeNode(data);
            node.leftNode = createBinaryTree(list);
            node.rightNode = createBinaryTree(list);
        }
        return node;
    }

    /**
     * 递归-前序遍历二叉树
     * @param node
     */
    public static void preOrderTraveral(TreeNode node) {
        if (node == null) {
            return;
        }
        System.out.print(node.data + " ");
        preOrderTraveral(node.leftNode);
        preOrderTraveral(node.rightNode);
    }

    /**
     * 非递归-前序遍历二叉树
     * @param root
     */
    public static void preOrderStackTraveral(TreeNode root) {
        Stack<TreeNode> stack = new Stack<>();
        TreeNode treeNode = root;
        while (treeNode != null || !stack.isEmpty()) {
            while (treeNode != null) {
                System.out.print(treeNode.data + " ");
                stack.push(treeNode);
                treeNode = treeNode.leftNode;
            }
            if (!stack.isEmpty()) {
                treeNode = stack.pop();
                treeNode = treeNode.rightNode;
            }
        }
    }

    /**
     * 递归-中序遍历二叉树
     * @param node
     */
    public static void midOrderTraveral(TreeNode node) {
        if (node == null) {
            return;
        }
        midOrderTraveral(node.leftNode);
        System.out.print(node.data + " ");
        midOrderTraveral(node.rightNode);
    }

    /**
     * 非递归-中序遍历二叉树
     * @param root
     */
    public static void midOrderStackTraveral(TreeNode root) {
        Stack<TreeNode> stack = new Stack<>();
        TreeNode treeNode = root;
        while (treeNode != null || !stack.isEmpty()) {
            while (treeNode != null) {
                stack.push(treeNode);
                treeNode = treeNode.leftNode;
            }
            if (!stack.isEmpty()) {
                treeNode = stack.pop();
                System.out.print(treeNode.data + " ");
                treeNode = treeNode.rightNode;
            }
        }

    }

    /**
     * 递归-后序遍历二叉树
     * @param node
     */
    public static void postOrderTraveral(TreeNode node) {
        if (node == null) {
            return;
        }
        postOrderTraveral(node.leftNode);
        postOrderTraveral(node.rightNode);
        System.out.print(node.data + " ");
    }

    /**
     * 非递归-后序遍历二叉树
     * 
     * 思路:利用辅助栈实现
     * 1、先把根节点压入辅助栈
     * 2、然后循环,判断辅助栈是否为空,如果不为空,则栈顶元素压入主栈,同时存在左右节点的,分别压入左右节点到辅助栈
     * 3、最后主栈遍历,按出栈顺序打印节点信息,得到到便是后序遍历到顺序节点信息
     * @param root
     */
    public static void postOrderStackTraveral(TreeNode root) {
        if (root == null) {
            return;
        }
        Stack<TreeNode> stack = new Stack<>();
        Stack<TreeNode> stack1 = new Stack<>();
        TreeNode treeNode = root;
        stack1.push(treeNode);
        while (!stack1.isEmpty()) {
            treeNode = stack1.pop();
            stack.push(treeNode);
            if (treeNode.leftNode != null) {
                stack1.push(treeNode.leftNode);
            }
            if (treeNode.rightNode != null) {
                stack1.push(treeNode.rightNode);
            }
        }
        while (!stack.isEmpty()) {
            treeNode = stack.pop();
            System.out.print(treeNode.data + " ");
        }
    }

    public static void main(String[] args) {
        LinkedList<Integer> linkedList = new LinkedList<>(Arrays.asList(3, 2, 9, null, null, 10, null, null, 8, null, 4));
        TreeNode node = createBinaryTree(linkedList);
        System.out.println("\n前序遍历:");
        preOrderTraveral(node);
        System.out.println("");
        preOrderStackTraveral(node);
        System.out.println("\n中序遍历:");
        midOrderTraveral(node);
        System.out.println("");
        midOrderStackTraveral(node);
        System.out.println("\n后序遍历:");
        postOrderTraveral(node);
        System.out.println("");
        postOrderStackTraveral(node);
    }
}

测试代码实现结果:

前序遍历:
3 2 9 10 8 4
3 2 9 10 8 4
中序遍历:
9 2 10 3 8 4
9 2 10 3 8 4
后序遍历:
9 10 2 4 8 3
9 10 2 4 8 3

后感:相对来说,非递归实现的后序遍历相对比较复杂,需要借助辅助栈,思路也是参考网上其他的实现的;

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值