数据结构中二叉树实现及部分操作

本文详细介绍了树和二叉树的基本概念,包括节点、根节点、度等,并深入探讨了二叉树的特性及其实现方法。文章还提供了二叉树的C++代码示例,包括递归和非递归的遍历方法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

谈二叉树之前,我们先来看看树的定义

树:由N(N>=0)个结点构成的集合。
对N>1的树:
1、有一个特殊的结点,称为根结点,根节点没有前驱结点
2、除根节点外,其余结点被分成M(M>0)个互不相交的集合T1、T2、……、Tm,其中每一个集合Ti(1<= i <= m)又是一棵结构与树类似的子树。每棵子树的根结点有且只有一个前驱,可以有0个或多个后继。
因此,树是递归定义的。
如下图所示:
这里写图片描述

再看一些和树有关的概念:

  • 节点:节点包含数据和指向其它节点的指针。
  • 根节点:树第一个结点称为根节点。
  • 节点的度:节点所拥有的子树的个数。
  • 树的度:树中所有节点的度的最大值称为该树的度。
  • 叶子节点:没有子节点的节点(度为0),也称为终端节点。
  • 父子节点:一个节点father指向另一个节点child,则child为孩子节点,father为父亲节点 。
  • 兄弟节点:具有相同父节点的节点互为兄弟节点。
  • 祖先节点:从根节点开始到该节点所经的所有节点都可以称为该节点的祖先。
  • 子孙节点:以某节点为根的子树中任一节点都称为该节点的子孙。
  • 树的高度:树中距离根节点最远节点的路径长度。

树中比较特殊的就是二叉树了,来看看二叉树的定义
二叉树:
一棵二叉树是结点的一个有限集合,该集合或者为空,或者是由一个根节点加上两棵分别称为左子树和右子树的二叉树组成。

二叉树的特点:
  • 每个结点最多有两棵子树,即二叉树不存在度大于2的结点(分支数最大不超过2)
  • 二叉树的子树有左右之分,其子树的次序不能颠倒

看张二叉树的示意图:
这里写图片描述
二叉树就是由上面这五种情况嵌套或组合而成

在介绍两种比较特殊的二叉树

1.完全二叉树

如果一棵具有N个结点的二叉树的结构与满二叉树的前N个结点的结构相同,称为完全二叉树。

这里写图片描述

2.满二叉树

在一棵二叉树中,如果所有分支结点都存在左子树和右子树,并且所有叶子节点都在同一层上。

这里写图片描述

从概念和图我们可以看出:满二叉树一定是完全二叉树,完全二叉树不一定是满二叉树。

接下来就是二叉树的具体实现了,直接上代码:

BinaryTree.h

#define _CRT_SECURE_NO_WARNINGS 1
#pragma once

#include <iostream>
using namespace std;
#include <stdlib.h>
#include <queue>
#include <stack>

//二叉树节点
template <class T>
struct BinaryTreeNode
{
    BinaryTreeNode(const T& data)
    :_pLeft(NULL)
    , _pRight(NULL)
    , _data(data)
    {}

    BinaryTreeNode<T>* _pLeft;//左子树
    BinaryTreeNode<T>* _pRight;//右子树
    T _data;//数据
};

