天天看點

劍指offer 從上往下列印二叉樹

題目描述

從上往下列印出二叉樹的每個節點,同層節點從左至右列印。

就是二叉樹的層次周遊,用隊列,根節點入隊列,每次列印一個節點時,如果該節點有子節點,則子節點放入隊尾,直到隊列中所有的節點都被列印出來。

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

/**
 * @author yuan
 * @date 2019/2/14
 * @description 層次周遊
 */
public class 從上往下列印二叉樹 {
    static class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;

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

        }

    }
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<Integer> result = new ArrayList<>();
        if (root == null) {
            return result;
        }

        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);

        while (!queue.isEmpty()) {
            TreeNode node = queue.poll();
            result.add(node.val);
            if (node.left != null) {
                queue.offer(node.left);

            }
            if (node.right != null) {
                queue.offer(node.right);
            }
        }
        return result;

    }
}