二叉树的层次遍历是指从根节点开始,沿着树的宽度一层一层向下遍历,比如下图的二叉树,层次遍历的顺序为:A-B-C-D-E-F-G,那层次遍历如何实现呢?
- 我们需要借助一个队列,由于队列是先进先出的,所以可以先让根入队,在根出队的同时打印根,并让根的左孩子入队,再让右孩子入队,这样一来,左孩子结点就存储在队头的位置,可以首先被访问;
- 被访问之后,左孩子出队的同时打印左孩子,并让左孩子的孩子入队;此时队列的下一个元素就是右孩子,右孩子出队的同时打印右孩子,并让右孩子的孩子入队;
- 以此类推,直到队列为空。
下面是C++实现树的BFS的源代码:
层序遍历用一维数组保存结果(20180421修改的新版本)
class Solution {
public:
vector<int> levelOrder(TreeNode* root)
{
vector<int> ret;
queue<TreeNode*> q;
if(root)
q.push(root);
while(!q.empty())
{
TreeNode* temp = q.front();
ret.push_back(temp->val);
q.pop();
if(temp->left)
q.push(temp->left);
if(temp->right)
q.push(temp->right);
}
return ret;
}
};