剑指offer-将二叉树打印为多行

ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
if (pRoot == null) {
return result;
}

TreeNode s1 = null;
TreeNode s2 = null;

Queue<TreeNode> queue1 = new LinkedList<>();
Queue<TreeNode> queue2 = new LinkedList<>(); // 备用队列（需要画图理解）

queue1.add(pRoot);
while (!queue1.isEmpty()) {
ArrayList<Integer> list = new ArrayList<Integer>();

// 将一层的所有值放入list中（queue1的作用），清空queue1并将数据转移到queue2
while (!queue1.isEmpty()) {
s1 = queue1.poll();
queue2.add(s1);
list.add(s1.val);
}
result.add(list);

// queue2保存相当于复制没有清空的queue1，然后将下一层所有节点转给queue1
while (!queue2.isEmpty()) {
s2 = queue2.poll();
if (s2.left != null) {
queue1.add(s2.left);
}
if (s2.right != null) {
queue1.add(s2.right);
}
}
}

return result;
}