题解 | #按之字形顺序打印二叉树#

按之字形顺序打印二叉树

http://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0

题解:层序遍历 + 双端队列

import java.util.*;

public class Solution {
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> list = new ArrayList<>();
        if(pRoot!=null){
            Deque<TreeNode> q = new LinkedList<>();
            q.add(pRoot);
            boolean leftToRight = true;
            while(!q.isEmpty()){
                int size = q.size();
                ArrayList<Integer> level = new ArrayList<>();
                for(int i=0;i<size;i++){
                    if(leftToRight){
                        TreeNode cur = q.removeFirst();
                        level.add(cur.val);
                        if(cur.left!=null)    q.addLast(cur.left);
                        if(cur.right!=null)    q.addLast(cur.right);
                    }else{
                        TreeNode cur = q.removeLast();
                        level.add(cur.val);
                        if(cur.right!=null)    q.addFirst(cur.right);
                        if(cur.left!=null)    q.addFirst(cur.left);
                    }
                    
                }
                list.add(level);
                leftToRight = !leftToRight;
            }
        }
        return list;
    }

}
全部评论

相关推荐

Cassifa:发的字比你都多的一律视为骗子或者想白嫖压榨实习生的
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务