最大宽度,其实如果算元素的话,还是挺简单的,但是这道题,按照满二叉树来做,我被困在了这里面,做了一个多小时,没做出来,每次做出一个版本就出现问题,我也是醉了,下面列出题目:
说实话,也怪我没好好读题,以为是一个简单的层次遍历就OK了,但是里面的坑还是挺多的,这里面我就介绍一个简单的标记位置吧,看到答案的时候也是眼前一亮,把每一行都用上一层的标记节点进行标记,然后每次放在一个List中,用最后一个数减去第一个数+1就是这一行的长度了,下面列出代码:
private int maxW = 1;
public int widthOfBinaryTree(TreeNode root) {
if (root == null)
return 0;
Queue<TreeNode> queue = new LinkedList<>();
LinkedList<Integer> indexList = new LinkedList<>();
queue.add(root);
indexList.add(1);
int size = 1;
while (!queue.isEmpty()) {
TreeNode node = queue.poll();
size--;
int index = indexList.removeFirst();
if (node.left != null) {
queue.add(node.left);
indexList.add(2 * index);
}
if (node.right != null) {
queue.add(node.right);
indexList.add(2 * index + 1);
}
if (size == 0) {
if (indexList.size() >= 2) {
maxW = Math.max(maxW, indexList.getLast() - indexList.getFirst() + 1);
}
size = queue.size();
}
}
return maxW;
}
新颖的是不用特殊的处理各个情况,只是把所有的都进行标记进行规划就OK了