二叉树的主要遍历方式有两种:
- DFS,Depth First Searc,深度优先
- BFS,Breadth First Search,广度优先
1.二叉树结构
1
/ \
2 3
/ \
4 5
2.构建节点和二叉树
class TreeNode {
// 节点的值
int val;
// 左节点
TreeNode left;
// 右节点
TreeNode right;
TreeNode(int x) {
val = x;
}
}
TreeNode num1 = new TreeNode(1);
TreeNode num2 = new TreeNode(2);
TreeNode num3 = new TreeNode(3);
TreeNode num4 = new TreeNode(4);
TreeNode num5 = new TreeNode(5);
num1.left = num2;
num1.right = num3;
num3.left = num4;
num3.right = num5;
3.DFS
3.1先序遍历(根->左->右)
public void preTraverse(TreeNode root) {
if (root == null) {
return;
}
// 获取节点值
res.add(root.val);
// 递归遍历左节点
preTraverse(root.left);
// 递归遍历右节点
preTraverse(root.right);
}
[1, 2, 3, 4, 5]
3.2中序遍历(左->根->右)
public void midTraverse(TreeNode root) {
if (root == null) {
return;
}
// 递归遍历左节点
midTraverse(root.left);
// 获取节点值
res.add(root.val);
// 递归遍历右节点
midTraverse(root.right);
}
[2, 1, 4, 3, 5]
3.3后序遍历(左->右->根)
public void afterTraverse(TreeNode root) {
if (root == null) {
return;
}
// 递归遍历左节点
afterTraverse(root.left);
// 递归遍历右节点
afterTraverse(root.right);
// 获取节点值
res.add(root.val);
}
[2, 4, 5, 3, 1]
4.BFS
public void breadthFirstSearch(TreeNode root) {
Queue<TreeNode> queue = new ArrayDeque<>();
queue.offer(root);
while (queue.peek() != null) {
// 队列头部元素
TreeNode poll = queue.poll();
// 左节点入队
TreeNode left = poll.left;
if (left != null) {
queue.offer(left);
}
// 右节点入队
TreeNode right = poll.right;
if (right != null) {
queue.offer(right);
}
res.add(poll.val);
}
}
[1, 2, 3, 4, 5]
Reference
https://www.cnblogs.com/anthony-dong/p/12250802.html