每天一道leetcode：剑指 Offer 32 - II. 从上到下打印二叉树 II（适合初学者）

今日份题目：

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

示例

例如:

给定二叉树: `[3,9,20,null,null,15,7]`,

```

3

/ \

9 20

/ \

15 7

```

返回其层次遍历结果：

```

\[3\], \[9,20\], \[15,7

]

```

提示

1. `节点总数 <= 1000`

题目思路

与上篇文章I中说的方法一样，都是使用BFS广度优先遍历对树进行层次遍历，与I不同处在于同层需要在一行vector中，由于队列中存在的元素是同一行的所有元素，所以每次对队列操作的时候，先插入一个空行（vector），然后在这个空行中放入同层元素。队列为空时操作结束，返回结果vector。

代码

/**
 * 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>> levelOrder(TreeNode* root) 
    {
        vector<vector<int> > ans;
        if(root==NULL) return {};
        queue<TreeNode*> p;
        //BFS
        p.push(root);
        while(!p.empty())
        {
            int n=p.size();
            ans.push_back(vector<int> ());
            for(int i=0;i<n;i++)
            {
                TreeNode* cur=p.front();
                p.pop();
                ans.back().push_back(cur->val);
                if(cur->left) p.push(cur->left);
                if(cur->right) p.push(cur->right);
            } 
        }
        return ans;
    }
};

提交结果

欢迎大家在评论区讨论，如有不懂的代码部分，欢迎在评论区留言！