按层遍历二叉树并打印换行

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;
	}
}

猜你喜欢

转载自blog.csdn.net/qq_25944759/article/details/82966502