利用層次遍歷的演算法，設定變數last指向當前層的最後一個節點，設定變數count記錄當前層已經訪問的節點的個數，當count等於last時，表示該層訪問結束。

層次遍歷在求樹的寬度、輸出某一層節點，某一層節點個數，每一層節點個數都可以採取類似的演算法。

樹的寬度：在樹的深度演算法基礎上，加一個記錄訪問過的層節點個數最多的變數max,在訪問每層前max與last比較，如果max比較大，max不變，如果max小於last，把last賦值給max;

package niuke.easy;

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

public class PrintBinaryTree {

	public static void main(String[] args) {

	}

	/**
	 * 利用層次遍歷的演算法，設定變數last指向當前層的最後一個節點，設定變數count記錄當前層已經訪問的節點的個數，當count等於last時，
	 * 表示該層訪問結束。 層次遍歷在求樹的寬度、輸出某一層節點，某一層節點個數，每一層節點個數都可以採取類似的演算法。
	 * 樹的寬度：在樹的深度演算法基礎上，加一個記錄訪問過的層節點個數最多的變數max
	 * ,在訪問每層前max與last比較，如果max比較大，max不變，如果max小於last，把last賦值給max;
	 * 
	 * @param pRoot
	 * @return
	 */
	public ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
		ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();

		if (pRoot == null) {
			return list;
		}

		int count = 0;
		int last = 0;
		TreeNode current = pRoot;
		LinkedList<TreeNode> queue = new LinkedList<TreeNode>();

		queue.offer(current);

		while (!queue.isEmpty()) {
			count = 0;
			last = queue.size();
			ArrayList<Integer> tempList = new ArrayList<Integer>();

			// 列印一層
			while (count < last) {
				current = queue.pop();
				tempList.add(current.val);

				count++;
				if (current.left != null) {
					queue.offer(current.left);
				}
				if (current.right != null) {
					queue.offer(current.right);
				}
			}
			list.add(tempList);
		}

		return list;
	}
}
 

程式碼2：

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.*;

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

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

    }

}
*/
public class Solution {
     ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        //先把需要的準備好
        ArrayList<ArrayList<Integer>> arrayList = new ArrayList<>();
        ArrayList<Integer> list = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();//queue 是一個介面
         
        if(pRoot != null){
            int nextNodes = 0;
            int needPrint = 1;//初始狀態，下一層節點0，需要被列印的結點1，下一層節點在迴圈中遞增
            queue.add(pRoot);
             
            while(!queue.isEmpty()){
                //隊不空的情況下
                TreeNode node = queue.peek();
                list.add(node.val);//只需要把結點的值存起來就行了
                 
                if(node.left != null){
                    queue.add(node.left);
                    ++nextNodes;
                }
                 
                if(node.right != null){
                    queue.add(node.right);
                    ++nextNodes;
                }
                 
                queue.poll();//移除佇列一個元素
                 
                --needPrint;
                if(needPrint == 0){
                    arrayList.add(new ArrayList(list));//一行打完了，到下一行
                    list.clear();
                    needPrint = nextNodes;
                    nextNodes = 0;
                }
            }
        }
         
        return arrayList;
    }
    
}