【劍指 offer】分行從上往下打印二叉樹

1、題目描述

從上往下打印出二叉樹的每個結(jié)點,同一層的結(jié)點按照從左到右的順序打印,每一層打印到一行。

樣例

輸入如下圖所示二叉樹[8, 12, 2, null, null, 6, null, 4, null, null, null]
? 8
? / \
12 2
??? /
? 6
? /
4
輸出:[[8], [12, 2], [6], [4]]

2、問題描述:

  • 分層打印,二叉樹。

3、問題關(guān)鍵:

  • 在每層的最后做個標記,每次遍歷到標記就換行。
  • 隊列,壓入當前層所有值后,再壓入一個NULL標記一下。
  • 如果當前層遍歷完,把當前層的值加入結(jié)果中,清空數(shù)組。

4、C++代碼:

class Solution {
public:
    vector<vector<int>> printFromTopToBottom(TreeNode* root) {
        vector<vector<int>> res;
        if (!root) return res;
        queue<TreeNode*> q;
        q.push(root);//壓入第一層。
        q.push(nullptr);//壓入標記。
        vector<int> level;
        while(q.size()) {
            auto t = q.front();
            q.pop();
            if(t) {//如果不是標記,就把值加入 數(shù)組。
                level.push_back(t->val); 
                if(t->left) q.push(t->left);//把左右孩子加入。
                if (t->right) q.push(t->right);
            }
            else {//遇到標記。
                if (level.size()) {//判斷數(shù)組是否為空,為空說明遍歷完了,break就可以了。
                    res.push_back(level);//把當前層加入結(jié)果中。
                    q.push(nullptr);給下一層最后加個標記。
                    level.clear();//清空數(shù)組,遍歷下一層。
                }
                else 
                    break;
            }
        }
        return res;
    }
};
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容