二叉树的主要遍历方式有两种:
DFS,Depth First Searc,深度优先
BFS,Breadth First Search,广度优先
二叉树结构
1
/ \
2 3
/ \
4 5
构建二叉树节点
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;
DFS,深度优先
先序遍历(根->左->右)
public void preTraverse(TreeNode root) {
if (root == null) {
return;
}
// 获取节点值
res.add(root.val);
// 递归遍历左节点
preTraverse(root.left);
// 递归遍历右节点
preTraverse(root.right);
}
[1, 2, 3, 4, 5]
中序遍历(左->根->右)
public void midTraverse(TreeNode root) {
if (root == null) {
return;
}
// 递归遍历左节点
midTraverse(root.left);
// 获取节点值
res.add(root.val);
// 递归遍历右节点
midTraverse(root.right);
}
[2, 1, 4, 3, 5]
后序遍历(左->右->根)
public void afterTraverse(TreeNode root) {
if (root == null) {
return;
}
// 递归遍历左节点
afterTraverse(root.left);
// 递归遍历右节点
afterTraverse(root.right);
// 获取节点值
res.add(root.val);
}
[2, 4, 5, 3, 1]
BFS 广度优先,层次遍历
private List<Integer> breadthFirstSearch(TreeNode root) {
if (null == root) return Collections.emptyList();
ArrayList<Integer> list = new ArrayList<>();
LinkedList<TreeNode> queue = new LinkedList<>();
// 添加一个元素并返回true
queue.offer(root);
// 返回队列头部的元素
while (queue.peek() != null) {
// 移除并返问队列头部的元素
TreeNode poll = queue.poll();
TreeNode left = poll.left;
if (null != left) {
queue.offer(left);
}
TreeNode right = poll.right;
if (null != right) {
queue.offer(right);
}
list.add(poll.val);
}