题解 | #把二叉树打印成多行#

把二叉树打印成多行

http://www.nowcoder.com/practice/445c44d982d04483b04a54f298796288

import java.util.*;
/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

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

    }

}
*/
这个题是之字形打印二叉树的一部分,也就是层序遍历二叉树
相对还是比较简单
总的思路是使用队列,维护的方式是取出队首并放入它的左右孩子
然后利用双重循环,每一次外层循环代表遍历完一层的节点,每一次内层循环代表遍历一个节点
外层维护一个用来装每一次节点VAL值的集合,以及一个用来统计下一层节点数的temp
内层执行取出队首并放入其左右子节点的操作,每取出一个节点temp加1.
内层结束把temp赋给sum,然后把row赋给总的集合
最后总循环结束,所以层数都被遍历完成,返回result总集合即可。
public class Solution {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
        if(pRoot == null)return result;//虽然是没什么必要,但是还是要判断一下数组越界
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        int sum = 1;//用来统计每层的个数
        queue.add(pRoot);
        while(!queue.isEmpty()){//注意==是数组判断,isEmpty是内容判断,所以判断引用类型用isEmpty
            int temp = 0;//用来存放下一层的节点个数
            ArrayList<Integer> row = new ArrayList<Integer>();
            while(sum > 0){
                TreeNode x = queue.poll();//取出队首节点
                row.add(x.val);//然后把它的值加入到行集合中
                sum--;
                if(x.left != null){
                    queue.add(x.left);
                    temp++;
                }
                if(x.right != null){
                    queue.add(x.right);
                    temp++;
                }
            }
            sum = temp;//要在整个一层搞定了再把temp赋值给sum。
            result.add(row);
        }
        return result;
    }
    
}
全部评论

相关推荐

头像
10-15 22:27
已编辑
门头沟学院 C++
罗格镇的小镇做题家:我投了hr打电话来说学历太低了不符合要求,建议投荣耀,结果荣耀也投了一定水花没有,非本211硕
投递华为等公司10个岗位
点赞 评论 收藏
分享
拉丁是我干掉的:把上海理工大学改成北京理工大学。成功率增加200%
点赞 评论 收藏
分享
1 收藏 评论
分享
牛客网
牛客企业服务