分行层序遍历
如果要分层把list输出到res中(也就是一层完了再打印一层),就要在while中加一个for来控制到什么时候停止,一定要注意在加的时候看节点是否为空,是空continue,那怎么知道for循环中的size呢,就是看队列中的节点个数,在开始前的节点个数就是这一层所有的节点个数,全部循环完也就是这一层的下一层节点也全部添加进去了
还要注意如果list长度不为0才能添加到res中,否则最后会多添加一个空list
小技巧是创建一个list,然后每次添加完之后都令list=new ArrayList,这样可以创建几个变量?
import java.util.ArrayList;
import java.util.LinkedList;
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> res=new ArrayList<List<Integer>>();
if(root==null)return res;
Queue<TreeNode> q=new LinkedList<TreeNode>();
q.offer(root);
while(!q.isEmpty()){
int size=q.size();
List<Integer> list=new ArrayList<Integer>();
for(int i=0;i<size;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;
}
}
分行之字层序遍历
首先想到树的层序遍历,用一个队列输出,然后想到逆序输出的方式是用list.add(0,a);或者是用deque,也可以实现双端插入。
还有一种通用的方法是设置两个栈,然后每次输出两行,首先加入一个s1,然后在循环1中向s2加入下一层的节点,此时是先加左后加右,这样等会s2输出的时候就是先输出右边的节点了,然后当s1空的时候。再进入s2的循环中,先输出的就是右边的节点,然后加下一层的节点,这个加的时候要先加右边的节点,这样输出的时候就会先输出左边的节点,正好符合下一层的输出。
class Solution {
public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
List<List<Integer>> res=new ArrayList<>();
if(root==null)return res;
Deque<TreeNode> dq=new LinkedList<>();
dq.offer(root);
boolean flag=false;
while(!dq.isEmpty()){
int size=dq.size();
Deque<Integer> list=new LinkedList<>();
for(int i=0;i<size;i++){
TreeNode node=dq.poll();
if(flag) list.addFirst(node.val);
else list.addLast(node.val);
if(node.left!=null)dq.offer(node.left);
if(node.right!=null)dq.offer(node.right);
}
res.add(new ArrayList(list));
flag=!flag;
}
return res;
}
}