剑指Offer第二十二题：从上往下打印二叉树（二叉树的层次遍历）

题目描述

从上往下打印出二叉树的每个节点，同层节点从左至右打印。

 

思路：这里思路还是引用下某大神的，利用队列实现。

1. 如上图所示，因为按层打印的顺序决定应该先打印根结点，所以我们从树的根结点开始分析。为了接下来能够打印值为8的结点的两个子结点，我们应该在遍历该结点时把值为6和10的两个结点保存到一个容器里，现在容器内就有两个结点了。
2. 按照从左到右打印的要求，我们先取出为6的结点。打印出值6之后把它的值分别为5和7的两个结点放入数据容器。此时数据容器中有三个结点，值分别为10、5和7。
3. 接下来我们从数据容器中取出值为10的结点。注意到值为10的结点比值为5、7的结点先放入容器，此时又比这两个结点先取出，这就是我们通常说的先入先出，因此不难看出这个数据容器应该是一个队列。由于值为5、7、9、11的结点都没有子结点，因此只要依次打印即可。

整个打印过程如下图所示：

代码如下：

class Solution {
public:
    vector<int> PrintFromTopToBottom(TreeNode* root) {
        vector<int>Data;
        if(root == NULL)
            return Data;
        
        queue<TreeNode*>mq;
            
        mq.push(root);
        while(!mq.empty())
        {
            Data.push_back(mq.front()->val);

            //弹出前，先把弹出元素的左右子树压入队列
            if(mq.front()->left != NULL)
                mq.push(mq.front()->left);
            if(mq.front()->right != NULL)
                mq.push(mq.front()->right);
            mq.pop();
        }   
        return Data;
    }  
};