数据结构之二叉树(C语言非递归实现)

1、思路

typedef struct TrNode
{
	int flag;  //标记 0代表此节点没有孩子 1代表此节点有一个左孩子 2 代表此节点有左右孩子
	char data;
	struct TrNode* leftChild; //左孩子
	struct TrNode* rightChild; //右孩子
}node;

数据：ABDH##I##E##CF##G##

递归转为非递归往往需要借助另一个数据结构 栈
数据结构之栈（顺序栈）

在每个节点中设立一个标记变量用来判断其左右孩子的情况，#代表没有孩子(左或右)。
第一步：判断是否为 # 是,则执行第二步，不是,则执行第三步
第二步：判断当前节点的标记 flag 如果为 0 则 flag = 1,如果为 1 则 flag = 2; 如果 flag 为2 则不断出栈直到 flag 不为2（因为 flag 为说明其左右孩子已经处理完毕）
第三步：判断当前节点的标记 flag 如果为 0 则申请新的节点，令当前节点(不是新节点)的 leftchild指向新节点，flag = 1,新节点入栈；如果 flag 为1 则 则申请新的节点，令当前节点(不是新节点)的 rightchild指向新节点，flag = 2,新节点入栈;

2、具体代码

node* createTree(ware*S,char*str)
{
	node* move_Pointer;
	int i = 0;
	if (str[0] == '\0')
	{
		printf("空树\n");
		exit(0);
	}
	node* rootNode = (node*)malloc(sizeof(node)); //根节点
	if (rootNode == NULL)
	{
		printf("根节点创建失败\n");
		exit(0);
	}
	move_Pointer = rootNode;
	//初始化根节点----------------
	rootNode->flag = 0;
	rootNode->data = str[i];
	rootNode->leftChild = NULL;
	rootNode->rightChild = NULL;
	//---------------------------
	Push(S, rootNode); //根节点入栈
	while (str[++i]!='\0')
	{
	    //第一步：判断是否为 # 不是,则执行第三步
		if (str[i] != '#')
		{
			node* childNode = (node*)malloc(sizeof(node));
			if (childNode == NULL)
			{
				printf("树节点创建失败\n");
				exit(0);
			}
			//初始化树节点----------------
			childNode->flag = 0;
			childNode->data = str[i];
			childNode->leftChild = NULL;
			childNode->rightChild = NULL;
			//---------------------------
			//判断当前节点的标记 flag 如果为 0 则申请新的节点，令当前节点(不是新节点)的 leftchild指向新节点，flag = 1,新节点入栈
			if (S->data[S->top]->flag == 0)
			{
				S->data[S->top]->leftChild = childNode;
				//printf("lefit:%p\n", S->data[S->top]->leftChild);
				S->data[S->top]->flag = 1;
				Push(S, childNode);
			}
			else if (S->data[S->top]->flag == 1)
			{   //如果 flag 为1 则 则申请新的节点，令当前节点(不是新节点)的 rightchild指向新节点，flag = 2,新节点入栈
				S->data[S->top]->rightChild = childNode;
				S->data[S->top]->flag = 2;
				Push(S, childNode);
			}
		}
		else if (str[i] == '#') //判断是否为 # 是,则执行第二步
		{
			if (S->top!=-1&&S->data[S->top]->flag == 0)
				S->data[S->top]->flag = 1; //判断当前节点的标记 flag 如果为 0 则 flag = 1
			else if (S->top!=-1&&S->data[S->top]->flag == 1)
			{
				S->data[S->top]->flag = 2; //如果为 1 则 flag = 2
				while (S->top!=-1&&S->data[S->top]->flag == 2)
				{
				    //如果 flag 为2 则不断出栈直到 flag 不为2（因为 flag 为说明其左右孩子已经处理完毕）
					Pop(S);
				}
			}
		}
    }
	printf("!!!二叉树建立完成!!!\n");
	printf("\n");
	return move_Pointer;
}

3、完整代码

#include<stdio.h>
#include<stdlib.h>
#define MAXSIZE 100

typedef struct TrNode
{
	int flag; //标记 0代表此节点没有孩子 1代表此节点有一个左孩子 2 代表此节点有左右孩
	char data;
	struct TrNode* leftChild; //左孩子
	struct TrNode* rightChild; //右孩子
}node;

