二叉树的锯齿形层次遍历

flag

软件学院大三党，每天一道算法题，第十八天

题目介绍

给定一个二叉树，返回其节点值的锯齿形层次遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。

思路

迭代：采用经过加工的广度遍历，引入depth层数，逐层将元素放入链表（奇数层插入到尾部，偶数层插入到头部），用队列的长度代表每层的元素个数，即内层循环的次数，再将下一层元素放入队列。

递归：类似深度优先遍历

关键代码
迭代

public static List<List<Integer>> zigzagLevelOrder(TreeNode root) {

    List<List<Integer>> lists=new ArrayList<>();
    if(root==null)
        return lists;
    Queue<TreeNode>queue=new LinkedList<>();
    queue.add(root);
    int depth=0;
    while (!queue.isEmpty()){
        List<Integer> tmp = new LinkedList<>();
        int count=queue.size();
        for(int i=0;i<count;i++){
            TreeNode temp=queue.poll();
            if(depth%2==0)
                tmp.add(temp.val);
            else//从首部添加元素
                tmp.add(0,temp.val);
            if(temp.left!=null)
                queue.add(temp.left);
            if(temp.right!=null)
                queue.add(temp.right);
        }
        lists.add(tmp);
        depth++;

    }
    return lists;

}

递归

public static List<List<Integer>> zigzagLevelOrder2(TreeNode root) {
    List<List<Integer>> res = new ArrayList<>();
    helper(res, root, 0);
    return res;

}

public static void helper(List<List<Integer>> lists, TreeNode root, int depth) {
    if (root == null)
        return;
    if (lists.size() == depth)
        lists.add(new LinkedList<>());
    if (depth % 2 == 0)
        lists.get(depth).add(root.val);
    else 
        lists.get(depth).add(0, root.val);
    helper(lists, root.left, depth + 1);
    helper(lists, root.right, depth + 1);
}

测试

TreeNode test=new TreeNode(1);
test.left=new TreeNode(2);
test.left.right=new TreeNode(4);
test.right=new TreeNode(9);
test.right.left=new TreeNode(4);
test.right.right=new TreeNode(3);
List<List<Integer>>l=zigzagLevelOrder(test);
for(int i=0;i<l.size();i++)
    System.out.println(l.get(i));

结果：