二叉树

Posted 2020-12-02 weekbo

#include<windows.h>
#include<stdio.h>

 

#define SUCCESS 1 // 执行成功

template<class T>
class TreeNode
{
public:
T element; //当前节点存储的数据
TreeNode<T>* pLeft; //指向左子节点的指针
TreeNode<T>* pRight; //指向右子节点的指针
TreeNode<T>* pParent; //指向父结点的指针

TreeNode(T& ele)
{
//初始化Node节点
memset(&element, 0, sizeof(TreeNode));
//为元素赋值
memcpy(&element, &ele, sizeof(T));
pLeft = pRight = pParent = NULL;
}
//重载== 比较两结点是否相等
bool operator==(TreeNode<T>* node)
{
return node->element == element ? true : false;
}
};

template<class T>
class BSortTree
{
public:
BSortTree(); //构造函数
~BSortTree(); //析构函数
public:
bool IsEmpty(); //判断树是否为空
DWORD Insert(T element); //新增节点
void Delete(T element); //删除节点
TreeNode<T>* Search(T element); //查找节点
TreeNode<T>* gen(); //获取根节点
void InOrderTraverse(TreeNode<T>* pNode); //中序遍历
void PreOrderTraverse(TreeNode<T>* pNode); //前序遍历
void PostOrderTraverse(TreeNode<T>* pNode); //后序遍历
private:
TreeNode<T>* GetMaxNode(TreeNode<T>* pNode); //获取以pNode为根的最大节点
TreeNode<T>* GetMinNode(TreeNode<T>* pNode); //获取以pNode为根的最小节点
TreeNode<T>* SearchNode(TreeNode<T>* pNode, T element); //获取以pNode为根的最小节点
DWORD InsertNode(T element, TreeNode<T>* pNode); //新增节点
TreeNode<T>* DeleteNode(T element, TreeNode<T>* pNode); //删除节点
void Clear(TreeNode<T>* pNode); //清空所有节点
private:
TreeNode<T>* m_pRoot; //根结点指针
int size; //树中元素总个数
};

template<class T>
BSortTree<T>::BSortTree()
{
m_pRoot = NULL;
size = 0;
}

template<class T>
BSortTree<T>::~BSortTree()
{
Clear(m_pRoot);
}

template<class T>
DWORD BSortTree<T>::Insert(T element)
{
//如果根节点为空
if (!m_pRoot)
{
m_pRoot = new TreeNode<T>(element);
size++;
return SUCCESS;
}
//如果根节点不为空
return InsertNode(element, m_pRoot);
}

template<class T>
TreeNode<T>* BSortTree<T>::GetMinNode(TreeNode<T>* pNode)
{
TreeNode<T>* zhi = NULL;
if (pNode != NULL)
{
int date = pNode->element;
if (pNode->element <= date)
{
zhi = (TreeNode<T>*) pNode->element;
}
GetMinNode(pNode->pLeft);
return zhi;
}

}

template<class T>
DWORD BSortTree<T>::InsertNode(T element, TreeNode<T>* pNode)
{
//将元素封装到节点中
TreeNode<T>* pNewNode = new TreeNode<T>(element);
//如果element == 当前节点 直接返回
if (element == pNode->element)
{
return SUCCESS;
}
//如果pNode的左子节点为NULL 并且element < 当前节点
if (pNode->pLeft == NULL && element < pNode->element)
{
pNode->pLeft = pNewNode;
pNewNode->pParent = pNode;
size++;
return SUCCESS;
}
//如果pNode的右子节点为NULL 并且element > 当前节点
if (pNode->pRight == NULL && element > pNode->element)
{
pNode->pRight = pNewNode;
pNewNode->pParent = pNode;
size++;
return SUCCESS;
}
//如果element<当前节点 且当前节点的左子树不为空
if (element < pNode->element)
{
InsertNode(element, pNode->pLeft);
}
else
{
InsertNode(element, pNode->pRight);
}
return SUCCESS;
}

