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

按之字形顺序打印二叉树

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

基本思路

难点在于不同层怎么按不同顺序打印节点值,不同层可以通过记录层数值判断是奇数层还是偶数层,打印节点值首先想的是如果偶数层就将队列中的下一层节点顺序反序,但是在for循环中加入下一层的节点一直是按左子树到右子树的顺序进行的,如果下一层节点的正确顺序为[5,7,9,11],反转前队列中节点为[9,11,5,7],反转后[7,5,11,9],还是不对。

正确做法是判断不同层后,将当前层数组保存的节点值反序就行了,因为数组中是从左到右的顺序存储节点,反转后是从右到左存储节点。

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 {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param pRoot TreeNode类 
     * @return int整型ArrayList<ArrayList<>>
     */
    public ArrayList<ArrayList<Integer>> Print (TreeNode pRoot) {
        // write code here
        ArrayList<ArrayList<Integer>> ret = new ArrayList<>();
        if (pRoot == null) {
            return ret;
        }

        int index = 1;
        Queue<TreeNode> q = new ArrayDeque<>();
        q.offer(pRoot);

        while(!q.isEmpty()) {
            int n = q.size();
            ArrayList<Integer> row = new ArrayList<>();

            index++;
            for (int i = 0; i < n; ++i) {
                TreeNode current = q.poll();
                row.add(current.val);
                if (current.left != null) {
                    q.offer(current.left);
                }
                if (current.right != null) {
                    q.offer(current.right);
                }
            }
            if (index % 2 != 0) {
                Collections.reverse(row);
            }
            ret.add(row);
        }

        return ret;
    }
}

全部评论

相关推荐

06-15 02:05
已编辑
南昌航空大学 数据分析师
Eason三木:你如果想干技术岗,那几个发公众号合唱比赛的经历就去掉,优秀团员去掉,求职没用。然后CET4这种不是奖项,是技能,放到下面的专业技能里或者单独列一个英语能力。 另外好好改改你的排版,首行缩进完全没有必要,行间距好好调调,别让字和标题背景黏在一起,你下面说能做高质量PPT你得展现出来啊,你这简历排版我用PPT做的都能比你做的好。 然后自我评价,你如果要干数据工程师,抗压能力强最起码得有吧。
简历中的项目经历要怎么写
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务