力扣剑指Offer 第6天 搜索与回溯算法(简单)剑指 Offer 32 - I. 从上到下打印二叉树 剑指 Offer 32 - II. 从上到下打印二叉树 II 剑指 Offer 32 - III. 从上到下打印二叉树 III
剑指 Offer 32 - I. 从上到下打印二叉树
题目
从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3
/ \
9 20
/ \
15 7
返回:
[3,9,20,15,7]
思路
使用队列
- 头结点入队
- 每次出队首元素,将首元素的左子树和右子树入队,将首元素值存入数组
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public int[] levelOrder(TreeNode root) {
if(root==null)return new int[]{};//特判是否为空
//层次遍历二叉树 队列
LinkedList<TreeNode>q=new LinkedList<>();
ArrayList<Integer>ans=new ArrayList<>();
q.addFirst(root);
TreeNode n;
while(!q.isEmpty()){
n=q.getFirst();
q.removeFirst();
ans.add(n.val);
if(n.left!=null)q.addLast(n.left);
if(n.right!=null)q.addLast(n.right);
}
int len =ans.size();
int[] a=new int[len];
for(int i=0;i<len;i++)a[i]=ans.get(i);
return a;
}
}
剑指 Offer 32 - II. 从上到下打印二叉
题目
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
思路
bfs
一层一层搜索并记录每一层节点数量。
- 使用一个计数器
cnt
计算下一层搜寻到的非空节点个数 - 每一层用
len=cnt
获取当前层的节点数量,同时重置下一层节点数量(cnt=0
)
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>>list=new ArrayList<>();
if(root==null)return list;//特判是否为空
//层次遍历二叉树 队列 bfs 记录当前步骤可行节点数量
LinkedList<TreeNode>q=new LinkedList<>();
q.addFirst(root);
int cnt=1,len=0;//len为当前层的节点数量
TreeNode n;
while(!q.isEmpty()){
len=cnt;cnt=0;//获取当前层节点数量 重置cnt
List<Integer>li=new ArrayList<>();
for(int i=0;i<len;i++){//遍历一层的节点
n=q.getFirst();
q.removeFirst();
li.add(n.val);
if(n.left!=null){
q.addLast(n.left);
cnt++;
}
if(n.right!=null){
q.addLast(n.right);
cnt++;
}
}
list.add(li);
}
return list;
}
}
剑指 Offer 32 - III. 从上到下打印二叉
题目
请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[20,9],
[15,7]
]
思路
bfs
一层层搜索,记录每层节点数量cnt
,记录当前层数tier
- 每次循环遍历一层的节点
cnt
- 如果
tier
为偶数层则翻转元素 实现之字型
打印
代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>>list=new ArrayList<>();
if(root==null)return list;//特判是否为空
//层次遍历二叉树 队列 bfs 记录当前步骤可行节点数量
LinkedList<TreeNode>q=new LinkedList<>();
int cnt=1,len=0,tier=0;
q.addFirst(root);
TreeNode n;
while(!q.isEmpty()){
len=cnt;cnt=0;tier++;//获取当前层节点数量 重置cnt 下一层层数tier
List<Integer>li=new ArrayList<>();
for(int i=0;i<len;i++){//遍历一层的节点
n=q.getFirst();
q.removeFirst();
li.add(n.val);
if(n.left!=null){
q.addLast(n.left);
cnt++;
}
if(n.right!=null){
q.addLast(n.right);
cnt++;
}
}
if((tier&1)==0)Collections.reverse(li);//翻转
list.add(li);
}
return list;
}
}