C++ 使用数组建立二叉树

本文内容来自C++ 使用数组建立二叉树 层序数组
仅供记录参考

1、输入数组要求

数组是按照层序输入的，当该结点为空时，用‘#’代替空的位置。

如：

图(a)中的二叉树的输入数组为：

int data[] = { 1, 2, 3, 4, 5, ‘#’, 6, ‘#’, ‘#’, 7, 8 };

图(b)中的二叉树的 输入数组为：

int data[] = { 1, 2, 3, 4, 5, ‘#’, 6, 7 };

2、数组创建二叉树的子函数

原创：东篱_

/***************************************************************************
*   @author:    东篱_
*   @date:      2018.4.8
*   @remark:    this code is for binary tree with array
*   @note:      The numbers in the input array is arranged by level order
****************************************************************************/

添加了个人注释以供理解

#include <iostream>
#include <queue>
using namespace std;
//二叉树节点定义
typedef struct BiTNode//二叉树的结点数据结构
{
	int data;
	BiTNode *lchild, *rchild;
};

//按层序输入，结点为空时，输入'#'
BiTNode *CreateBiTree(int *a, int n, int start)//*a为data，n为数组长度，start为根节点
{
	if (a[start] == '#')return NULL;//当根节点为空，即空树

	BiTNode* root = new BiTNode;//新建一个根结点
	root->data = a[start];//给根结点root的成员变量data,lchild,rchild赋初值
	root->lchild = NULL;
	root->rchild = NULL;

	int lnode = 2 * start + 1;//用根节点确定左节点的位置
	int rnode = 2 * start + 2;//用根节点确定右节点的位置

	if (lnode > n - 1) root->lchild = NULL;
	else root->lchild = CreateBiTree(a, n, lnode);//lnode替换start，为左子树的根节点

	if (rnode > n - 1) root->rchild = NULL;
	else root->rchild = CreateBiTree(a, n, rnode);//rnode替换start，为右子树的根节点

	return root;
}

//先序遍历函数（递归完成）
void PreOrderTraverse(BiTNode *T)
{
	if (T){
		cout << T->data << " ";
		PreOrderTraverse(T->lchild);
		PreOrderTraverse(T->rchild);
	}
}

//层序遍历--队列
void LevelOrderTraverse(BiTNode *T)
{
	queue<BiTNode *> Q;//定义一个queue二叉树对象Q
	if (T == NULL) return;
	Q.push(T);//入队根指针  push：入队
	while (!Q.empty()){
		BiTNode *cur = Q.front();//当前指针指向队列的第一个元素
		Q.pop();// pop：出队
		cout << cur->data << " ";
		if (cur->lchild) Q.push(cur->lchild);
		if (cur->rchild) Q.push(cur->rchild);
	}
	cout << endl;
}

/*
使用sizeof(arr) / sizeof(arr[0])求数组长度，但是要注意：
    sizeof()函数是求数组所占的内存空间大小（不是长度）。
    当在函数中使用该方法求数组长度时，由于数组作为函数参数传入函数的过程中会“退化”为指针，
    因此将会导致sizeof(arr)求出的是一个指针的内存空间大小，
    而非数组的内存空间大小–>从而导致计算的长度错误
    
解决办法
    在声明数组之后就计算出数组长度，然后将数组长度作为函数参数也一起传入
    如：
        int arr[] = {10,20,30,3,12,14};
        int len = sizeof(arr) / sizeof(arr[0]);
        int *find = findPrime(arr, len);
*/

此处内容参考链接：c中自定义函数通过sizeof来输出数组的长度为何不正确？【原创】

void test1(){
	BiTNode *t;
	int data[] = { 1, 2, 3, 4, 5, '#', 6, '#', '#', 7, 8 };
	//前序遍历：1 2 4 5 7 8 3 6
	t = CreateBiTree(data, sizeof(data) / sizeof(data[0]), 0);

	printf("The pre order is :   ");
	PreOrderTraverse(t);
	cout << endl;

	printf("The level order is : ");
	LevelOrderTraverse(t);
}

void test2(){
	BiTNode *t;
	int data[] = { 1, 2, 3, 4, 5, '#', 6, 7 };
	//前序遍历：1 2 4 7 5 3 6
	t = CreateBiTree(data, sizeof(data) / sizeof(data[0]), 0);

	printf("The pre order is :   ");
	PreOrderTraverse(t);
	cout << endl;

	printf("The level order is : ");
	LevelOrderTraverse(t);
}
int main()
{
	test1();
	test2();

	return 0;
}

运行结果: