package com.zyl.algorithm;
import java.util.LinkedList;
public class PrintBinaryTreeByFloor {
public static void main(String[] args) {
// 初始化一颗二叉树
Node head = new Node(12);
head.left = new Node(34);
head.right = new Node(32);
head.left.left = new Node(1);
head.left.right = new Node(2);
head.right.right = new Node(3);
head.left.left.left = new Node(5);
head.left.left.right = new Node(6);
print(head);
}
/*
* 要求:
* 按层遍历二叉树,打印换行
* 思路:
* 使用两个标记变量last和nlast,分别表示当前行的最后一个和下一行的最后一个,都初始化为head节点
* 使用队列LinkedList,先放入head节点,之后每抛出一个节点就把它不为空的孩子节点加入队列
* 1.抛出一个节点
* 2.如果其左右节点不为空则加入其孩子节点,没加入一个节点,将其值赋值给nlast节点
* 3.如果抛出的节点是last节点,说明一行打印完毕,需要换行.且现在的nlast节点是下一行的最右节点,将其赋值给last. 否则正常打印不换行
* 4.每加入一个节点就将其赋值给nlast,表示现有的最右节点
* 5.如果队列不为空则继续
*/
public static void print(Node head) {
LinkedList<Node> queue = new LinkedList<>();
Node last = head;
Node nlast = head;
queue.push(head);
while (queue.size() != 0) {
Node node = queue.removeFirst();
if (node.left != null) {
queue.add(node.left);
nlast = node.left;
}
if (node.right != null) {
queue.add(node.right);
nlast = node.right;
}
if (node.equals(last)) {
System.out.println(node.value);
last = nlast;
} else {
System.out.print(node.value + " ");
}
}
}
}
class Node {
int value;
Node left;
Node right;
public Node(int v) {
value = v;
}
}
按层遍历二叉树并打印换行
最新推荐文章于 2022-06-08 19:06:35 发布