单词吸血鬼源代码 二叉树操作

/*****************************************************************
程序名：单词吸血鬼
作者：  常涛

GNU认证

  程序功能简介：可以提取出以ASCII码保存的文件中的单词并以字典顺序输出

*****************************************************************/

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

struct node
{
   struct node *left;   /* tree to the left */
   struct node *right;  /* three to the right */
   char   *word;        /*  word for this tree */
};

/*the top of the tree */
static struct node *root=NULL;

/****************************************************
* 内存出错报警                                     *
****************************************************/

void memory_error(char* text)
{
   fprintf(stderr,text);
   exit(8);
}

/****************************************************
* save_string --将string保存到堆防止栈溢出.         *
*                                                   *
* 参数：                                            *
*        string -- 要存的字符串  .                  *
*                                                   *
* 返回值：                                          *
*      指针                                         *
****************************************************/

char *save_string(char *string)
{
   char *new_string;      /*初学者易犯的错误，对没有初始化的指针赋值*/

   new_string = malloc((unsigned)(strlen(string))+1);  //关键步骤,new和malloc一样，都是分配在堆的

   if(new_string == NULL)
   {
       memory_error("没有足够的内存用于分配");
   }

   strcpy(new_string, string);     //相当有用的函数

   return(new_string);             
}

/****************************************************
* enter -- 向树中加入一个值                         *
*                                                   *
* 参数                                              *
*      node -- 正在操作的树                         *
*      word -- 要输入的值                           *
****************************************************/

void enter(struct node **node, char *word)
{
   int result;                  /*暂存比较结果p*/

   char *save_string(char *);   /*s数据入堆*/
   /*
    * 如果满树开始建立新的内存块
   */

   if((*node) == NULL)
   {
       (*node) = malloc(sizeof(struct node));

       if((*node) == NULL)
       {
    memory_error("没有足够的内存用于分配");
    return;
       }

       (*node) -> left = NULL;              
       (*node) -> right = NULL;

       (*node) -> word = save_string(word);   //这步因该想办法理解它的好处

   }

   /* 选择数据保存的位置 */
   result = strcmp((*node) -> word, word);
  
   /*如果已经存在要存入的数据，则返回 */

   if(result == 0) return;
  
   /* 数据将会依据result的值来选择存储点 */

   if(result < 0)
   {
       enter(&(*node) -> right, word);
   }
   else
   {
       enter(&(*node) -> left,  word);
   }
}

/*****************************************************************
* scan -- 扫描并读入文件          .                              *
*                                                                *
* 参数：                                                         *
*       name -- 要读人的文件名和                                 *
*****************************************************************/

void scan(char *name)
{
   char word[100];            /*每次操作的数据*/
   int index=0;               /* 字符索引  */ 
   int ch = 0;                /* 当前字符r    */
   FILE *in_file;             /* 输入的文件指针           */

   in_file = fopen(name,"r");
   if (in_file == NULL)
   {
       fprintf(stderr,"Error:Unable to open %s/n", name);
       exit(8);
   }
   while(1)
   {
       while(1)
       {
    ch = fgetc(in_file);

    if(isalpha(ch) || (ch == EOF))
    {
        break;
    }
       }

       if(ch == EOF) break;

       word[0] = ch;
       for(index = 1; index < sizeof(word); ++index)
       {
    ch = fgetc(in_file);
    word[index]=ch;

    if(!isalpha(ch)) break;
       }
       word[index] = '/0';

       enter(&root,word);
   }

   fclose(in_file);
}

/*************************************************************
* print_tree -- 输出整个树                                   *
*                                                            *
* 参数                                                       *
*    top -- 要输出树的根节点                                 *
*************************************************************/

void print_tree(struct node *top)
{
   if(top == NULL) return;        /* 元树则返回 */
   print_tree(top -> left);
   printf("%s/n",top -> word);
   print_tree(top -> right);
}

int main(int argc, char *argv[])
{
 
   if(argc !=2)
   {
       fprintf(stderr,"错误:没有足够的参数/n");
       fprintf(stderr,"     在命令行窗口输入 /n");
       fprintf(stderr,"     用法如下:/n");
       fprintf(stderr,"     wordsvampire '文件名'(文件名因该是全称就是要有扩展名)/n");
       system("pause");
       exit(8);
   }
  
   scan(argv[1]);
   print_tree(root);
   system("pause");
   return(0);
}