二叉树的层序遍历

二叉树的层序遍历算是一道经典的算法题目，面试中经常会碰到，特此总结一下：

/**************************************************
* 1.二叉树层序遍历，基于广度优先搜索
* 2.程序借助队列进行操作，先将根节点压入队列，定义一个temp节点，temp为当前要打印的节点，判断temp的左右子节点，存在则压入队列，为后续输出做准备
* 3.每层需要一个换行操作，为此定义两个值，parentsize来记录当前层未遍历节点数目，childsize记录下一层节点数目
* 4.当parentsize == 0时，进入换行操作，将parentsize更新为下一行的节点数目
*/
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;

struct Node{
    Node *pleft;
    Node *pright;
    int  data;
};
//声明函数
void PrintNodeByLevel(Node *root);
void Link(Node *nodes, int parent, int left, int right);

int main()
{
    Node test1[9] = {0};  
    int i;  

    for (i = 0; i < 9; i++)   
    {  
        test1[i].data = i;  
    }  

    Link(test1, 1, 2, 3);  
    Link(test1, 2, 4, 5);  
    Link(test1, 3, -1, 6);  
    Link(test1, 5, 7, 8);  

    PrintNodeByLevel(&test1[1]);  

    return 0;  
}

//构造树
void Link(Node *nodes, int parent, int left, int right)
{
    if (left != -1)   
    {  
        nodes[parent].pleft = &nodes[left];  
    }  
    if (right != -1)   
    {  
        nodes[parent].pright = &nodes[right];  
    }  
}

//层序遍历函数
void PrintNodeByLevel(Node *root)
{
    int parentsize = 1, childsize = 0;
    Node *temp;
    queue<Node *> q;    //q是一个node*类型的队列
    q.push(root);
    do
    {
        temp = q.front();
        cout << temp->data << " ";
        q.pop();
        if(temp->pleft != NULL)
        {
            q.push(temp->pleft);
            childsize++;
        }
        if(temp->pright != NULL)
        {
            q.push(temp->pright);
            childsize++;
        }

        parentsize--;
        if(parentsize == 0)
        {
            parentsize = childsize;
            childsize = 0;
            cout << endl;
        }
    } while(!q.empty());
}