- 广度优先搜索(层序遍历)迭代法,用队列
- 题目
思路:用队列,num统计每一层的结点数。队列不为空为循环条件。如果num不为0,出队列,并把它的左右孩子入队列。num为0,表示一层遍历完成,将这一层添加到resul中。(resul的定义的意思是往resul中添加List),当想清除list不能用clear,因为resul中存的list的地址,如果给list清空,resul中数据也会清空。但是可以初始化list,得到新的地址再存下一层结点。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
//层序遍历(队列)
//num记录每一层的结点个数
Queue<TreeNode>queue=new LinkedList<TreeNode>();
ArrayList<Integer>list=new ArrayList<>();
List<List<Integer>>resul=new ArrayList<>();
int num=0;
TreeNode node;
//头结点不为空,添加头结点
if(root!=null){
queue.add(root);
num=queue.size();
}else{
//为空则返回resul
return resul;}
while(!queue.isEmpty())
{
//出栈,入栈它的左右孩子
if(num!=0) {
//出栈
node=queue.poll();
//存入数组
list.add(node.val);
//存左结点
if(node.left!=null)
queue.add(node.left);
//存右结点
if(node.right!=null)
queue.add(node.right);
num--;
}else {
//一层的数添加到结果数组
resul.add(list);
//不能用list.clear(),因为resul里存的list的地址
//如果用该方法,list就为空,resul也会为空
// 这里我们初始化list,它会指向新的对象,因此地址也会不一样
list=new ArrayList<>();
num=queue.size();
}
}
//最后一层的节点因为没有孩子,当它们都存入list中
//由于栈为空,没有进行else里面的添加操作,因此要自己补上
resul.add(list);
return resul;
}
}