题目要求逐层的扫描打印二叉树,并且每一层变换扫描顺序(从左至右或从右至左)。 思路就是从根节点开始将当前层的所有节点存入list中,扫描下一层的节点只需从这个list的尾部开始扫描加入子节点即可。 在这个过程中需要维护一个变量来记录当前是从左至右扫描还是从右至左扫描,从而确定是先扫描左子结点还是右子结点。
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public ArrayList<ArrayList<Integer>> zigzagLevelOrder(TreeNode root) {
ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
ArrayList<TreeNode> temp = new ArrayList<TreeNode>();
ArrayList<Integer> tmp = new ArrayList<Integer>();
if( root == null )
{
return res;
}
temp.add(root);
tmp.add(root.val);
res.add(tmp);
int right = 1;
while( temp.size() != 0 )
{
ArrayList<TreeNode> temp1 = new ArrayList<TreeNode>();
ArrayList<Integer> tmp1 = new ArrayList<Integer>();
if( right == 1 )
{
for( int i = temp.size()-1;i>=0;i-- )
{
if( temp.get(i).right != null )
{
temp1.add(temp.get(i).right);
tmp1.add( temp.get(i).right.val );
}
if( temp.get(i).left != null )
{
temp1.add(temp.get(i).left);
tmp1.add( temp.get(i).left.val );
}
}
}
else
{
for( int i = temp.size()-1;i>=0;i-- )
{
if( temp.get(i).left != null )
{
temp1.add(temp.get(i).left);
tmp1.add( temp.get(i).left.val );
}
if( temp.get(i).right != null )
{
temp1.add(temp.get(i).right);
tmp1.add( temp.get(i).right.val );
}
}
}
right = 1-right;
temp = temp1;
if( tmp1.size() != 0 )
{
res.add(tmp1);
}
}
return res;
}
}