力扣-树-层次遍历
1.从上到下打印二叉树(键指Offer 32 -1)
- 题目概述:从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印
- 题目范例:
- 解题思路:一层一层遍历其实就是使用广度优先搜索BFS。用队列保存节点,秉承先进先出的原则
- java代码
class Solution {
public int[] levelOrder(TreeNode root) {
if(root==null){
return new int[0];
}
List<Integer> lst=new ArrayList<>();
Queue<TreeNode> queue=new LinkedList<>();
//BFS
BFS(queue,root,lst);
int[] outcome=new int[lst.size()];
for (int i = 0; i < lst.size(); i++) {
outcome[i]=lst.get(i);
}
return outcome;
}
public List<Integer> BFS(Queue<TreeNode> queue,TreeNode root,List<Integer> lst){
queue.add(root);
while(!queue.isEmpty()){
TreeNode node=queue.poll();
lst.add(node.val);
if(node.left!=null){
queue.add(node.left);
}
if(node.right!=null){
queue.add(node.right);
}
}
return lst;
}
}
- python代码
import collections
class Solution:
def levelOrder(self, root: TreeNode) -> List[int]:
if not root:
return []
lst=[]
queue=collections.deque()
queue.append(root)
while(queue):
#popleft取左边,pop取右边
node=queue.popleft()
lst.append(node.val)
if(node.left):
queue.append(node.left)
if(node.right):
queue.append(node.right)
return lst
2.二叉树的层序遍历(LeetCode 102)
- 题目概述:给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)
- 题目范例:
- 解题思路:本题和上题大体思路一样,只需要多一步将每一层的结点按从左到右顺序保存在一个集合中。这里最重要就是要思考怎样能将一层结点存放在一起,其实当我们将队列现有结点依次poll出来就是该层结点,而新加入的结点即为下一层的结点集,直到二叉树遍历结束
- java代码
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
if(root==null){
return new ArrayList<>();
}
List<List<Integer>> lst=new ArrayList<>();
Queue<TreeNode> queue=new LinkedList<>();
queue.add(root);
while(!queue.isEmpty()){
List<Integer> temp=new ArrayList<>();
int LevelNum= queue.size();
for (int i = 0; i < LevelNum; i++) {
TreeNode node=queue.poll();
temp.add(node.val);
if(node.left!=null){
queue.add(node.left);
}
if(node.right!=null){
queue.add(node.right);
}
}
lst.add(temp);
}
return lst;
}
}
- python代码
import collections
class Solution:
def levelOrder(self, root: TreeNode) -> List[List[int]]:
if not root:
return []
lst=[]
queue=collections.deque()
queue.append(root)
while(queue):
temp=[]
for i in range(len(queue)):
node = queue.popleft()
temp.append(node.val)
if(node.left):
queue.append(node.left)
if(node.right):
queue.append(node.right)
lst.append(temp)
return lst