題目描述

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

實驗平臺：牛客網

解決思路：

這題跟之前的劍指offer-題23：從上往下列印二叉樹很類似，就是多了一個需要換行列印的條件。這裡增加了2個變數，一個是下一層需要列印的結點數:nextNodesNum，還有一個是這一層還需列印的剩餘結點數:NeedToAdd，當其值為0時，表示這一層已經列印完畢。程式碼如下：

java:

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


/*
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>> list = new ArrayList<ArrayList<Integer>>();
        if (pRoot != null) {
            Queue<TreeNode> queue = new LinkedList<TreeNode>
 
();
            ArrayList<Integer> oneRow = new ArrayList<>();
            queue.offer(pRoot);
            int nextNodesNum = 0;
            int NeedToAdd = 1;
            while (!queue.isEmpty()) {
                TreeNode tempNode = queue.poll();
                oneRow.add(new Integer(tempNode.
 
val));
                if (tempNode.left != null) {
                    queue.offer(tempNode.left);
                    nextNodesNum++;
                }
                if (tempNode.right != null) {
                    queue.offer(tempNode.right);
                    nextNodesNum++;
                }
                NeedToAdd--;
                if (NeedToAdd == 0) {
                    list.add(oneRow);
                    oneRow = new ArrayList<Integer>();
                    NeedToAdd = nextNodesNum;
                    nextNodesNum = 0;
                }
            }
        }
        return list;
    }
}

python:

略