typedef struct Ware
{
	struct TrNode* data[MAXSIZE];
	int top; //栈顶指针
}ware;


node* Pop(ware* S); //出栈
void initWare(ware* S); //初始化栈
void Push(ware* S, node* trNode); //压栈
node* createTree(ware* S, char* str); //建树
void PerOrderTraversal(ware* S, node* rootNode); //前序遍历


int main()
{
	char* str = "ABDH##I##E##CF##G##";
	//char* str = "ABD##E##CF##G##";
	ware S; //栈
	initWare(&S);  //初始化栈
	node* rootNode = createTree(&S, str);  //建树
	PerOrderTraversal(&S,rootNode); //前序遍历
	free(rootNode);
	return 0;
}

void initWare(ware* S) //初始化栈
{
	for (int i = 0; i < MAXSIZE; i++)
		S->data[i] = NULL;
	S->top = -1;
}


//压栈
void Push(ware* S,node*trNode)
{
	if (S->top == MAXSIZE - 1)
	{
		printf("栈已满\n");
		exit(0);
	}
	S->top++;
	S->data[S->top] = trNode;
	/*printf("入栈成功\n");*/
}


//出栈
node* Pop(ware* S)
{
	if (S->top==-1)
	{
		printf("没有数据可以出栈\n");
		return NULL;
	}
	else
	{
		S->top--;
		return S->data[S->top];
	}
	//printf("%d\n", S->top);
}


//建树
node* createTree(ware*S,char*str)
{
	node* move_Pointer;
	int i = 0;
	if (str[0] == '\0')
	{
		printf("空树\n");
		exit(0);
	}
	node* rootNode = (node*)malloc(sizeof(node)); //根节点
	if (rootNode == NULL)
	{
		printf("根节点创建失败\n");
		exit(0);
	}
	move_Pointer = rootNode;
	//初始化根节点----------------
	rootNode->flag = 0;
	rootNode->data = str[i];
	rootNode->leftChild = NULL;
	rootNode->rightChild = NULL;
	//---------------------------
	Push(S, rootNode); //根节点入栈
	while (str[++i]!='\0')
	{
		if (str[i] != '#')
		{
			node* childNode = (node*)malloc(sizeof(node));
			if (childNode == NULL)
			{
				printf("树节点创建失败\n");
				exit(0);
			}
			//初始化树节点----------------
			childNode->flag = 0;
			childNode->data = str[i];
			childNode->leftChild = NULL;
			childNode->rightChild = NULL;
			//---------------------------
			if (S->data[S->top]->flag == 0)
			{
				S->data[S->top]->leftChild = childNode;
				//printf("lefit:%p\n", S->data[S->top]->leftChild);
				S->data[S->top]->flag = 1;
				Push(S, childNode); //入栈
			}
			else if (S->data[S->top]->flag == 1)
			{
				S->data[S->top]->rightChild = childNode;
				S->data[S->top]->flag = 2;
				Push(S, childNode);  //入栈
			}
		}
		else if (str[i] == '#')
		{
			if (S->top!=-1&&S->data[S->top]->flag == 0)
				S->data[S->top]->flag = 1;
			else if (S->top!=-1&&S->data[S->top]->flag == 1)
			{
				S->data[S->top]->flag = 2;
				while (S->top!=-1&&S->data[S->top]->flag == 2)
				{
					Pop(S);  //出栈
				}
				//这段出栈的代码是二叉树非递归实现的核心代码
			}
		}
    }
	printf("!!!二叉树建立完成!!!\n");
	printf("\n");
	return move_Pointer;
}


//前序遍历
void PerOrderTraversal(ware* S, node* rootNode)
{
	node* M;
	//node* T = rootNode;
	initWare(S);
	if (rootNode == NULL)
		exit(0);
	Push(S, rootNode);
	printf("前序遍历:");
	while (S->top!=-1)
	{
	
		M = S->data[S->top];
		Pop(S);
		if (M->rightChild != NULL)
			Push(S, M->rightChild);
		if (M->leftChild != NULL)
			Push(S, M->leftChild);
		printf("%c", M->data);
	}
	printf("\n");
	printf("\n");
}

后期会出二叉树的三种遍历方式的非递归实现，请大家多多支持！！