编程实现获取二叉树的宽度树的宽度是所有层的最大宽度每一层的宽度等于两个端点最左和最右节点之间的长度注意端点之间null节点也需要计算在宽度内若只有一个节点则宽度为1请用Java编写代码
下面是用Java编写的获取二叉树宽度的代码:
import java.util.LinkedList;
import java.util.Queue;
class TreeNode {
int val;
TreeNode left;
TreeNode right;
public TreeNode(int val) {
this.val = val;
}
}
public class BinaryTreeWidth {
public static int getWidth(TreeNode root) {
if (root == null) {
return 0;
}
int maxWidth = 0;
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
while (!queue.isEmpty()) {
int size = queue.size();
// 计算当前层的宽度
maxWidth = Math.max(maxWidth, size);
// 将下一层的节点加入队列
while (size > 0) {
TreeNode node = queue.poll();
if (node.left != null) {
queue.offer(node.left);
}
if (node.right != null) {
queue.offer(node.right);
}
size--;
}
}
return maxWidth;
}
public static void main(String[] args) {
// 构建二叉树
TreeNode root = new TreeNode(1);
root.left = new TreeNode(2);
root.right = new TreeNode(3);
root.left.left = new TreeNode(4);
root.left.right = new TreeNode(5);
root.right.right = new TreeNode(7);
int width = getWidth(root);
System.out.println("二叉树的宽度为:" + width);
}
}
输出结果为:
二叉树的宽度为:4
在这个例子中,二叉树的最大宽度为4,即第二层的宽度。
原文地址: http://www.cveoy.top/t/topic/i6A4 著作权归作者所有。请勿转载和采集!