二叉树遍历算法

先序遍历

先序遍历即先遍历根节点在遍历左节点和右节点,利用栈的特点先进后出可以先压根节点入栈,弹出根节点后,在把右孩子入栈,然后左孩子入栈,这样出栈顺序就是先根节点再左孩子,右孩子;

中序遍历

先把根节点入栈,然后把左节点全部压入栈中,一直到叶子节点,然后开始出栈,令右孩子成为子树根节点重复上面的步骤,最后压入栈的顺序如下图;


image.png

后序遍历

后序遍历可以仿照先序遍历,先序遍历是根左右,而后序遍历是左右根,其实先序遍历入栈是左右孩子换位,得到的结果就是根右左,这样反转得到的就是后序遍历;

层次遍历

层次遍历通过队列实现,直接从根节点开始把左右孩子放入队列中,这样每次都是上一层先进队列的先出队列;

 /**先序遍历*/
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        if(!Objects.isNull(root)) {
            Stack<TreeNode> stack = new Stack<>();
            stack.push(root);
            TreeNode cur;
            while (!stack.empty()) {
                cur = stack.pop();
                list.add(cur.val);
                if(!Objects.isNull(cur.right)) {
                    stack.push(cur.right);
                }
                if(!Objects.isNull(cur.left)) {
                    stack.push(cur.left);
                }
            }
        }
        return list;
    }
    /**中序遍历*/
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        if(Objects.nonNull(root)) {
            Stack<TreeNode> stack = new Stack<>();
            TreeNode cur = root;
            while (!stack.empty() || Objects.nonNull(cur)) {
                while (Objects.nonNull(cur)) {
                    stack.push(cur);
                    cur = cur.left;
                }
                cur = stack.pop();
                list.add(cur.val);
                cur = cur.right;
            }
        }
        return list;
    }
    /**后序遍历*/
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        if(!Objects.isNull(root)) {
            Stack<TreeNode> stack = new Stack<>();
            stack.push(root);
            TreeNode cur;
            while (!stack.empty()) {
                cur = stack.pop();
                list.add(cur.val);
                if(!Objects.isNull(cur.left)) {
                    stack.push(cur.left);
                }
                if(!Objects.isNull(cur.right)) {
                    stack.push(cur.right);
                }
            }
        }
        Collections.reverse(list);
        return list;
    }
    /**层次遍历*/
    public List<Integer> levelOrder(TreeNode root) {
        List<Integer> list = new ArrayList<>();
        if(Objects.nonNull(root)) {
            Queue<TreeNode> queue = new LinkedList<>();
            queue.offer(root);
            TreeNode cur;
            while (!queue.isEmpty()) {
                cur = queue.poll();
                list.add(cur.val);
                if(Objects.nonNull(cur.left)) {
                    queue.offer(cur.left);
                }
                if(Objects.nonNull(cur.right)) {
                    queue.offer(cur.right);
                }
            }
        }
        return list;
    }
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 前言 二叉树的前序遍历,中序遍历,后序遍历是面试中常常考察的基本算法,关于它的概念这里不再赘述了,还不了解的同学可...
    Jesse1995阅读 16,612评论 0 3
  • 摘要:二叉树主要有3种遍历算法,分为为先序、中序、后序。本文对二叉树的3种遍历算法的遍历规则进行介绍,并给出3种遍...
    杨赟快跑阅读 18,110评论 5 18
  • 图文解说 1、先序遍历又称为根左右 遍历规则: 先遍历根节点,然后遍历左节点,最后遍历呦节点。 举例 首先遍历A节...
    萧修阅读 800评论 0 0
  • 先序遍历 思路:先根节点->左子树->右子树;二叉树如下图: 先序遍历结果:ABDEGCFHI 中序遍历 思路:先...
    JerrySi阅读 257评论 0 0
  • 1,前序遍历 2,中序遍历 3,后序遍历 4,队列层级遍历 5,计算二叉树节点数 一,首先定义一个二叉树的节点 二...
    王元阅读 133评论 0 0