前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >二叉树最大宽度与leetcode662的二叉树最大宽度

二叉树最大宽度与leetcode662的二叉树最大宽度

作者头像
gzq大数据
发布2021-11-24 13:28:01
4080
发布2021-11-24 13:28:01
举报
文章被收录于专栏:大数据那些事大数据那些事

首先先实现二叉树的最大宽度,也就是某一层最多的节点个数,使用了两种方法,第一种是使用一个map数组盛放当前节点以及当前节点的层数。

代码语言:javascript
复制
public static int maxWidthUseMap(Node head) {
		if (head == null) {
			return 0;
		}
		Queue<Node> queue = new LinkedList<>();
		queue.add(head);
		// key 在 哪一层,value
		HashMap<Node, Integer> levelMap = new HashMap<>();
		levelMap.put(head, 1);
		int curLevel = 1; // 当前你正在统计哪一层的宽度
		int curLevelNodes = 0; // 当前层curLevel层,宽度目前是多少
		int max = 0;
		while (!queue.isEmpty()) {
			Node cur = queue.poll();
			int curNodeLevel = levelMap.get(cur);
			if (cur.left != null) {
				levelMap.put(cur.left, curNodeLevel + 1);
				queue.add(cur.left);
			}
			if (cur.right != null) {
				levelMap.put(cur.right, curNodeLevel + 1);
				queue.add(cur.right);
			}
			if (curNodeLevel == curLevel) {
				curLevelNodes++;
			} else {
				max = Math.max(max, curLevelNodes);
				curLevel++;
				curLevelNodes = 1;//已经有一个当前节点了
			}
		}
		max = Math.max(max, curLevelNodes);
		return max;
	}

第二种方法是使用两个linkedList,每一次倒空都代表着一层的结束:

代码语言:javascript
复制
 private static int findMax(TreeNode node) {
        if (node == null) {
            return 0;
        }
        LinkedList<TreeNode> nodes = new LinkedList<>();
        LinkedList<TreeNode> cache = new LinkedList<>();
        int max = -1;
        nodes.add(node);
        while (true) {
            while (!nodes.isEmpty()) {//一次倒空的循环
                max = Math.max(nodes.size(), max);
                TreeNode poll = nodes.poll();
                if (poll.left != null) {
                    cache.add(poll.left);
                }
                if (poll.right != null) {
                    cache.add(poll.right);
                }
            }
            if (cache.size() == 0) {
                break;//如果此时下一层已经没有了就没有必要继续了,直接跳出循环
            }
            LinkedList<TreeNode> tmp = nodes;//进行互换,这样子此时的nodes就变成了下一层的所有nodes
            nodes = cache;
            cache = tmp;
        }
        return max;
    }

leetcode上的变种:

在这里插入图片描述
在这里插入图片描述

代码实现如下,使用了while里面套一层for循环,循环里面是存每一层的有节点的最左和最右下标的,并且这里每一次for循环就是一层的循环结束:

代码语言:javascript
复制
  private static int findMax(TreeNode node) {
        if (node == null) {
            return 0;
        }
        int max = 1;
        LinkedList<TreeNode> nodes = new LinkedList<TreeNode>();
        LinkedList<Integer> list = new LinkedList<Integer>();
         list.add(0);
            
        nodes.offer(node);
        while(!nodes.isEmpty()){
        int count = nodes.size();
       
            for(;count>0;count--){
            TreeNode poll = nodes.poll();
            int curIndex = list.removeFirst();
            if(poll.left!=null){
                int left = curIndex*2+1;
                list.offer(left);
                nodes.offer(poll.left);
            }
            if(poll.right!=null){
                int right = curIndex*2+2;
                list.offer(right);
                nodes.offer(poll.right);
            }
            
            }
            if(list.size()>1){/// list 中 size 为 1 的情况下,宽度也为 1,没有必要计算。
              max = Math.max(max,list.getLast()-list.getFirst()+1);
            }
        }
        return max;
}
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2021-11-21 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档