判断有没有找到要删除的节点我用的是设置一个k标志,如果找到了令k等于1,最后在main函数中根据k的值输出。
#include<stdio.h>
#include<stdlib.h>
typedef int TElemType;
typedef struct BiNode {
TElemType data;
struct BiNode* lchild, * rchild;
}BiTNode, *BiTree;
static int k = 0; //这是为了find函数内的标记处不被覆盖掉,并且改变值以后,在main函数中仍能保持
void DeleteChild(BiTree &T) {
if (!T) return;
if (T) {
DeleteChild(T->lchild);
DeleteChild(T->rchild);
T->data = 0;
T = NULL;
}
}
void find(BiTree T,int s) { //寻找要删除的节点,调用DeleteChid()函数
if (T == NULL) return;
if (T->data == s) {
DeleteChild(T);
k = 1; //做一个标记,在main函数中判断,如果k==0那么意味着没有找到要删除的节点,输出0
}
if (T != NULL) {
find(T->lchild, s);
find(T->rchild, s);
}
}
void InOrderTraverse(BiTree T) {
if (!T) return;
if (T->data == 0) return;
InOrderTraverse(T->lchild);
printf("%d ", T->data);
InOrderTraverse(T->rchild);
}
void CreatBiTree(BiTree &T) {
int a;
scanf_s(" %d", &a);
if (a == 0) T = NULL;
else {
T = new BiTNode;
T->data = a;
CreatBiTree(T->lchild);
CreatBiTree(T->rchild);
}
}
int main() {
int n = 0,s=0;
BiTree tree = (BiTree)malloc(sizeof(BiTNode));
BiTree d = (BiTree)malloc(sizeof(BiTNode));
CreatBiTree(tree);
scanf_s("%d",&n);
while (n) {
k = 0;
scanf_s("%d",&s);
find(tree,s);
if (k == 0) {
printf("0");
}
else {
InOrderTraverse(tree);
}
printf("\n");
n--;
}
return 0;
}