//二叉树
template<class T>
class BinaryTree
{
    typedef BinaryTreeNode<T>* pNode;
    typedef BinaryTreeNode<T> Node;
public:
    //无参构造函数
    BinaryTree()
        : _pRoot(NULL)
    {}
    //带参数的构造函数
    BinaryTree(T* arr, size_t size, const T& invalid)
    {
        size_t index = 0;
        _CreateBinaryTree(_pRoot, arr, size, index, invalid);
    }
    //拷贝构造函数
    BinaryTree(const BinaryTree<T>& bt)
    {
        _pRoot = _CopyBinaryTree(bt._pRoot);
    }
    //赋值运算符重载(两种方式均可)
    //BinaryTree& operator=(const BinaryTree<T>& bt)
    //{
    //  if (this != &bt)
    //  {
    //      _DestoryBinaryTree(_pRoot);
    //      _pRoot = _CopyBinaryTree(bt._pRoot);
    //  }
    //  return *this;
    //}
    BinaryTree& operator=(const BinaryTree<T>& bt)
    {
        swap(_pRoot, bt._pRoot);
        return *this;
    }
    //析构函数
    ~BinaryTree()
    {
        _DestoryBinaryTree(_pRoot);
    }
    //递归前序遍历
    void PreOrder()
    {
        _PreOrder(_pRoot);
        cout << endl;
    }
    //递归中序遍历
    void InOrder()
    {
        _InOrder(_pRoot);
        cout << endl;
    }
    //递归后序遍历
    void PostOrder()
    {
        _PostOrder(_pRoot);
        cout << endl;
    }
    //层序遍历(用队列)
    void LevelOrder()
    {
        queue<pNode> q;
        if (_pRoot)
            q.push(_pRoot);
        while (!q.empty())
        {
            pNode cur = q.front();
            q.pop();
            cout << cur->_data << " ";
            if (cur->_pLeft)
            {
                q.push(cur->_pLeft);
            }
            if (cur->_pRight)
            {
                q.push(cur->_pRight);
            }
        }
        cout << endl;
    }
    //非递归前序遍历
    void PreOrderNoR()
    {
        _PreOrderNoR2(_pRoot);
    }
    //非递归中序遍历
    void InOrderNoR()
    {
        _InOrderNoR(_pRoot);
    }
    //非递归后序遍历
    void PostOrderNoR()
    {
        _PostOrderNoR(_pRoot);
    }
    //二叉树节点个数
    size_t Size()
    {
        return _Size(_pRoot);
    }
    //二叉树叶子节点个数
    size_t LeafSize()
    {
        return _LeafSize(_pRoot);
    }
    //二叉树第K层节点个数
    size_t KSize(size_t k)
    {
        return _KSize(_pRoot, k);
    }
    //二叉树高度
    size_t Height()
    {
        return _Height(_pRoot);
    }
    //寻找二叉树中的节点
    pNode Find(const T& data)
    {
        return _Find(_pRoot, data);
    }
    //寻找某节点的双亲节点
    pNode FindParent(pNode node)
    {
        return _FindParent(_pRoot, node);
    }
    //寻找某节点的左孩子节点
    pNode FindLeftChild(pNode node)
    {
        return _FindLeftChild(_pRoot, node);
    }
    //寻找某节点的右孩子节点
    pNode FindRightChild(pNode node)
    {
        return _FindRightChild(_pRoot, node);
    }

//封装函数
private:
    //创建二叉树(前序遍历)
    void _CreateBinaryTree(pNode& pRoot, T* arr, size_t size, size_t& index, const T& invalid)
    {
        if (index < size&&invalid != arr[index])
        {
            pNode pNewNode = new Node(arr[index]);
            _CreateBinaryTree(pNewNode->_pLeft, arr, size, ++index, invalid);
            _CreateBinaryTree(pNewNode->_pRight, arr, size, ++index, invalid);
            pRoot = pNewNode;
        }
    }
    //拷贝二叉树
    pNode _CopyBinaryTree(pNode pRoot)
    {
        if (pRoot == NULL);
        {
            return NULL;
        }
        pNode tmp = new Node(pRoot->_data);
        tmp->_pLeft = _CopyBinaryTree(pRoot->_pLeft);
        tmp->_pRight = _CopyBinaryTree(pRoot->_pRight);
        return tmp;
    }
    //递归前序遍历
    void _PreOrder(pNode pRoot)
    {
        if (pRoot)
        {
            cout << pRoot->_data << " ";
            _PreOrder(pRoot->_pLeft);
            _PreOrder(pRoot->_pRight);
        }
    }
    //递归中序遍历
    void _InOrder(pNode pRoot)
    {
        if (pRoot)
        {
            _InOrder(pRoot->_pLeft);
            cout << pRoot->_data << " ";
            _InOrder(pRoot->_pRight);
        }
    }
    //递归后序遍历
    void _PostOrder(pNode pRoot)
    {
        if (pRoot)
        {
            _PostOrder(pRoot->_pLeft);
            _PostOrder(pRoot->_pRight);
            cout << pRoot->_data << " ";
        }
    }
    //非递归前序遍历
    //方法一:访问根之后直接访问左子树访问完
    void _PreOrderNoR(pNode pRoot)
    {
        stack<pNode> s;
        pNode cur = pRoot;
        while (cur || !s.empty())
        {   
            while (cur)
            {
                cout << cur->_data << " ";
                s.push(cur);
                cur = cur->_pLeft;
            }//此时左子树遍历完
            //取栈顶元素访问右子树(此时栈顶元素已经访问)
            pNode top = s.top();
            s.pop();
            //子问题
            cur = top->_pRight;
        }
        cout << endl;
    }
    //方法二:
    void _PreOrderNoR2(pNode pRoot)
    {
        if (pRoot == NULL)
            return;
        stack<pNode> s;
        s.push(pRoot);//根节点入栈
        while (!s.empty())
        {
            pNode top = s.top();
            cout << top->_data << " ";
            s.pop();
            if (top->_pRight)//先压右子树
                s.push(top->_pRight);
            if (top->_pLeft)//再压左子树
                s.push(top->_pLeft);
        }
        cout << endl;
    }
    //非递归中序遍历
    void _InOrderNoR(pNode pRoot)
    {
        stack<pNode> s;
        pNode cur = pRoot;
        while (cur || !s.empty())
        {
            while (cur)
            {
                s.push(cur);
                cur = cur->_pLeft;
            }//直到最左节点
            //取栈顶元素访问该元素并访问右子树
            pNode top = s.top();
            cout << top->_data << " ";
            s.pop();
            //子问题
            cur = top->_pRight;
        }
        cout << endl;
    }
    //非递归后序遍历
    void _PostOrderNoR(pNode pRoot)
    {
        stack<pNode> s;
        pNode cur = pRoot;
        pNode prev = NULL;//标记最近访问过的节点
        while (cur || !s.empty())
        {
            while (cur)
            {
                s.push(cur);
                cur = cur->_pLeft;
            }//直到最左节点

            pNode top = s.top();//取栈顶元素
            if (top->_pRight == NULL || top->_pRight == prev)//判断该节点的右子树是否为空
            {                                                //再判断是不是第一次在栈顶取到该元素 
                cout << top->_data << " ";
                prev = top;
                s.pop();
            }
            else//不为空,访问右子树
            {
                cur = top->_pRight;
            }
        }
        cout << endl;
    }
    //后序遍历递归销毁二叉树
    void _DestoryBinaryTree(pNode pRoot)
    {
        if (pRoot)
            return;
        _DestoryBinaryTree(pRoot->_pLeft);
        _DestoryBinaryTree(pRoot->_pRight);
        delete pRoot;
        pRoot = NULL;
    }
    //求节点个数
    size_t _Size(pNode pRoot)
    {
        if (!pRoot)
        {
            return 0;
        }
        return _Size(pRoot->_pLeft) + _Size(pRoot->_pRight) + 1;
    }
    //求叶子节点个数
    size_t _LeafSize(pNode pRoot)
    {
        if (!pRoot)
            return 0;
        if (pRoot->_pLeft == NULL && pRoot->_pRight == NULL)
            return 1;
        return _LeafSize(pRoot->_pLeft) + _LeafSize(pRoot->_pRight);
    }
    //求第K层节点的个数
    size_t _KSize(pNode pRoot, size_t k)
    {
        if (pRoot == NULL)
            return 0;
        if (k == 1)
            return 1;
        return _KSize(pRoot->_pLeft, k - 1) + _KSize(pRoot->_pRight, k - 1);
    }
    //求二叉树高度
    size_t _Height(pNode pRoot)
    {
        if (!pRoot)
            return 0;
        if (pRoot->_pLeft == NULL&&pRoot->_pRight == NULL)
            return 1;
        size_t LeftHeight = _Height(pRoot->_pLeft);
        size_t RightHeight = _Height(pRoot->_pRight);
        return LeftHeight > RightHeight ? LeftHeight + 1 : RightHeight + 1;
    }
    //找二叉树中的节点
    pNode _Find(pNode pRoot, const T& data)
    {
        if (pRoot == NULL)
            return NULL;
        if (pRoot->_data == data)
            return pRoot;
        pNode ret = NULL;
        ret = _Find(pRoot->_pLeft, data);
        if (ret)
            return ret;
        return _Find(pRoot->_pRight, data);
    }
    //找二叉树中节点的双亲节点
    pNode _FindParent(pNode pRoot, pNode node)
    {
        //树为空树   寻找的节点为空节点   寻找的节点是根节点(无双亲节点)
        if (pRoot == NULL || node == NULL || pRoot == node)
        {
            return NULL;
        }
        if (node == pRoot->_pLeft || node == pRoot->_pRight)
            return pRoot;
        if (pRoot->_pLeft)
            return _FindParent(pRoot->_pLeft, node);
        return _FindParent(pRoot->_pRight, node);
    }
    //寻找节点的左孩子
    pNode _FindLeftChild(pNode pRoot, pNode node)
    {
        if (node == NULL || pRoot == NULL)
        {
            return NULL;
        }
        pNode cur = _Find(pRoot, node->_data);
        return cur->_pLeft;
    }
    //寻找节点的右孩子
    pNode _FindRightChild(pNode pRoot, pNode node)
    {
        if (node == NULL || pRoot == NULL)
        {
            return NULL;
        }
        pNode cur = _Find(pRoot, node->_data);
        return cur->_pRight;
    }

private:
    pNode _pRoot;
};

