查找树（二叉树）的构建以及分层遍历

代码均通过vs2008编译运行

头文件代码：

#include <iostream>
#include <stack>//包含stack和queue，为了方便遍历数据存储
#include <queue>
struct BTree//定义查找树的结构
{
    int m_data;
    struct BTree* m_pLeft;
    struct BTree* m_pRight;
};

struct BTree* insert(struct BTree* root,int& data)//查找树的生成，采用递归操作
{
    if (root==NULL)
    {
        root = (struct BTree*)malloc( sizeof(struct BTree) );
        root->m_data=data;
        root->m_pLeft=NULL;
        root->m_pRight=NULL;
    }
    else
    {
        if (root->m_data>data)
        {
            root->m_pRight=insert(root->m_pRight,data);
        }
        else
        {
            root->m_pLeft=insert(root->m_pLeft,data);
        }
    }
    return root;
}

void show_tree(struct BTree* root)//输出树的内容，顺序输出
{
    /* recursion */
    if (root->m_pLeft!=NULL||root->m_pRight!=NULL)
    {
        if (root->m_pLeft!=NULL)
        {
            show_tree(root->m_pLeft);
        }
        if (root->m_pRight!=NULL)
        {
            show_tree(root->m_pRight);
        }
    }

    /* backtracking */
    printf("%d\t",root->m_data);
    return;
}

std::stack<int> range_tree(struct BTree* root,std::stack<int>& int_stack)//在栈中存储树的数据，满足LIFO输出
{
    if (root->m_pLeft!=NULL||root->m_pRight!=NULL)
    {
        if (root->m_pLeft!=NULL)
        {
            int_stack=range_tree(root->m_pLeft,int_stack);
        }
        if (root->m_pRight!=NULL)
        {
            int_stack=range_tree(root->m_pRight,int_stack);
        }
    }

    /* backtracking */
    int_stack.push(root->m_data);
    return int_stack;

}

std::queue<int> queue_tree(struct BTree* root,std::queue<int>& int_queue)//在队列中存储树的数据，满足FIFO
{
    if (root->m_pLeft!=NULL||root->m_pRight!=NULL)
    {
        if (root->m_pLeft!=NULL)
        {
            int_queue=queue_tree(root->m_pLeft,int_queue);
        }
        if (root->m_pRight!=NULL)
        {
            int_queue=queue_tree(root->m_pRight,int_queue);
        }
    }

    /* backtracking */
    int_queue.push(root->m_data);
    return int_queue;

}

 

 

主文件：

#include "tree.h"

int main(void)
{
    std::stack<int> int_value;
    std::queue<int> int_queue;
    struct BTree* number={0};//初始化
    int i=0;
    while (std::cin>>i)
    {
        number=insert(number,i);
    }
    show_tree(number);
    std::cout<<std::endl;

    int_value=range_tree(number,int_value);
    std::stack<int>::size_type stack_cap = int_value.size();
    for (std::stack<int>::size_type i=0;i!=stack_cap;++i)
    {
        std::cout<<int_value.top()<<"\t";
        if (int_value.size()!=0)
        {
            int_value.pop();//注意要出栈
        }
        else break;
    }
    std::cout<<std::endl;

    int_queue=queue_tree(number,int_queue);
    std::queue<int>::size_type queue_cap = int_queue.size();
    for (std::queue<int>::size_type iter=0;iter!=queue_cap;++iter)
    {
        std::cout<<int_queue.front()<<"\t";
        int_queue.pop();//注意要出队
       
    }
    std::cout<<std::endl;
    system("pause");
    return NULL;
}