美文网首页
二叉树的初始化、前序、中序、后序遍历

二叉树的初始化、前序、中序、后序遍历

作者: HITZGD | 来源:发表于2018-11-26 21:25 被阅读0次

问题:

给定二叉树的初始化数据,怎样动态建立一个二叉树呢?

比如我们给定这样的一组数据:{ 1, 2, 3, 4, 0, 5, 6, 0, 7 }(假设0代表空),则我们构建的二叉树是这样的:

 1
/ \

2 3
/ /
4 5 6

7
思路分析:

我们可以使用一个队列,队首出一个元素,队未进两个元素,而这两个元素正好是这个队首元素的左右节点。

头文件 (TreeNode.h)

#pragma once
#include <iostream>
#include <vector>
#include <queue>
using namespace std;

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};


//初始化一个二叉树
TreeNode* initBTree(int elements[], int size);

//树的前序遍历
void preOrder(TreeNode *root, vector<int> &result);

//树的中序遍历
void inOrder(TreeNode *root, vector<int> &result);

//树的后序遍历
void postOrder(TreeNode *root, vector<int> &result);

/*树的深度*/
int btreedepth(TreeNode *bt);

/*树的节点数*/
int btreecount(TreeNode *bt)

//vector的遍历
void traverse(vector<int> nums);

函数(TreeNode.cpp)

#include "TreeNode.h"

/*初始化一个二叉树*/
TreeNode* initBTree(int elements[], int size)
{
    if (size < 1)
    {
        return NULL;
    }
    //动态申请size大小的指针数组
    TreeNode **nodes = new TreeNode*[size];
    //将int数据转换为TreeNode节点
    for (int i = 0; i < size; i++)
    {
        if (elements[i] == 0)
        {
            nodes[i] = NULL;
        }
        else
        {
            nodes[i] = new TreeNode(elements[i]);
        }
    }
    queue<TreeNode*> nodeQueue;
    nodeQueue.push(nodes[0]);

    TreeNode *node;
    int index = 1;
    while (index < size)
    {
        node = nodeQueue.front();
        nodeQueue.pop();
        nodeQueue.push(nodes[index++]);
        node->left = nodeQueue.back();
        nodeQueue.push(nodes[index++]);
        node->right = nodeQueue.back();
    }
    return nodes[0];
}

/*树的前序遍历*/
void preOrder(TreeNode *root, vector<int> &result)
{
    if (root)
    {
        result.push_back(root->val);
        preOrder(root->left, result);
        preOrder(root->right, result);
    }
}

/*树的中序遍历*/
void inOrder(TreeNode *root, vector<int> &result)
{
    if (root)
    {
        inOrder(root->left, result);
        result.push_back(root->val);
        inOrder(root->right, result);
    }
}

/*树的后序遍历*/
void postOrder(TreeNode *root, vector<int> &result)
{
    if (root)
    {
        postOrder(root->left, result);
        postOrder(root->right, result);
        result.push_back(root->val);
    }
}

/*树的深度*/
int btreedepth(TreeNode *bt)
{
    if (bt == NULL)
        return 0;
    else
    {
        int dep1 = btreedepth(bt->left);
        int dep2 = btreedepth(bt->right);
        if (dep1>dep2)
            return dep1 + 1;
        else
            return dep2 + 1;
    }
}

/*树的节点数*/
int btreecount(TreeNode *bt)
{
    if (bt == NULL)
        return 0;
    else
        return btreecount(bt->left) + btreecount(bt->right) + 1;

}

/*树的叶子树*/
int btreeleafcount(TreeNode *bt)
{
    if (bt == NULL)
        return 0;
    else if (bt->left == NULL&&bt->right == NULL)
        return 1;
    else return btreeleafcount(bt->left) + btreeleafcount(bt->right);
}

/*输出树*/
void printbtree(TreeNode *bt)
{
    if (bt == NULL)
        return;
    else
    {
        cout << bt->val;
        if (bt->left != NULL || bt->right != NULL)
        {
            cout << '(';
            printbtree(bt->left);
            if (bt->right != NULL)
                cout << ',';
            printbtree(bt->right);
            cout << ')';
        }
    }
}


/*Vector遍历*/
void traverse(vector<int> nums)
{
    vector<int>::size_type size = nums.size();
    for (size_t i = 0; i < size; i++)
    {
        cout << nums[i] << ' ';
    }
    cout << endl;
}

相关文章

网友评论

      本文标题:二叉树的初始化、前序、中序、后序遍历

      本文链接:https://www.haomeiwen.com/subject/epmiqqtx.html