剑指offer面试题23 从上往下打印二叉树

解题思路:

考察层次遍历二叉树,可以利用一个队列来实现。当一个节点从队列头部移除时,若其存在左子节点,则左子节点入队列;若存在右子节点,则右子节点入队列,如此循环下去,直到队列为空,证明队列中的所有元素都已经访问过了,结束循环。

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;

class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}

public class Solution {

	/**
	 * 解题思路:
	 * 考察层次遍历二叉树,可以利用一个队列来实现。当一个节点从队列头部移除时,若其包含左子节点,则左子节点入队列;若包含右子节点,则右子节点
	 * 入队列,如此循环下去,指导队列为空结束循环
	 */
	public ArrayList PrintFromTopToBottom(TreeNode root) {
		
		ArrayList arrayList = new ArrayList<>();
		if (root == null) {
			return arrayList;
		}
		
		Queue queue = new LinkedList<>();
		//将根节点入队
		queue.offer(root);
		
		while (!queue.isEmpty()) {
			//取出队首元素
			TreeNode first = queue.peek();
			//访问队首元素
			arrayList.add(first.val);
			//队首元素出队列
			queue.poll();
			if (first.left != null) {
				queue.offer(first.left);
			}
			if (first.right != null) {
				queue.offer(first.right);
			}
		}
		return arrayList;
	}

}


你可能感兴趣的:(剑指offer面试题,剑指offer,面试题,算法,二叉树)