码迷,mamicode.com
首页 > 其他好文 > 详细

从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。

时间:2018-03-12 22:57:01      阅读:598      评论:0      收藏:0      [点我收藏+]

标签:ret   lin   array   node   print   tree   入队   linked   ISE   

这里需要使用两个变量:now 当前层没有打印的节点个数
next 下一层的节点个数
每当当前层打印完毕 就把下一的节点个数(next)赋值给now

import java.util.ArrayList;
import java.util.*;

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

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

    }

}
*/
/*
这里需要使用两个变量:now  当前层没有打印的节点个数
                  next 下一层的节点个数
  每当当前层打印完毕   就把下一的节点个数(next)赋值给now
*/
public class Solution {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
         ArrayList<ArrayList<Integer>> container=new ArrayList<ArrayList<Integer>>();
        if (pRoot==null) return container;
       ArrayList<Integer>inner=new ArrayList<Integer>();
       Queue<TreeNode> queue= new LinkedList<TreeNode>();
       queue.offer(pRoot);
       int now=1;//当前层中没有打印的节点数   根节点没有打印 初始值取1
       int next=0;//下一层中的节点总数   初始时  下一层还没有入队列  未统计  初始取值为0
       while(!queue.isEmpty()){
           TreeNode t=queue.poll();
           inner.add(t.val);
           now--;
           if(t.left!=null){
               queue.offer(t.left);
               next++;
           }
           if(t.right!=null){
               queue.offer(t.right);
               next++;
           }
           if(now==0){//当前层打印完成   把下一层的节点总数赋值给now   next清零
               container.add(inner);
               inner=new ArrayList<Integer>();
               now=next;
               next=0;
           }
       }
        return container;
    }
    
}

  

从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。

标签:ret   lin   array   node   print   tree   入队   linked   ISE   

原文地址:https://www.cnblogs.com/bolianggufeng/p/8552152.html

(0)
(1)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!