分层遍历二叉树-编程之美

最新推荐文章于 2015-11-16 19:28:22 发布

原创最新推荐文章于 2015-11-16 19:28:22 发布 · 778 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

130 篇文章

订阅专栏

翻看编程之美的3.10：分层遍历二叉树，觉得这个基本问题会经常遇到：

问题1：给定一棵二叉树，要求按分层遍历该二叉树，即从上到下的层次访问该二叉树（每一层将单独输出一行），每一层要求访问的顺序为从左到右，并将节点依次编号。
问题2：写一个函数，打印二叉树中某层次的节点（从左到右），其中根节点为第0层，函数原型为int PrintNodeAtLevel(Node* root, int level)，成功返回1，失败返回0.

这里就问题1和其扩展问题给了简单。

实现了一个最基本的方法：

#include<iostream>
#include<vector>
using namespace std;
typedef struct Node{
        int data;
        struct Node* lChild;
        struct Node* rChild;
}Node;

int  PrintNodeByLevel(Node * root)
{
        if(root==NULL)
                return 0;
        vector<Node*> vec;
        vec.push_back(root);
        int cur=0;
        int last=1;
        while(cur<vec.size()){
                last=vec.size();
                while(cur<last){
                        cout<<vec[cur]->data<<" ";
                        if(vec[cur]->lChild!=NULL){
                                vec.push_back(vec[cur]->lChild);
                        }
                        if(vec[cur]->rChild!=NULL){
                                vec.push_back(vec[cur]->rChild);
                        }
                        cur++;
                }
                cout<<endl;
        }
        return 0;
}
//扩展问题1
int PrintNodeByLevelRes(Node * root)
{
        if(root==NULL)
                return 0;
        vector<Node*> vec;
        vec.push_back(root);
        int cur=0;
        int last=1;
        while(cur<vec.size()){
                vec.push_back(NULL);
                last=vec.size();
                while(cur<last){
                        if(vec[cur]!=NULL){
                                if(vec[cur]->rChild!=NULL)
                                        vec.push_back(vec[cur]->rChild);
                                if(vec[cur]->lChild!=NULL)
                                        vec.push_back(vec[cur]->lChild);
                       }
                        cur++;
                }
        }
        for(int i=vec.size()-1;i>=0;i--){
                if(vec[i]==NULL)
                        cout<<endl;
                else
                        cout<<vec[i]->data<<" ";
        }
        return 0;
}
//扩展问题2
int PrintNodeByLevelResRL(Node * root)
{
        if(root==NULL)
                return 0;
        vector<Node*> vec;
        vec.push_back(root);
        int cur=0;
        int last=1;
        while(cur<vec.size()){
                vec.push_back(NULL);
                last=vec.size();
                while(cur<last){
                        if(vec[cur]!=NULL){
                                if(vec[cur]->lChild!=NULL)
                                        vec.push_back(vec[cur]->lChild);
                                if(vec[cur]->rChild!=NULL)
                                        vec.push_back(vec[cur]->rChild);
                        }
                        cur++;
                }
        }
        for(int i=vec.size()-1;i>=0;i--){
                if(vec[i]==NULL)
                        cout<<endl;
                else
                        cout<<vec[i]->data<<" ";
        }
        return 0;
}

Node *  initTree()
{
        Node * array[8];
        for(int i=0;i<8;i++){
                array[i]=new Node;
                array[i]->data=1+i;
                array[i]->rChild=NULL;
                array[i]->lChild=NULL;
        }
        array[0]->lChild=array[1];
        array[0]->rChild=array[2];
        array[1]->lChild=array[3];
        array[1]->rChild=array[4];
        array[2]->lChild=array[5];
        array[4]->lChild=array[6];
        array[4]->rChild=array[7];

        return array[0];
}
int main()
{
        Node *root;
        root=initTree();
        PrintNodeByLevel(root);
        PrintNodeByLevelRes(root);
        PrintNodeByLevelResRL(root);

        return 0;
}

参考文献：

http://blog.youkuaiyun.com/sunmenggmail/article/details/8044729