深度优先遍历(DFS)
递归算法的三步骤:(以前序遍历为例)
- 确定递归函数的参数和返回值:因为要打印遍历节点的值,所以参数要传入ArrayList存放节点的值,不需要有返回值,所以返回类型为void。
- 确定终止条件:当前遍历节点为空,本层递归就结束了。
- 确定单层递归的逻辑:前序遍历是中左右的顺序。
前序遍历(递归法)中—>左—>右
class TreeNode {
int val;
TreeNode left;
TreeNode right;
public TreeNode() {}
public TreeNode(int val) {
this.val = val;
}
}
public class RecursiveTraversal {
public void preOrder(TreeNode root, ArrayList<Integer> res) {
if (root == null) {
return; // 递归终止条件为:当前遍历的节点为空
}
// 单层递归逻辑
res.add(root.val); // 中
preOrder(root.left, res); // 左
preOrder(root.right, res); // 右
}
List<Integer> preOrderReverse(TreeNode root) {
ArrayList<Integer> res = new ArrayList<>();
preOrder(root, res);
return res;
}
public static void main(String[] args) {
TreeNode t1 = new TreeNode(1);
TreeNode t2 = new TreeNode(2);
TreeNode t3 = new TreeNode(3);
t1.left = t2;
t1.right = t3;
RecursiveTraversal rt = new RecursiveTraversal();
LinkedList<Integer> res = rt.preOrderReverse(t1);
System.out.println(res); // [1, 2, 3]
}
}
中序遍历和后序遍历只需要修改单层递归逻辑的顺序即可:
中序遍历(递归法)左—>中—>右
public void inOrder(TreeNode root, LinkedList<Integer> res) {
if (root == null) {
return; // 递归终止条件为:当前遍历的节点为空
}
// 单层递归逻辑
preOrder(root.left, res); // 左
res.add(root.val); // 中
preOrder(root.right, res); // 右
}
后序遍历(递归法)左—>右—>中
public void postOrder(TreeNode root, LinkedList<Integer> res) {
if (root == null) {
return; // 递归终止条件为:当前遍历的节点为空
}
// 单层递归逻辑
preOrder(root.left, res); // 左
preOrder(root.right, res); // 右
res.add(root.val); // 中
}