C++——二叉树的层序遍历

二叉树的层序遍历：一般基于队列的实现

• 首先将二叉树的根节点push到队列中。
• 判断队列不为空就输出队头元素。
• 判断当前对头节点是否有孩子节点，有则push到队列中。
• 循环操作，直到队列为空。

代码如下：

void FloorPrint_QUEUE(TreeNode* Tree) //层序遍历_队列实现
{
    queue < TreeNode* > q;
    if (Tree != NULL)
        q.push(Tree);   //根节点进队列
    while (q.empty() == false)  //队列不为空
    {
        TreeNode* node=q.front();
        cout << node->val; 

        if (node->left != NULL)   //如果有左孩子，入队
            q.push(node->left);   

        if (node->right != NULL)   //如果有右孩子，入队
            q.push(node->right);
        q.pop();  //已经遍历过的节点出队列
    }
}

本题由于对遍历的结果存储有要求，将同一层的数据存在一个容器中，因此对基本的层序遍历代码做了一些处理，用一个临时容器存储每层遍历的结果，然后push到最终求解的容器vec中，实现vec中的每个元素就是存储当前层数据的一个容器。

for循环中就是遍历某一层的数据处理过程，经过for循环，容器lev中存储的是某层的所有节点值，然后push到vec中，继续处理下一层，直到队为空也就是遍历完毕。

代码如下：

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 * };
 */

class Solution {
public:
    /**
     * 
     * @param root TreeNode类 
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > levelOrder(TreeNode* root) {
        // write code here
        vector<vector<int>> vec;
        if(root==nullptr) return vec;
        queue<TreeNode*> que;
        que.push(root);
        while(!que.empty())
        {
            int n=que.size();
             vector<int>lev;
            TreeNode* node;
            for(int i=0;i<n;i++)
            {
                node=que.front();
                que.pop();
                lev.push_back(node->val);
                if(node->left) que.push(node->left);
                if(node->right) que.push(node->right);
            }
            vec.push_back(lev);
        }
         return vec; 
    }
};