题解 | #把二叉树打印成多行#

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
//本题用广度优先遍历即可，问题是怎么把一行的数都放在同一个数组里一起输出
//可以用两个变量实现，一个是nowLast表示当前行的最后一个元素，另一个变量是nextLaTeX表示下一行的最后一个元素
//nextLast每次更新成最近一个加入到队列中的元素
class Solution {
public:
    vector<int> B;
    vector<vector<int> >A;
    void bfs(TreeNode* root){
        if(root==NULL)
            return;
        queue<TreeNode*> que;
        TreeNode*nowLast=root,*nextLast=NULL;
        que.push(root);
        while(!que.empty()){//当队列不为空时
            TreeNode*node=que.front();
            que.pop();
            B.push_back(node->val);
            if(node->left!=NULL){
                que.push(node->left);
                nextLast=node->left;
            }
            if(node->right!=NULL){
                que.push(node->right);
                nextLast=node->right;
            }
            if(node==nowLast){//说明当前行已经来到尽头了，接下来就要换行了
                A.push_back(B);
                B.clear();
                nowLast=nextLast;
            }
        }
    }
        vector<vector<int> > Print(TreeNode* pRoot) {
            if(pRoot==NULL)
                return A;
        bfs(pRoot);
            return A;
        }
    
};