【数据结构与算法】二叉树的 4 种遍历

简介: 【数据结构与算法】二叉树的 4 种遍历

前言

二叉树作为一种重要的数据结构,在算法中起到了承前启后的作用,它是数组和链表的延伸,也是图的基础。所以学习二叉树的相关知识是十分有必要的,而在相关的操作中,二叉树的遍历是最频繁的,今天就来看看二叉树的 4 种遍历方法!


二叉树数据结构

所谓二叉树,指的是每个结点最多有两个分支的树结构,其分支通常被称为“左子树”和“右子树”,而且他们的次序是固定的,不能随意颠倒,一棵二叉树的示例如下:



image.png

class TreeNode{
    int val;
    // 左子树
    TreeNode left;
    // 右子树
    TreeNode right;
}

前序遍历

也叫做先序遍历,首先访问根节点,然后遍历左子树,最后再遍历右子树。而在遍历左右子树时,仍然按照先访问根节点,然后遍历左子树,最后遍历右子树的方式,直到二叉树为空则返回!

遍历的方式又主要分为递归和迭代的方式,其具体实现如下所示。

递归

public ArrayList<Integer> preOrderReverse(TreeNode root){
    ArrayList<Integer> list = new ArrayList<>();
    preOrder(root, list);
    return list;
}
public void preOrder(TreeNode root, ArrayList<Integer> list){
    if(root != null){
        list.add(root.val);     
        preOrder(root.left, list);
        preOrder(root.right, list);
    }
}

迭代

/**
* 用栈来进行迭代,由于栈是一种 先进后出 的数据结构,要输出的顺序是 中、左、右
* 所以我们优先将根节点加入 stack 后,然后先加入右子树,再加入左子树
*/
public ArrayList<Integer> preOrderReverse(TreeNode root){
    // 栈,先进后出
    Stack<TreeNode> stack = new Stack<>();
    ArrayList<Integer> list = new ArrayList<>();
    if(root != null){
        // 入栈
        stack.push(root);
        while(!stack.empty()){
            // 出栈
            TreeNode node = stack.pop();
            list.add(node.val);
            // 栈是一种先进后出的数据结构,所以先入右子树,再入左子树
            if(node.right != null){
                stack.push(node.right);
            }
            if(node.left != null){
                stack.push(node.left);
            }
        }
    }
    return list;
}

中序遍历

首先遍历左子树,然后访问根节点,最后再遍历右子树。而在遍历左右子树时,仍然按照先遍历左子树,然后访问根节点,最后遍历右子树的方式,直到二叉树为空则返回!

遍历的方式又主要分为递归和迭代的方式,其具体实现如下所示。

递归

public ArrayList<Integer> inOrderReverse(TreeNode root){
    ArrayList<Integer> list = new ArrayList<>();
    inOrder(root, list);
    return list;
}
public void inOrder(TreeNode root, ArrayList<Integer> list){
    if(root != null){     
        inOrder(root.left, list);
        list.add(root.val);
        inOrder(root.right, list);
    }
}

迭代

/**
* 中序遍历,按照 左、中、右 的顺序打印
* 所以优先将左子树压入栈中,接着处理中间节点,最后处理右子树
*
*/
public ArrayList<Integer> inOrderReverse(TreeNode root){
    ArrayList<Integer> list = new ArrayList<>();
    Stack<TreeNode> stack = new Stack<TreeNode>();
    TreeNode curr = root;
    while(curr != null || !stack.isEmpty()){
        // 节点不为空就一直压栈
        while(curr != null){
            stack.push(curr);
            // 考虑左子树
            curr = curr.left;     
        }
        // 节点为空,出栈
        curr = stack.pop();
        // 加入当前值
        list.add(curr.val);
        // 考虑右子树
        curr = curr.right;
    }
    return list;
}

后序遍历

后序遍历首先遍历左子树,然后遍历右子树,最后访问根结点,在遍历左、右子树时,仍然先遍历左子树,然后遍历右子树,最后遍历根结点,直到二叉树为空则返回!

遍历的方式又主要分为递归和迭代的方式,其具体实现如下所示。

递归

