二叉树的层序遍历

二叉树的层序遍历

1、题目

题目要求：

给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。

示例：二叉树：[3,9,20,null,null,15,7]

    3
   / \
  9  20
 /  \
15   7

返回其层次遍历结果：

[
  [3],
  [9,20],
  [15,7]
]

2、代码思路

先将根节点入队，然后执行以下操作，直至队列为空

1. 将该层的节点从队列中取出，放入待返回的集合中
2. 将该层的节点的下一层节点依次放入队列中

3、代码实现

3.1、二叉树节点的定义

// 二叉树节点的定义
class Node {
	public Integer data;
	public Node left;
	public Node right;

	public Node(Integer data) {
		this.data = data;
	}

	public String toString() {
		return data.toString();
	}
}

3.2、二叉树的定义

// 二叉树的定义
class BinaryTree {
	public Node root;

	public List<ArrayList<Node>> levelOrder() {
		if (root == null) {
			System.out.println("二叉树为空~~~");
		}

		// 队列
		Queue<Node> queue = new LinkedList<Node>();
		// 根节点加入队列
		queue.add(root);
		
		// 记录层序遍历的结果
		List<ArrayList<Node>> levelOrderSequence = new ArrayList<ArrayList<Node>>();
		
		// 一直输出，直到队列为空
		while (!queue.isEmpty()) {
			// 记录当前层有多少个
			int curLevelCount = queue.size();
			
			// 当前层节点的集合
			ArrayList<Node> curLevelNodes = new ArrayList<Node>();
			
			for (int i = 0; i < curLevelCount; i++) {
				// 当前层的节点
				Node curNode = queue.remove();
				
				// 添加至集合中
				curLevelNodes.add(curNode);

				// 如果左节点不为空，加入队列（在下一层排队呢）
				if (curNode.left != null) {
					queue.add(curNode.left);
				}

				// 如果左节点不为空，加入队列（在下一层排队呢）
				if (curNode.right != null) {
					queue.add(curNode.right);
				}
			}
			
			// 将该层的节点集合放入集合中
			levelOrderSequence.add(curLevelNodes);
		}

		return levelOrderSequence;
	}

}

3.3、测试代码

代码

public static void main(String[] args) {

    Node node1 = new Node(1);
    Node node2 = new Node(2);
    Node node3 = new Node(3);
    Node node4 = new Node(4);
    Node node5 = new Node(5);
    Node node6 = new Node(6);
    Node node7 = new Node(7);

    // 构造二叉树
    node1.left = node2;
    node1.right = node3;
    node2.left = node4;
    node2.right = node5;
    node3.left = node6;
    node3.right = node7;

    BinaryTree binaryTree = new BinaryTree();
    binaryTree.root = node1;

    // 执行层序遍历
    List<ArrayList<Node>> allLevelNodes = binaryTree.levelOrder();
    for (ArrayList<Node> curLevelNodes : allLevelNodes) {
        System.out.println(curLevelNodes.toString());
    }
}

程序运行结果

[1]
[2, 3]
[4, 5, 6, 7]

3.4、全部代码

public class BinaryTreeLevelOrder {

	public static void main(String[] args) {

		Node node1 = new Node(1);
		Node node2 = new Node(2);
		Node node3 = new Node(3);
		Node node4 = new Node(4);
		Node node5 = new Node(5);
		Node node6 = new Node(6);
		Node node7 = new Node(7);

		// 构造二叉树
		node1.left = node2;
		node1.right = node3;
		node2.left = node4;
		node2.right = node5;
		node3.left = node6;
		node3.right = node7;

		BinaryTree binaryTree = new BinaryTree();
		binaryTree.root = node1;

		// 执行层序遍历
		List<ArrayList<Node>> allLevelNodes = binaryTree.levelOrder();
		for (ArrayList<Node> curLevelNodes : allLevelNodes) {
			System.out.println(curLevelNodes.toString());
		}
	}
    
}

// 二叉树的定义
class BinaryTree {
	public Node root;

	public List<ArrayList<Node>> levelOrder() {
		if (root == null) {
			System.out.println("二叉树为空~~~");
		}

		// 队列
		Queue<Node> queue = new LinkedList<Node>();
		// 根节点加入队列
		queue.add(root);
		
		// 记录层序遍历的结果
		List<ArrayList<Node>> levelOrderSequence = new ArrayList<ArrayList<Node>>();
		
		// 一直输出，直到队列为空
		while (!queue.isEmpty()) {
			// 记录当前层有多少个
			int curLevelCount = queue.size();
			
			// 当前层节点的集合
			ArrayList<Node> curLevelNodes = new ArrayList<Node>();
			
			for (int i = 0; i < curLevelCount; i++) {
				// 当前层的节点
				Node curNode = queue.remove();
				
				// 添加至集合中
				curLevelNodes.add(curNode);

				// 如果左节点不为空，加入队列（在下一层排队呢）
				if (curNode.left != null) {
					queue.add(curNode.left);
				}

				// 如果左节点不为空，加入队列（在下一层排队呢）
				if (curNode.right != null) {
					queue.add(curNode.right);
				}
			}
			
			// 将该层的节点集合放入集合中
			levelOrderSequence.add(curLevelNodes);
		}

		return levelOrderSequence;
	}

}

// 二叉树节点的定义
class Node {
	public Integer data;
	public Node left;
	public Node right;

	public Node(Integer data) {
		this.data = data;
	}

	public String toString() {
		return data.toString();
	}
}