C语言创建二叉树过程中遇到的一些问题

其实这些问题以前也遇到过，不过没总结，遇到又要搞段时间，现在总结下。

比如我们要创建颗如下二叉树，按前序输入，则为AB#D##C##：

开始代码写成下面这样：

#define TElemType char

typedef struct BiTNode {
	TElemType data;
	struct BiTNode *lchild,*rchild;
}BiTNode,*BiTree;  

void CreateBiTree(BiTree T) {
	TElemType ch;
	scanf("%c",&ch);
	if(ch == '#') {	
		T = NULL;
	} else {
		T = (BiTree)malloc(sizeof(BiTNode));
		T->data = ch;
		CreateBiTree(T->lchild);
		CreateBiTree(T->rchild);
		return T;
	}
}

其实这种错误和下面这种a，b无法被交换本质上是一样的，只是int类型变成了结构体指针类型：

#include<stdio.h>

void swap(int a,int b) {
	int temp = a;
	a = b;
	b = temp;
}

int main() {
	int a = 3,b = 4;
	swap(a,b);
	printf("%d %d\n",a,b);
}

所以类比两个数交换时的处理方式， 自然可以想到用下面这种方式来处理:

void CreateBiTree(BiTree *T) {
	TElemType ch;
	scanf("%c",&ch);
	if(ch == '#') {
		*T = NULL;
	} else {
		*T = (BiTree)malloc(sizeof(BiTNode));
		(*T)->data = ch;
		CreateBiTree(&(*T)->lchild);
		CreateBiTree(&(*T)->rchild);
	}
}

就是不是传结构体指针的值，而是传结构体指针的指针。

指针的指针不好理解，还有一种处理方式如下所示：

BiTree CreateBiTree() {
	TElemType ch;
	scanf("%c",&ch);
	if(ch == '#') {	
		return NULL; 
	} else {
		BiTree T = (BiTree)malloc(sizeof(BiTNode));
		T->data = ch;
		T->lchild = CreateBiTree();
	 	T->rchild = CreateBiTree();
		return T;
	}
}

我开始写的程序就错在在递归调用的时候没有将返回值赋值给父节点的左右孩子。

最后给个正确的完整代码，并用前序遍历输出了构造的二叉树：

#include<stdio.h>
#include<stdlib.h>

#define TElemType char

typedef struct BiTNode {
	TElemType data;
	struct BiTNode *lchild,*rchild;
}BiTNode,*BiTree;  

BiTree CreateBiTree() {
	TElemType ch;
	scanf("%c",&ch);
	if(ch == '#') {	
		return NULL; 
	} else {
		BiTree T = (BiTree)malloc(sizeof(BiTNode));
		T->data = ch;
		T->lchild = CreateBiTree();
	 	T->rchild = CreateBiTree();
		return T;
	}
}

void traverse(BiTree r) {
	if(r == NULL) {
		return;
	}
	printf("%c",r->data);
	traverse(r->lchild);
	traverse(r->rchild);
}

int main() {
	BiTree root;
	root = CreateBiTree();
	traverse(root);
	printf("\n");
}

如果你要用指针的指针来构建的话完整代码如下：

#include<stdio.h>
#include<stdlib.h>

#define TElemType char

typedef struct BiTNode {
	TElemType data;
	struct BiTNode *lchild,*rchild;
}BiTNode,*BiTree;  

void CreateBiTree(BiTree *T) {
	TElemType ch;
	scanf("%c",&ch);
	if(ch == '#') {
		*T = NULL;
	} else {
		*T = (BiTree)malloc(sizeof(BiTNode));
		(*T)->data = ch;
		CreateBiTree(&(*T)->lchild);
		CreateBiTree(&(*T)->rchild);
	}
}

void traverse(BiTree r) {
	if(r == NULL) {
		return;
	}
	printf("%c",r->data);
	traverse(r->lchild);
	traverse(r->rchild);
}

int main() {
	BiTree root;
	CreateBiTree(&root);
	traverse(root);
	printf("\n");
}