AcWing 44. 分行從上往下打印二叉樹

地址 https://www.acwing.com/problem/content/description/42/

題目描述
從上到下按層打印二叉樹，同一層的結點按從左到右的順序打印，每一層打印到一行。

樣例

樣例
輸入以下圖所示二叉樹[8, 12, 2, null, null, 6, null, 4, null, null, null]
    8
   / \
  12  2
     /
    6
   /
  4

輸出：[[8], [12, 2], [6], [4]]

 

算法1
在上一題的基礎上修改代碼 使用null指針做爲分割
注意邊界條件

C++ 代碼

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> res;
    queue<TreeNode*>  que;

    void bfs()
    {

        while(!que.empty()){
            vector<int> lineVec;
            while(!que.empty() && NULL != que.front() ){
                TreeNode* p = que.front();
                que.pop();
                lineVec.push_back(p->val);
                if(p->left!=NULL)
                    que.push(p->left);
                if(p->right!=NULL)
                    que.push(p->right);
            }
            res.push_back(lineVec);
            if(!que.empty()){
                que.pop();
            }
            if(!que.empty()){
                que.push(NULL);
            }
        }
    }


    vector<vector<int>> printFromTopToBottom(TreeNode* root) {
        if(root == NULL) return res;
        que.push(root); que.push(NULL);
        bfs();

        return res;
    }
};

做者：defddr
連接：https://www.acwing.com/solution/acwing/content/3661/
來源：AcWing
著做權歸做者全部。商業轉載請聯繫做者得到受權，非商業轉載請註明出處。