[模版总结] - 树的基本算法1 - 遍历

树结构定义

一种非线性存储结构,具有存储"一对多"关系的数据元素集合

种类

  • General Tree
    • Trie
    • B/B+ 树
  • 二叉树
    • 满/完满/完全二叉树
      • 完美BT : 除了叶子结点外所有节点都有两个字节点,每一层都完满填充
      • 完全BT: 除最后一层以外其他每一层都完美填充,最后一层从左到右紧密填充
      • 完满BT: 除了叶子结点外所有节点都有两个字节点
    • 二叉搜索树 BST
      • 平衡BST
        • 红黑树
        • 伸展树
        • 自平衡二叉查找树AVL
        • 替罪羊树
    • 线索二叉树
    • 霍夫曼树/最优二叉树

二叉树遍历方式

所有二叉树基本遍历时间复杂度均为:, N代表结点数量。

前序遍历 (根左右)

递归写法

java 复制代码
class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        dc(root, res);
        return res;
        
    }

    private void dc(TreeNode root, List<Integer> res) {
        if (root==null) return;

        res.add(root.val);
        dc(root.left, res);
        dc(root.right, res);
    }
}

中序遍历 (左根右)

递归写法

java 复制代码
class Solution {
    List<Integer> res;
    public List<Integer> inorderTraversal(TreeNode root) {
        res = new ArrayList<>();
        dc(root);

        return res;
    }

    private void dc(TreeNode root) {
        if (root==null) return;

        dc(root.left);
        res.add(root.val);
        dc(root.right);
    }
}

后续遍历 (左右根)

java 复制代码
class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        dc(root, res);
        return res;
    }

    private void dc(TreeNode root, List<Integer> res) {

        if (root==null) return;

        dc(root.left, res);
        dc(root.right, res);
        res.add(root.val);
    }    
}

层级遍历 I - 自上而下

树/图类层级遍历直接BFS即可

java 复制代码
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();

        if (root==null) return res;
        Queue<TreeNode> q = new LinkedList<>();
        q.add(root);

        while (!q.isEmpty()) {
            int size = q.size();

            List<Integer> tmp = new ArrayList<>();
            for (int i=0; i<size; i++) {
                TreeNode curr = q.poll();
                tmp.add(curr.val);
                if (curr.left!=null) q.add(curr.left);
                if (curr.right!=null) q.add(curr.right);
            }
            res.add(tmp);
        }

        return res;
    }
}

层级遍历 II - 自下而上

java 复制代码
class Solution {
    public List<List<Integer>> levelOrderBottom(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        
        if (root==null) return res;
        Queue<TreeNode> q = new LinkedList<>();
        q.add(root);

        while (!q.isEmpty()) {
            int size = q.size();
            List<Integer> tmp = new ArrayList<>();
            for (int i=0; i<size; i++) {
                TreeNode curr = q.poll();
                if (curr.left!=null) q.add(curr.left);
                if (curr.right!=null) q.add(curr.right);
                tmp.add(curr.val);
            }

            res.add(0, tmp);
        }

        return res;
    }
}

ZigZag 遍历

java 复制代码
class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        dfs(root, 0, res);
        return res;
    }

    private void dfs(TreeNode root, int height, List<List<Integer>> res) {
        if (root==null) return;
        if (res.size()<=height) res.add(new ArrayList<>());

        if (height%2==0) {
            res.get(height).add(root.val);
        } else {
            res.get(height).add(0, root.val);
        }

        dfs(root.left, height+1, res);
        dfs(root.right, height+1, res);
    }
}

一些特别的遍历:

逐列遍历

, N表示dfs遍历时间复杂度,C表示列数,R表示每一列的行数

java 复制代码
class Solution {
    TreeMap<Integer, List<Pair<Integer, Integer>>> colMap;
    public List<List<Integer>> verticalOrder(TreeNode root) {
        if (root==null) return new ArrayList<>();

        colMap = new TreeMap<>();
        dfs(root, 0, 0);

        List<List<Integer>> res = new ArrayList<>();

        for (int idx: colMap.keySet()) {
            Collections.sort(colMap.get(idx), (a, b) -> {
                return a.getKey()-b.getKey();
            });

            List<Integer> tmp = new ArrayList<>();
            for (Pair<Integer, Integer> a: colMap.get(idx)) {
                tmp.add(a.getValue());
            }

            res.add(tmp);
        }

        return res;

    }

    private void dfs(TreeNode root, int row, int col) {
        if (root==null) return;

        colMap.putIfAbsent(col, new ArrayList<>());
        colMap.get(col).add(new Pair<>(row, root.val));

        dfs(root.left, row+1, col-1);
        dfs(root.right, row+1, col+1);
    }
}
相关推荐
kangkang-2 小时前
PC端基于SpringBoot架构控制无人机(三):系统架构设计
java·架构·无人机
许小燚4 小时前
线性表——双向链表
数据结构·链表
界面开发小八哥4 小时前
「Java EE开发指南」如何用MyEclipse创建一个WEB项目?(三)
java·ide·java-ee·myeclipse
董董灿是个攻城狮4 小时前
5分钟搞懂什么是窗口注意力?
算法
Dann Hiroaki4 小时前
笔记分享: 哈尔滨工业大学CS31002编译原理——02. 语法分析
笔记·算法
idolyXyz4 小时前
[java: Cleaner]-一文述之
java
一碗谦谦粉4 小时前
Maven 依赖调解的两大原则
java·maven
netyeaxi5 小时前
Java:使用spring-boot + mybatis如何打印SQL日志?
java·spring·mybatis
收破烂的小熊猫~5 小时前
《Java修仙传:从凡胎到码帝》第四章:设计模式破万法
java·开发语言·设计模式
猴哥源码5 小时前
基于Java+SpringBoot的动物领养平台
java·spring boot