public ArrayList<Integer> postOrderReverse(TreeNode root){
    ArrayList<Integer> list = new ArrayList<>();
    postOrder(root, list);
    return list;
}
public void postOrder(TreeNode root, ArrayList<Integer> list){
    if(root != null){     
        postOrder(root.left, list);
        postOrder(root.right, list);
        list.add(root.val);
    }
}

迭代

public ArrayList<Integer> postOrderReverse(TreeNode root){
    List<Integer> list = new ArrayList<Integer>();
    Stack<TreeNode> stack = new Stack<TreeNode>();
    TreeNode current = root;
    // 用来区分之前的结点是否被访问过
    TreeNode last = null;  
    while(current != null || !stack.isEmpty()){
        // 到树的最左面
        if(current != null){    
            stack.push(current);
            current = current.left;
        }else{
            //看最左结点有没有右子树
            current = stack.peek();    
            if(current.right != null && current.right != last){
                current = current.right;
                stack.push(current);
                //右子树再到最左
                current = current.left;     
            }else{
                //访问该结点,并标记被访问
                current = stack.pop();    
                list.add(current.val);
                last = current;
                current = null;
            }
        }
    }
    return list;
}

层次遍历

层次遍历也叫做广度优先遍历,它会优先访问离根节点最近的节点,其实现一般借助队列实现。

遍历的方式又主要分为递归和迭代的方式,其具体实现如下所示。

递归

public List<List<Integer>> levelOrder(TreeNode root) {
    List<List<Integer>> lists = new ArrayList<List<Integer>>();
    if(root != null){            
        // 根节点不为 null,递归
        dfs(1, root, lists);
    }
    return lists;
}
// index : 层数
public void dfs(int index, TreeNode root, List<List<Integer>> lists){
    // 若 lists 中序列数小于层数,则将 lists 中加入一个空的序列
    if(lists.size() < index){
        lists.add(new ArrayList<Integer>());
    }
    // 然后将当前节点加入 lists 的子序列中
    lists.get(index - 1).add(root.val);
    // 以上就处理完 root 节点
    // 接着处理左右子树即可,处理时,层数到下一次,所以要 +1
    if(root.left != null){
        dfs(index + 1, root.left, lists);
    }
    if(root.right != null){
        dfs(index + 1, root.right, lists);
    }
}

迭代

ArrayList<ArrayList<Integer>> levelOrder(TreeNode root){
    List<List<Integer>> res = new ArrayList<>();
    Queue<TreeNode> queue = new ArrayDeque<>();
    if (root != null) {
        queue.add(root);
    }
    while (!queue.isEmpty()) {
        // 获取当前队列的长度,这个长度相当于当前这一层的节点个数
        int n = queue.size();
        // 将队列中的元素都拿出来(也就是获取这一层的节点),放到临时list中
        // 如果节点的左/右子树不为空,也放入队列中
        List<Integer> level = new ArrayList<>();
        int i = 0;
        while(i < n){
            TreeNode node = queue.poll();
            level.add(node.val);
            if (node.left != null) {
                queue.add(node.left);
            }
            if (node.right != null) {
                queue.add(node.right);
            }
            i++;
        }
        // 将临时list加入最终返回结果中
        res.add(level);
    }
    return res;
}

总结

以上就是数据结构二叉树的 4 种遍历,如果你有更多关于各种遍历的实现,欢迎留言交流呀!

目录
相关文章
|
6月前
|
存储 算法 Java
算法系列之数据结构-二叉树
树是一种重要的非线性数据结构,广泛应用于各种算法和应用中。本文介绍了树的基本概念、常见类型(如二叉树、满二叉树、完全二叉树、平衡二叉树、B树等)及其在Java中的实现。通过递归方法实现了二叉树的前序、中序、后序和层次遍历,并展示了具体的代码示例和运行结果。掌握树结构有助于提高编程能力,优化算法设计。
187 10
 算法系列之数据结构-二叉树
