1.问题描述
给定一个二叉树,分别进行前中后序的遍历。
例如:给你数组【5,8,9,4,7】
2.解决办法
1.首先理解前中后序遍历的形式
- 前序:根节点->左节点->右节点
- 中序:左节点->根节点->右节点
- 前序:左节点->右节点->根节点
2.递归
以后序为例:
定义函数postorder(root) 表示当前遍历到 root 节点的答案。按照定义,我们只要递归调用 postorder(root->left) 来遍历 root 节点的左子树,然后递归调用 postorder(root->right) 来遍历 root 节点的右子树,最后将 root 节点的值加入答案即可,递归终止的条件为碰到空节点。所以只需要改变添加节点的位置{res.add(root.val);}即可实现前后中序的遍历;
3.代码实现
前序遍历
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
postorder(root, res);
return res;
}
public void postorder(TreeNode root, List<Integer> res) {
if (root == null) {
return;
}
res.add(root.val);
postorder(root.left, res);
postorder(root.right, res);
}
}
中序遍历
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
postorder(root, res);
return res;
}
public void postorder(TreeNode root, List<Integer> res) {
if (root == null) {
return;
}
postorder(root.left, res);
res.add(root.val);
postorder(root.right, res);
}
}
后序遍历
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<Integer>();
postorder(root, res);
return res;
}
public void postorder(TreeNode root, List<Integer> res) {
if (root == null) {
return;
}
postorder(root.left, res);
postorder(root.right, res);
res.add(root.val);
}
}