Gane la oferta de la impresión del árbol binario en varias líneas

título Descripción

Impresión de la árbol binario en capas de arriba a abajo, de izquierda a derecha con el nodo de la capa de salida. Cada línea de la capa de salida.

pensamiento

El uso de la idea de la cola

código

class Solution {
public:
    vector<vector<int> > Print(TreeNode* pRoot) {
        vector<vector<int> > res;
        vector<int> res_p;
        if(pRoot == NULL)
            return res;
        
        queue<TreeNode*> q_Tree;
        q_Tree.push(pRoot);
        res_p.push_back(pRoot->val);
        res.push_back(res_p);
        res_p.clear();
        int count = 1;
        
        while(q_Tree.size()>0)
        {
            int number = q_Tree.size();
            while(number > 0)
            {
                TreeNode* current = q_Tree.front();
                q_Tree.pop();
                if(current->left)
                {
                    q_Tree.push(current->left);
                    res_p.push_back(current->left->val);
                }
                if(current->right)
                {
                    q_Tree.push(current->right);
                    res_p.push_back(current->right->val);
                }
                number--;
            }
            if(res_p.size()>0)
            {
                res.push_back(res_p);
                res_p.clear();
            }
            count++;
        }
        return res;
    }
};