template<class T>
void BSortTree<T>::Clear(TreeNode<T>* pNode)
{
if (pNode != NULL)
{
Clear(pNode->pLeft);
Clear(pNode->pRight);
delete pNode;
pNode = NULL;
}
}

template<class T>
bool BSortTree<T>::IsEmpty()
{
return size == 0 ? true : false;
}

template<class T>
TreeNode<T>* BSortTree<T>::Search(T element)
{
return SearchNode(m_pRoot, element);
}

template<class T>
TreeNode<T>* BSortTree<T>::gen()
{
return m_pRoot;
}

template<class T>
TreeNode<T>* BSortTree<T>::SearchNode(TreeNode<T>* pNode, T element)
{
if (pNode == NULL) //如果节点为NULL
{
return NULL;
}
else if (element == pNode->element) //如果相等
{
return pNode;
} //如果比节点的元素小 向左找
else if (element < pNode->element)
{
return SearchNode(pNode->pLeft, element);
}
else //如果比节点的元素大 向右找
{
return SearchNode(pNode->pRight, element);
}
}

template<class T>
void BSortTree<T>::Delete(T element)
{
DeleteNode(element, m_pRoot);
}

template<class T>
TreeNode<T>* BSortTree<T>::DeleteNode(T element, TreeNode<T>* pNode)
{
if (pNode == NULL) //如果节点为NULL
{
return NULL;
}
else if (element == pNode->element) //如果相等 找到了数据
{
if (pNode->pRight != NULL && pNode->pLeft == NULL)//右子树
{
TreeNode<T>* py = pNode->pParent;
py->pLeft = pNode->pRight;
delete pNode;
}
else if (pNode->pLeft != NULL && pNode->pRight == NULL)//左子树
{
TreeNode<T>* pz = pNode->pParent;
pz->pLeft = pNode->pLeft;
delete pNode;
}
else if (pNode->pRight == NULL && pNode->pLeft == NULL)
{
TreeNode<T>* pN;
pN = pNode->pParent;
if (pN->pLeft != NULL && pN->pRight == NULL)
{
delete pN->pLeft;
pN->pLeft = NULL;
}
if (pN->pRight != NULL && pN->pLeft == NULL)
{
delete pN->pRight;
pN->pRight = NULL;
}

delete pN->pRight;
pN->pRight = NULL;
}
else if (pNode->pRight != NULL && pNode->pLeft != NULL)
{
TreeNode<T>* dataMin = GetMinNode(pNode->pRight);
TreeNode<T>* deg = pNode->pRight;
pNode->element = (int)dataMin;
DeleteNode(pNode->element, deg);

}
} //如果比节点的元素小 向左找
else if (element < pNode->element)
{
DeleteNode(element, pNode->pLeft);
}
else //如果比节点的元素大 向右找
{
DeleteNode(element, pNode->pRight);
}
return NULL;
}

template<class T>
void BSortTree<T>::InOrderTraverse(TreeNode<T>* pNode)
{
if (pNode != NULL)
{
InOrderTraverse(pNode->pLeft);
printf("%d ", pNode->element);
InOrderTraverse(pNode->pRight);
}

}

void TestInsert()
{
//12 8 5 9 17 15 13
/*
12

8 17

5 9 15

13

*/

BSortTree<int> tree;

tree.Insert(12);
tree.Insert(8);
tree.Insert(5);
tree.Insert(9);
tree.Insert(17);
tree.Insert(15);
tree.Insert(13);
}

void TestSerch()
{
//12 8 5 9 17 15 13
/*
12

8 17

5 9 15

13
*/

BSortTree<int> tree;

tree.Insert(12);
tree.Insert(8);
tree.Insert(5);
tree.Insert(9);
tree.Insert(17);
tree.Insert(15);
tree.Insert(13);

TreeNode<int>* p = tree.Search(13);
printf("%p %d ", p, p->element);

tree.Delete(13);
//中序遍历所有节点
tree.InOrderTraverse(tree.gen());

}

int main()
{
TestSerch();
getchar();
return 0;
}