一、思路:
在按层遍历的根底上进行改写。
(1)筹备currentLevelEnd(示意以后层的完结节点)、nextLevelEnd(示意下一层的完结节点)、currentWeight(示意以后层的宽度)、maxWeight(示意最大宽度)四个变量。
(2)currentLevelEnd批改为二叉树的头节点(其余均为默认值),头节点入队列。
(3)从队列中弹出一个节点,每出队一个节点则currentWeight加一,出队节点有左孩子则左孩子入队,有右孩子则右孩子入队,有节点入队则将nextLevelEnd批改为以后入队的节点(为下一层查找做筹备)。
(4)判断第3步出队的节点是否等于currentLevelEnd。
是,则批改maxWeight为以后最大宽度,currentWeight重置,currentLevelEnd批改为nextLevelEnd,nextLevelEnd重置;
否,则继续执行第3步。
(5)始终执行第3、4步,直到队列为空。
/** * @author Java和算法学习:周一 */public static int treeMaxWidth(Node head) { if (head == null) { return 0; } Queue<Node> queue = new LinkedList<>(); Node currentLevelEnd = head; Node nextLevelEnd = null; int currentWidth = 0; int maxWidth = 0; queue.offer(head); while (!queue.isEmpty()) { Node current = queue.poll(); // 每次从队列中弹出节点时,以后层的宽度都加一 currentWidth++; // 有孩子节点入队时就批改下一层的完结节点 if (current.left != null) { queue.offer(current.left); nextLevelEnd = current.left; } if (current.right != null) { queue.offer(current.right); nextLevelEnd = current.right; } // 如果以后出队的节点是以后层的完结节点 if (current == currentLevelEnd) { // 批改最大宽度 maxWidth = Math.max(maxWidth, currentWidth); // 以后层宽度重置 currentWidth = 0; // 以后层的完结节点批改,表明下一次循环时开始统计下一层的宽度 currentLevelEnd = nextLevelEnd; // 下一层完结节点重置 nextLevelEnd = null; } } return maxWidth;}
二、对数器源码
/** * 对数器办法 * * @author Java和算法学习:周一 */public static int treeMaxWidth1(Node head) { if (head == null) { return 0; } Queue<Node> queue = new LinkedList<>(); queue.add(head); // key:节点,value:节点在哪一层 HashMap<Node, Integer> levelMap = new HashMap<>(16); levelMap.put(head, 1); // 以后来到哪一层 int curLevel = 1; // 以后层的宽度 int curLevelWidth = 0; // 最大宽度 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) { curLevelWidth++; } else { // 以后层曾经遍历结束 max = Math.max(max, curLevelWidth); // 来到下一层 curLevel++; // 以后节点曾经是下一层的了,也就是下一层曾经有一个节点了,宽度天然是1 curLevelWidth = 1; } } // 最初一层的宽度没有和max比拟,所以这里要再比拟一次 max = Math.max(max, curLevelWidth); return max;}
本文全副代码:https://github.com/monday-pro/algorithm-study/blob/master/src/basic/binarytree/TreeMaxWidth.java