test.cpp

#define _CRT_SECURE_NO_WARNINGS 1

#include "BinaryTree.h"


void FunTest1()
{
    char arr1[] = { 'A', 'B', 'D', '#', 'G', '#', '#', '#', 'C', 'E', '#', '#', 'F' };
    BinaryTree<char> bt1(arr1, sizeof(arr1) / sizeof(arr1[0]), '#');
    bt1.PreOrder();
    bt1.InOrder();
    bt1.PostOrder();
    bt1.LevelOrder();
    cout << bt1.Size() << endl;
    cout << bt1.LeafSize() << endl;
    cout << bt1.KSize(3) << endl;
    cout << bt1.Height() << endl;
    cout << bt1.KSize(4) << endl;

    BinaryTreeNode<char>* ret1 = bt1.Find('B');
    BinaryTreeNode<char>* ret2 = bt1.FindParent(ret1);
    BinaryTreeNode<char>* ret3 = bt1.FindLeftChild(ret1);
    BinaryTreeNode<char>* ret4 = bt1.FindRightChild(ret1);
    bt1.PreOrderNoR();
    bt1.InOrderNoR();
    bt1.PostOrderNoR();
}

int main()
{
    FunTest1();
    system("pause");
    return 0;
}

结果在这里不做演示,读者可以自行测试。
(注:对二叉树中常见的面试题做出整理后,会附上链接)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值