技術標籤：劍指offer佇列列表二叉樹資料結構java

題目描述

從上到下按層列印二叉樹，同一層結點從左至右輸出。每一層輸出一行。

分析

從題目可以得知，按層列印，並且每層都放在一個ArrayList中，最後再外巢狀一個ArrayList。這樣就可以達到 [[8],[6,10],[5,7,9,11]] 效果。

我們先看看層次遍歷：

public class PrintFromTopToBottom {

    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
        ArrayList<
 
Integer> list=new ArrayList<Integer>();  //存放結果
        Queue<TreeNode> queue= new LinkedList<TreeNode>();   //輔助佇列
        if (root == null){
            return list;
        }
        queue.offer(root); //根節點入隊
        //佇列不為空，執行迴圈
        while (!queue.isEmpty()){
            TreeNode node=
 
queue.poll();
            list.add(node.val);     //將佇列元素輸出

            //如果有左節點，就把左節點加入
            if (node.left!=null){
                queue.offer(node.left);
            }
            //如果有右節點，就把右節點加入
            if(node.right!=null){
                queue.offer(node.right);
            }
        }
        return
 
 list;
    }
}

在此基礎上我們可以進行修改新增，加一個curList，表示每層需要列印的結點集合，如[8]，[6,10]，然後加一個for迴圈，將每個結點都新增進去，注意每次都要新建立一個curList，這樣每次都是新的，不會有之前的元素。

本題實現程式碼如下：

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

public class Solution {
    //從上倒下按層列印，即層次遍歷，利用佇列輔助，再利用另一個ArrayList存每一行的元素值
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> list = new ArrayList<>();
        if (pRoot == null) {
            return list;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(pRoot);
        // 當前處理的元素
        TreeNode curNode;
        while (!queue.isEmpty()) {
            // 當前層的節點數量
            int size = queue.size();
            // 當前層的節點值，按照應該列印的順序新增到列表中
            ArrayList<Integer> curList = new ArrayList<>();
            // 遍歷當前層的元素
            for (int i=0; i<size; i++) {
                curNode = queue.poll(); //彈出
                System.out.println(curNode.val); //列印訪問
                curList.add(curNode.val);
                // 新增左右孩子
                if (curNode.left != null) {
                    queue.offer(curNode.left);
                }
                if (curNode.right != null) {
                    queue.offer(curNode.right);
                }
           }
            // 當前行元素遍歷的列表加入list
            list.add(curList);
        }
        return list;
    }
    
}