1.二叉树的递归遍历
回顾了递归的套路:1.返回值+传入参数
2.结束条件
3.写具体的每一个递归单位
而在二叉树的三种遍历都没有返回值,传入当前node节点和所给框架下的返回值数组res
当传入的节点是null时结束递归
每个递归单位包含输出当前节点+递归左孩子+递归右孩子,改变输出的位置即可实现三种遍历
144.二叉树的前序遍历
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);}
}
145.二叉树的后序遍历
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);}
}
94.二叉树的中序遍历
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.二叉树的迭代遍历
144.二叉树的前序遍历
众所周知递归都可以改成循环,本质上还是自己模拟递归栈的过程
前序遍历模拟最简单,因为处理的节点和先访问的节点一致,可以直接将节点的左右孩子入栈
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;}
}
- 145.二叉树的后序遍历(opens new window)
稍微改动下前序遍历的代码,左右孩子改下顺序再将最后的结果逆序就可以得到后序遍历的结果
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;}
}
94.二叉树的中序遍历
中序遍历没办法像前序遍历直接模拟,需要一路向左入栈,左孩子为空后处理,再查看右孩子
代码上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;}
}