二叉树的最大宽度

初始化节点

	public static class Node {
		public int value;
		public Node left;
		public Node right;

		public Node(int data) {
			this.value = data;
		}
	}

1.使用容器

public static int maxWidthUseMap(Node head) {
   if(head == null){
      return 0;
   }
   Queue<Node> queue = new LinkedList<>();
   queue.add(head);
   // level endNode
   Map<Integer, Node> map = new HashMap<>(16);
   int curLevel = 1;
   int size = 0;
   int maxSize = 0;
   map.put(curLevel,head);
   while (!queue.isEmpty()){
      head = queue.poll();
      size++;
      if(head.left != null ){
         queue.add(head.left);
         map.put(curLevel + 1 ,head.left);
      }
      if(head.right != null ){
         queue.add(head.right);
         map.put(curLevel + 1 ,head.right);
      }
      if(head == map.get(curLevel)){
         maxSize = Math.max(size,maxSize);
         size = 0;
         curLevel++;
      }
   }
   return maxSize;
} 

2.不使用容器

public static int maxWidthNoMap(Node head) {
   if(head == null){
      return 0;
   }
   Queue<Node> queue = new LinkedList<>();
   queue.add(head);


   // record tree level end node
   Node curEnd = head;
   Node nextEnd = null;

   int maxWidth = 0;
   int size = 0;
   while (!queue.isEmpty()){
      head = queue.poll();
      size++;
      if(head.left != null){
         queue.add(head.left);
         //update next level end
         nextEnd = head.left;
      }
      if(head.right != null){
         queue.add(head.right);
         //update next level end
         nextEnd = head.right;
      }
      // is current level end ,require max and reset size and curEnd = nextEnd
      if(head == curEnd){
         maxWidth = Math.max(size,maxWidth);
         size = 0;
         curEnd = nextEnd;
      }

   }
   return maxWidth;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值