|
10月前
|
算法
分享一些提高二叉树遍历算法效率的代码示例
这只是简单的示例代码,实际应用中可能还需要根据具体需求进行更多的优化和处理。你可以根据自己的需求对代码进行修改和扩展。
264 64
|
8月前
|
Java C++
【C++数据结构——树】二叉树的基本运算(头歌实践教学平台习题)【合集】
本关任务:编写一个程序实现二叉树的基本运算。​ 相关知识 创建二叉树 销毁二叉树 查找结点 求二叉树的高度 输出二叉树 //二叉树节点结构体定义 structTreeNode{ intval; TreeNode*left; TreeNode*right; TreeNode(intx):val(x),left(NULL),right(NULL){} }; 创建二叉树 //创建二叉树函数(简单示例,手动构建) TreeNode*create
182 12
|
8月前
|
C++
【C++数据结构——树】二叉树的性质(头歌实践教学平台习题)【合集】
本文档介绍了如何根据二叉树的括号表示串创建二叉树,并计算其结点个数、叶子结点个数、某结点的层次和二叉树的宽度。主要内容包括: 1. **定义二叉树节点结构体**:定义了包含节点值、左子节点指针和右子节点指针的结构体。 2. **实现构建二叉树的函数**:通过解析括号表示串,递归地构建二叉树的各个节点及其子树。 3. **使用示例**:展示了如何调用 `buildTree` 函数构建二叉树并进行简单验证。 4. **计算二叉树属性**: - 计算二叉树节点个数。 - 计算二叉树叶子节点个数。 - 计算某节点的层次。 - 计算二叉树的宽度。 最后,提供了测试说明及通关代
163 10
|
8月前
|
存储 算法 测试技术
【C++数据结构——树】二叉树的遍历算法(头歌教学实验平台习题) 【合集】
本任务旨在实现二叉树的遍历,包括先序、中序、后序和层次遍历。首先介绍了二叉树的基本概念与结构定义,并通过C++代码示例展示了如何定义二叉树节点及构建二叉树。接着详细讲解了四种遍历方法的递归实现逻辑,以及层次遍历中队列的应用。最后提供了测试用例和预期输出,确保代码正确性。通过这些内容,帮助读者理解并掌握二叉树遍历的核心思想与实现技巧。
259 3
|
9月前
|
存储 算法 Python
文件管理系统中基于 Python 语言的二叉树查找算法探秘
在数字化时代,文件管理系统至关重要。本文探讨了二叉树查找算法在文件管理中的应用,并通过Python代码展示了其实现过程。二叉树是一种非线性数据结构,每个节点最多有两个子节点。通过文件名的字典序构建和查找二叉树,能高效地管理和检索文件。相较于顺序查找,二叉树查找每次比较可排除一半子树,极大提升了查找效率,尤其适用于海量文件管理。Python代码示例包括定义节点类、插入和查找函数,展示了如何快速定位目标文件。二叉树查找算法为文件管理系统的优化提供了有效途径。
146 5
|
9月前
|
数据库
数据结构中二叉树,哈希表,顺序表,链表的比较补充
二叉搜索树,哈希表,顺序表,链表的特点的比较
数据结构中二叉树,哈希表,顺序表,链表的比较补充
|
8月前
|
数据采集 存储 算法
【C++数据结构——图】图的遍历(头歌教学实验平台习题) 【合集】
本文介绍了图的遍历算法,包括深度优先遍历(DFS)和广度优先遍历(BFS)。深度优先遍历通过递归方式从起始节点深入探索图,适用于寻找路径、拓扑排序等场景;广度优先遍历则按层次逐层访问节点,适合无权图最短路径和网络爬虫等应用。文中提供了C++代码示例,演示了如何实现这两种遍历方法,并附有测试用例及结果,帮助读者理解和实践图的遍历算法。
319 0
|
10月前
|
机器学习/深度学习 存储 算法
数据结构实验之二叉树实验基础
本实验旨在掌握二叉树的基本特性和遍历算法,包括先序、中序、后序的递归与非递归遍历方法。通过编程实践,加深对二叉树结构的理解,学习如何计算二叉树的深度、叶子节点数等属性。实验内容涉及创建二叉树、实现各种遍历算法及求解特定节点数量。
256 4
|
10月前
|
C语言
【数据结构】二叉树(c语言)(附源码)
本文介绍了如何使用链式结构实现二叉树的基本功能,包括前序、中序、后序和层序遍历,统计节点个数和树的高度,查找节点,判断是否为完全二叉树,以及销毁二叉树。通过手动创建一棵二叉树,详细讲解了每个功能的实现方法和代码示例,帮助读者深入理解递归和数据结构的应用。
683 8

热门文章

最新文章