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

树结构定义

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

种类

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

二叉树遍历方式

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

前序遍历 (根左右)

• 题目：Leetcode 144

递归写法 

由于前序的特性，他可以展示树结构的继承关系，所以通常前序遍历会用在复制/打印树结构，比如序列化/反序列化，打印文件系统结构。

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);}
}

中序遍历 (左根右)

• 题目：Leetcode 94

递归写法

中序遍历最常用就是打印BST结构

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);}
}

后续遍历 (左右根) 

•  题目：Leetcode 145​​​​​​

后续遍历由于特性是先搜索叶子结点，所以可以用来做一些叶子结点操作(删除叶子结点)，还有一些归并操作(计算算术表达式)

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 - 自上而下

• 题目：Leetcode 102

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

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 - 自下而上

• 题目：Leetcode 107

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 遍历

• 题目：Leetcode 103​​​​​

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表示每一列的行数

• 题目：Leetcode 314

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);}
}