给定一棵二叉树,对该二叉树进行之字型的层次遍历


题目的意思就是说第1层(根结点)从左边开始遍历,在第二层则从右边开始遍历,第三层又从左边开始如此递归下去,直到遍历完整棵二叉树;

如给定一个二叉树的树组[1,2,3,4,null,null,5],null表示对应节点为空,则应该返回[[1],[3,2],[4,5]];详情可参考leetcode;
二叉树节点的定义如下:

public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }

具体的解决思路:用一个flag来标记是将当前的节点前插到还是后插到当前层的list中,遍历完一层后,flag赋值为它的非。其余思想和二叉树的层次遍历一样,用队列来保存需要访问的元素,直到队列为空;

LinkdeList中的add()方法可以指定插入的地点,默认add(object)是将元素插入到list已经存在的元素的后面,假设当前list中的元素为{1,2},则当我们执行完方法add(4)后,list中的元素位置为{1,2,4};;当我们指定插入的位置为0里add(0,object),则将元素插入到已有元素的前面,我们继续执行完add(0,5)后,则list中的元素就变为{5,1,2,4};

具体的代码如下:

public class Solution {
    public List<List<Integer>> zigzagLevelOrder(TreeNode root) {
        List<List<Integer>> result=new LinkedList<List<Integer>>();
        Queue<TreeNode> nodeQueue=new LinkedList<TreeNode>();
        boolean flag=true;
        if(null==root){
            return result;
        }
        nodeQueue.offer(root);
        while(0!=nodeQueue.size()){
            int size=nodeQueue.size();
            List level=new LinkedList<Integer>();
            for(int i=0;i<size;i++){
                 root=nodeQueue.remove();
                 if(null!=root.left){
                    nodeQueue.offer(root.left);
                 }
                 if(null!=root.right){
                    nodeQueue.offer(root.right);
                 }
                if(flag){
                  level.add(root.val);
                }else{
                 level.add(0,root.val);
                }
            }
            flag=!flag;
            result.add(level);
        }
        return result;
    }
}

代码在leetcode编译通过,我就不贴运行结果了

posted @ 2015-07-30 20:50  woodWu  阅读(896)  评论(0编辑  收藏  举报