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

按之字形顺序打印二叉树

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

增加标志位,隔一行进行翻转

import java.util.*;

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        //增加一个标志变换方向
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        if(pRoot==null) return res;
        Queue<TreeNode> queue = new LinkedList<>();
        boolean flag = true;
        queue.add(pRoot);
        while(!queue.isEmpty()){
            int size = queue.size();
            ArrayList<Integer> temp = new ArrayList<>();
            for(int i = 0;i < size;i++){
                pRoot = queue.poll();
                if(pRoot.left!=null) queue.add(pRoot.left);
                if(pRoot.right!=null) queue.add(pRoot.right);
                temp.add(pRoot.val);
            }
            if(!flag) Collections.reverse(temp);
            res.add(temp);
            flag = !flag;
            
        }
        return res;
    }

}
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务