1.二叉树的递归遍历
回顾了递归的套路:1.返回值+传入参数
2.结束条件
3.写具体的每一个递归单位
而在二叉树的三种遍历都没有返回值,传入当前node节点和所给框架下的返回值数组res
当传入的节点是null时结束递归
每个递归单位包含输出当前节点+递归左孩子+递归右孩子,改变输出的位置即可实现三种遍历
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
pre(root,res);
return res;
}
void pre(TreeNode node, List<Integer> res){
if(node == null) return;
res.add(node.val);
pre(node.left,res);
pre(node.right,res);
}
}
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
post(root,res);
return res;
}
void post(TreeNode node, List<Integer> res){
if(node == null) return;
post(node.left,res);
post(node.right,res);
res.add(node.val);
}
}
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
in(root,res);
return res;
}
void in(TreeNode node, List<Integer> res){
if(node == null) return;
in(node.left,res);
res.add(node.val);
in(node.right,res);
}
}
2.二叉树的迭代遍历
众所周知递归都可以改成循环,本质上还是自己模拟递归栈的过程
前序遍历模拟最简单,因为处理的节点和先访问的节点一致,可以直接将节点的左右孩子入栈
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
if (root == null) return res;
Deque<TreeNode> s = new LinkedList<>();
s.push(root);
while (!s.isEmpty()) {
TreeNode cur = s.pop();
res.add(cur.val);
if (cur.right != null) s.push(cur.right);
if (cur.left != null) s.push(cur.left);
}
return res;
}
}
稍微改动下前序遍历的代码,左右孩子改下顺序再将最后的结果逆序就可以得到后序遍历的结果
Java里的集合进行逆序:Collections.reverse(res)
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
if (root == null) return res;
Deque<TreeNode> s = new LinkedList<>();
s.push(root);
while (!s.isEmpty()) {
TreeNode cur = s.pop();
res.add(cur.val);
if (cur.left != null) s.push(cur.left);
if (cur.right != null) s.push(cur.right);
}
Collections.reverse(res);
return res;
}
}
中序遍历没办法像前序遍历直接模拟,需要一路向左入栈,左孩子为空后处理,再查看右孩子
代码上while内的条件直接作为两个分支处理
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
List<Integer> res = new LinkedList<>();
if (root == null) return res;
Deque<TreeNode> s = new LinkedList<>();
TreeNode cur = root;
while(cur != null || !s.isEmpty()){
if (cur != null) {
s.push(cur);
cur = cur.left;
}else {
cur = s.pop();
res.add(cur.val);
cur = cur.right;
}
}
return res;
}
}
3.二叉树的层序遍历
按照广度优先搜索,从根节点(第一层)开始,输出节点后将节点的左右孩子入队
注意代码中队列使用offer和poll,以及容器的size要在循环前保存
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
if (root == null) return res;
Deque<TreeNode> q =new LinkedList<>();
q.offer(root);
while(!q.isEmpty()){
List<Integer> list=new ArrayList<>();
int n = q.size();
for (int i = 0; i < n; i++) {
TreeNode node = q.poll();
list.add(node.val);
if (node.left != null) q.offer(node.left);
if (node.right != null) q.offer(node.right);
}
res.add(list);
}
return res;
}
}