关于分层遍历二叉树

来源:互联网 发布:外贸企业搜索软件 编辑:程序博客网 时间:2024/05/22 07:00

问题:给定一棵二叉树,要求按分层遍历二叉树,即从上到下按层次访问该二叉树,每一层要求访问顺序为从左到右。

解:首先从根节点出发,依次将每层的节点从左到右压入一个数组,并用一个游标cur记录当前访问的节点,另一个游标Lost指示当前层次的最后一个节点的下一个位置,以cur==last作为当前层次访问的结束条件,在访问某一层的同时将该层的所有节点的子节点压入数组,直到访问完所有层次。


代码:

#include <iostream>#include <vector>using namespace std;typedef struct Node  //节点存储格式{    int date;    Node* lchild;    Node* rchild;}Node,*PNode;int pos = 0;PNode buildtree(int* a, int N)//建立二叉树{    if(pos == N-1)        return NULL;    if(a[pos] == 0)    {        pos++;        return NULL;    }    PNode pnode = new Node;    pnode->lchild = NULL;    pnode->rchild = NULL;    pnode->date = a[pos];    pos++;    pnode->lchild = buildtree(a, N);    pnode->rchild = buildtree(a, N);    return pnode;}void priorder(Node *node)//先序遍历二叉树,主要作用是检查树是否建立完整。{    if(node)    {        cout << node->date << " ";        priorder(node->lchild);        priorder(node->rchild);    }}void printbylevel(PNode pnode){    if(pnode == NULL)        return;    vector<PNode> vec;//存储临时节点值    vec.push_back(pnode);    int cur = 0;    int last = 1;    while(cur < vec.size())    {        last = vec.size();        while(cur != last)        {            cout << vec[cur]->date << " ";
<pre name="code" class="cpp"><span style="white-space:pre"></span>    if(vec[cur]->lchild)                vec.push_back(vec[cur]->lchild);
if(vec[cur]->rchild) vec.push_back(vec[cur]->rchild); cur++; } cout << endl; }}int main(){ int a[] = {1,2,4,0,0,5,7,0,0,8,0,0,3,0,6,0,0}; PNode Pnode = NULL; Pnode = buildtree(a, 17); priorder(Pnode); cout << endl; printbylevel(Pnode); return 0;}



0 0
原创粉丝点击