共计 2757 个字符,预计需要花费 7 分钟才能阅读完成。
一、思路:
在按层遍历的根底上进行改写 。
(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
正文完