我们亲爱的项目经理真是有创意，他说你给我写得二叉树挺好的；
功能还算可以：插入节点，能够删除节点；
可是有时候我们只是需要查找树的某个节点是否存在；
所以我希望你能够给我一个find功能；
还有就是，我在跟BOSS交流的时候，他根本不相信你已经把链表写成树的结构啦；
他希望你能够更明显的把树的结构打印出来；
所以，你需要实现一个show功能；
这应该不难吧；
Problem
我们需要实现的其实是遍历的两个变体；
首先确定我们的思路，想到都是需要遍历的；
那么我们是不是又想到了有递归的方式和非递归的方式呢；
int find_recursive(Node *root, int data)
{
//TODO
}
我们首先实现的是递归的，大致的结构应该是这样的；
还记得吗？递归的前面几行一般是递归的结束条件；
因此，我们的代码应该是这样的：
int find_recursive(Node *root, int data)
{
if(root == NULL)
return ;
}
其次，我们应当考虑找到目标的时候，应该是怎么确定的呢？
应该是root->data 与参数data相等的时候，我们希望程序返回啦；
因此，我们需要的代码可能会是这样的；
int find_recursive(Node *root, int data)
{
if(root == NULL){
return ;
}else if(root->data == data){
return ;
}
}
那，在当前的节点上如果没有找到目标呢；
我们需要的应该是确定往左找还是往右找；
所以需要把代码写成如下的形式；
int find_recursive(Node *root, int data)
{
if(root == NULL){
return ;
}else if(root->data == data){
return ;
}else{
int dir = root->data < data;
return find_recursive(root->link[dir],data);
}
}
dir的值只能是0或者1，代表的意义是什么大家应该很清楚啦；
它的作用是什么也应该很清楚啦；
最后执行递归；
通常我们需要客户操作的是树，因此我们需要一个包装函数；
int find(Tree *tree,int data)
{
return find_recursive(tree->root,data);
}
好啦，我们可以测试一下啦；
我首先把需要用到得代码附上；
#include <stdio.h>
#include <stdlib.h>
#include <time.h> typedef struct _node {
int data;
struct _node *link[];
}Node; typedef struct _tree{
struct _node *root;
}Tree; Tree * init_tree()
{
Tree *temp = (Tree*)malloc(sizeof(Tree));
temp->root = NULL;
return temp;
} Node * make_node(int data)
{
Node *temp = (Node*)malloc(sizeof(Node));
temp->link[] = temp->link[] = NULL;
temp->data = data;
return temp;
} int insert(Tree *tree, int data)
{
if(tree->root == NULL){
tree->root = make_node(data);
}else{
Node * it = tree->root;
int dir ;
for(;;){
dir = it->data < data;
if(it->data == data){
return ;
}else if(it->link[dir] == NULL){
break;
}
it = it->link[dir];
}
it->link[dir] = make_node(data);
}
return ;
}
测试主函数代码如下：
int main(void)
{
Tree * tree = init_tree();
insert(tree,);
insert(tree,);
insert(tree,);
insert(tree,);
printf("looking for %d...%s\n",,find(tree,) ? "Got it":"Not exist");
printf("looking for %d...%s\n",,find(tree,) ? "Got it":"Not exist");
return ;
}
运行结果如下：
好啦！可能对于客户来说，这样把查询的数据写死啦，并把那么好；
别人可能想查某个数据就输入某个数据，可能更实用；
这是非常简单的功能，我不会去讲解，但是也附上一份简单的实现；
大家写的肯的能够更加漂亮；
void tips(){
time_t curTime;
struct tm *ts;
time_t testTime = time(&curTime);
ts = localtime(&curTime);
//struct tm *tsgm = gmtime(&curTime);
puts("*************************************************************");
printf("Date:%s",asctime(ts));
puts("Please input right cmd");
puts("1)Search");
puts("2)Quit");
puts("*************************************************************");
}
int search(Tree *tree)
{
char cmd;
int isExist;
int target;
tips();
while((cmd=getchar())!=EOF){
if(cmd==''){
puts("-------------------------------------------------------------");
printf("Please input the target:");
scanf("%d",&target);
isExist=find(tree,target);
if(isExist){
printf("Target %d is exists \n",target);
puts("-------------------------------------------------------------");
puts("\n");
continue;
//break;
}else{
puts("Target Not exitst");
puts("-------------------------------------------------------------");
puts("\n");
continue;
//break;
}
}else if(cmd==''){
puts("bye~");
break;
}else{
tips();
}
}
return ;
}
运行结果如下：
这个可能用户体验要较好些；
注意别忘记包涵头文件time.h
好啦查询工作算是完成啦；
接下来是下一个任务；
Problem
为啦更加形象的表明我们一直在写得是一棵树；
我想有必要show也就是秀秀我们的成果啦；
这次咱们先看要实现的效果，如下图；
我们的树根就是8，而~符合代表就是叶子啦；
这样旋转90°的树可能你不一定喜欢；
但是，如果你想再旋转90度的话你就自己想想吧；
好啦，接下来就是实现啦；
不管它咱们显示，咱们始终记住，我们需要遍历；
为简单，我们使用递归实现；
void structure_recursive(Node *root,int level)
{
if(root == NULL){
//TODO
}else{
//TODO;
}
}
这个遍历方式结构就怎么简单；
但是请思考思考，我们的输出到终端上得数据是先大的，然后是小的；
显然是只有中序遍历才能得到按正确的大小顺序输出；
所以我们需要记住这点哈；
void structure_recursive(Node *root,int level)
{
if(root == NULL){
//TODO
}else{
structure_recursive(root->link[],level + );
printf("%d\n",root->data);
structure_recursive(root->link[],level + );
}
}
你会发现，我首先写else部分，仔细看看else部分你会发现我们的中序遍历很明显啦；
可是呢!!它又有别于我们前面写得中序遍历方式，我们首先从右子树下手啦；
以前都是从左子树的，这就是为什么我们输出的结果是从大到小往下的啦；
好啦！TODO到底应该写成什么样子呢？
想想要是我们一直递归到最后要是递归一次我们就得增加一个level的计数；
到最后发现叶子是NULL我们就直接输出level个tab然后跟一个~表示叶子；
所以代码如下：
void structure_recursive(Node *root,int level)
{
int i;
if(root == NULL){
for(i = ;i<level;i++){
putchar('\t');
}
puts("~");
}else{
structure_recursive(root->link[],level + );
printf("%d\n",root->data);
structure_recursive(root->link[],level + );
}
}
这里已经差不多完事啦，但是我们先去写一个包装函数；
void structure(Tree *tree)
{
structure_recursive(tree->root,);
}
注意包装函数里面的0表示是从最左边开始打印的；
我们先不去完善我们的structure_recursive函数，我们直接测试，看看结果；
这个可不是我们想看到的结果哈；
原因是什么呢；
我们还没有完善我们的函数呢；
void structure_recursive(Node *root,int level)
{
int i;
if(root == NULL){
for(i = ;i<level;i++){
putchar('\t');
}
puts("~");
}else{
structure_recursive(root->link[],level + );
for(i = ;i < level;i++){
putchar('\t');
}
printf("%d\n",root->data);
structure_recursive(root->link[],level + );
}
}
好啦，我们的效果出来啦；
好啦！工作完成啦；
总结一下吧；
不，还没有到总结的时候；
明天的任务还有呢：
我们的树不能一直老想着用递归实现，在很多时候它是受限制的；
递归对空间要求特别多，所以我们下次想想有没有非递归的方法实现其他的函数；
最后附上我们今天所有代码吧：
#include <stdio.h>
#include <stdlib.h>
#include <time.h> typedef struct _node {
int data;
struct _node *link[];
}Node; typedef struct _tree{
struct _node *root;
}Tree; Tree * init_tree()
{
Tree *temp = (Tree*)malloc(sizeof(Tree));
temp->root = NULL;
return temp;
} Node * make_node(int data)
{
Node *temp = (Node*)malloc(sizeof(Node));
temp->link[] = temp->link[] = NULL;
temp->data = data;
return temp;
} int insert(Tree *tree, int data)
{
if(tree->root == NULL){
tree->root = make_node(data);
}else{
Node * it = tree->root;
int dir ;
for(;;){
dir = it->data < data;
if(it->data == data){
return ;
}else if(it->link[dir] == NULL){
break;
}
it = it->link[dir];
}
it->link[dir] = make_node(data);
}
return ;
}
int remove_node(Tree *tree, int data)
{
if(tree->root != NULL)
{
Node *p = NULL;
Node *succ;
Node *it = tree->root;
int dir;
for(;;){
if( it == NULL){
return ;
}else if(it->data == data){
break;
}
dir = it->data < data;
p = it;
it = it->link[dir];
}
/***********************************************************************/ if(it->link[] != NULL && it->link[] != NULL){
p = it;
succ = it->link[];
while(succ->link[] != NULL){
p = succ;
succ = succ->link[];
}
it->data = succ->data;
p->link[p->link[] == succ] = succ->link[];
free(succ);
/***********************************************************************/
}else{ /*it->link[0] == NULL || it->link[1] == NULL */
dir = it->link[] == NULL;
if( p == NULL){
tree->root = it->link[dir];
}else{
p->link[p->link[] == it] = it->link[dir];
}
free(it);
}
} return ;
}
int find_recursive(Node *root, int data)
{
if(root == NULL){
return ;
}else if(root->data == data){
return ;
}else{
int dir = root->data < data;
return find_recursive(root->link[dir],data);
}
} int find(Tree *tree,int data)
{
return find_recursive(tree->root,data);
} void tips(){
time_t curTime;
struct tm *ts;
time_t testTime = time(&curTime);
ts = localtime(&curTime);
//struct tm *tsgm = gmtime(&curTime);
puts("*************************************************************");
printf("Date:%s",asctime(ts));
puts("Please input right cmd");
puts("1)Search");
puts("2)Quit");
puts("*************************************************************");
}
int search(Tree *tree)
{
char cmd;
int isExist;
int target;
tips();
while((cmd=getchar())!=EOF){
if(cmd==''){
puts("-------------------------------------------------------------");
printf("Please input the target:");
scanf("%d",&target);
isExist=find(tree,target);
if(isExist){
printf("Target %d is exists \n",target);
puts("-------------------------------------------------------------");
puts("\n");
continue;
//break;
}else{
puts("Target Not exitst");
puts("-------------------------------------------------------------");
puts("\n");
continue;
//break;
}
}else if(cmd==''){
puts("bye~");
break;
}else{
tips();
}
}
return ;
} void structure_recursive(Node *root,int level)
{
int i;
if(root == NULL){
for(i = ;i<level;i++){
putchar('\t');
}
puts("~");
}else{
structure_recursive(root->link[],level + );
for(i = ;i < level;i++){
putchar('\t');
}
printf("%d\n",root->data);
structure_recursive(root->link[],level + );
}
} void structure_recursive(Node *root,int level)
{
int i;
if(root == NULL){
for(i = ;i<level;i++){
putchar('\t');
}
puts("~");
}else{
structure_recursive(root->link[],level + );
for(i = ;i < level;i++){
putchar('\t');
}
printf("%d\n",root->data);
structure_recursive(root->link[],level + );
}
} void structure(Tree *tree)
{
structure_recursive(tree->root,);
} int main(void)
{
Tree * tree = init_tree();
insert(tree,);
insert(tree,);
insert(tree,);
insert(tree,);
insert(tree,);
insert(tree,);
structure(tree);
return ;
}