基础知识(青铜挑战)
- 理解递归思想:调用自己、压栈、明确终止条件
实战训练(白银挑战)
递归实现二叉树的前、中、后序遍历
-
我的建议是直接把代码背下来
-
你当然可以尝试去理解,也不是很难想明白,但是不要钻牛角尖了,递归这玩意儿,很不友好
java
/**
* 前序遍历,将结果返回到list中
*
* @param root
* @param res
*/
public static void preOrder(TreeNode root, List<Integer> res) {
if (root == null) {
return;
}
res.add(root.val);
preOrder(root.left, res);
preOrder(root.right, res);
}
java
/**
* 中序遍历,将结果返回到list中
*
* @param root
* @param res
*/
public static void inOrder(TreeNode root, List<Integer> res) {
if (root == null) {
return;
}
inOrder(root.left, res);
res.add(root.val);
inOrder(root.right, res);
}
java
/**
* 后序遍历,将结果返回到list中
*
* @param root
* @param res
*/
public static void postOrder(TreeNode root, List<Integer> res) {
if (root == null) {
return;
}
postOrder(root.left, res);
postOrder(root.right, res);
res.add(root.val);
}
迭代实现二叉树的前、中、后序遍历
前序遍历
java
public static List<Integer> preOrderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
if (root == null) {
return res;
}
Deque<TreeNode> stack = new LinkedList<TreeNode>();
TreeNode node = root;
while (!stack.isEmpty() || node != null) {
while (node != null) {
res.add(node.val);
stack.push(node);
node = node.left;
}
node = stack.pop();
node = node.right;
}
return res;
}
中序遍历(2023/09/12午)
java
public static List<Integer> inorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
Deque<TreeNode> stack = new LinkedList<TreeNode>();
while (root != null || !stack.isEmpty()) {
while (root != null) {
stack.push(root);
root = root.left;
}
root = stack.pop();
res.add(root.val);
root = root.right;
}
return res;
}
-
前序遍历和中序遍历的代码不复杂,还是注意:挺不好理解的,不要钻牛角尖,那么我们接下来看下后序遍历
后序遍历
-
这个相较于前两个更难实现,但可以这样想:
-
前序遍历的顺序是:中、左、右,我们简单该两个参数,实现遍历顺序为中、右、左
-
再将遍历结果reserve反转,不就得到左、右、中的遍历顺序了吗?
-
java
public static List<Integer> postOrderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
if (root == null) return res;
Stack<TreeNode> stack = new Stack<>();
TreeNode node = root;
while (!stack.isEmpty() || node != null) {
while (node != null) {
res.add(node.val);
stack.push(node);
node = node.right;
}
node = stack.pop();
node = node.left;
}
Collections.reverse(res);
return res;
}