二叉排序树求解pku 1002

原题链接：

http://acm.pku.edu.cn/JudgeOnline/problem?id=1002

     这道题就是让把字母串形式的电话号码转换成纯数字形式，记录这些电话号码重复的次数，最后按照电话号码的字典序从小到大输出重复两次以上的电话号码，如果没有重复的电话号码，就输出 No duplicates.
     用二叉查找树很适合存储这些电话号码，最后按照中序遍历输出结果即可。

程序主要流程：
     1. 对每一个输入的字符串进行转换，转换成整数形式的电话号码 A；
     2. 查找二叉排序中是否有A，有，则计数器加1；没有，插入A；
     3.中序遍历二叉排序树，输出结果（把整数转换字符串输出）

Version 1：
     用链表形式表示二叉排序树，二叉树结点由malloc动态分配，插入算法采用递归。
     复杂度：内存：3176K 耗时：719MS

     #include <stdio.h> #include <stdlib.h> #include <string.h> struct node { int key;//电话号码对应的整数 int num;//出现次数 struct node* left_child; struct node* right_child; }; int flag=1;//全局结点,标记是否有重复值 //查找二叉排序树，如果树为空或者号码已经存在（计数器加1），返回NULL,否则返回最后查找的位置 struct node* search(struct node* root, int key) { if(!root) return NULL; int tmp = root->key - key; if(tmp==0) //root->key == key { //找到 root->num++; return NULL; } if(tmp>0) { //查找左子树，如果左子树为空，则root即为要插入的位置 if(!root->left_child) //==NULL return root; return search(root->left_child,key); } //tmp<0 //查找右子树，如果右子树为空，则root即为要插入的位置 if(!root->right_child) return root; return search(root->right_child, key); } //插入新节点 int insert(struct node* node,int key) { struct node* new_node = (struct node*)malloc(sizeof(struct node)); new_node->key =key; new_node->num=1; new_node->left_child=new_node->right_child=NULL; //一下判断插在左子树，还是右子树 int tmp = node->key - key; if(tmp==0) return 0; if(tmp>0) node->left_child = new_node; else node->right_child = new_node; return 1; } //装换成整数 int convert(char* str) { char new_str[8]={'/0'}; char* tmp =new_str; while(*str) { switch(*(str++)) { case 'A' : case 'B' : case 'C' : *(tmp++)='2'; break; case 'D' : case 'E' : case 'F' : *(tmp++)='3'; break; case 'G' : case 'H' : case 'I' : *(tmp++)='4'; break; case 'J' : case 'K' : case 'L' : *(tmp++)='5'; break; case 'M' : case 'N' : case 'O' : *(tmp++)='6'; break; case 'P' : case 'R' : case 'S' : *(tmp++)='7'; break; case 'T' : case 'U' : case 'V' : *(tmp++)='8'; break; case 'W' : case 'X' : case 'Y' : *(tmp++)='9'; break; case '-' : break; default: *(tmp++)=*(str-1); } } return atoi(new_str); } void build_tree(struct node ** root_pointer) { freopen("in.txt","r",stdin); int n; struct node* root=*root_pointer; scanf("%d",&n); while(n--) { char line[50]; scanf("%s",line); //printf("%s",line); int key = convert(line); //printf("%s/n",new_str); if(!root) { *root_pointer = root = (struct node *) malloc(sizeof(struct node)); root->key = key; root->num=1; root->left_child = NULL; root->right_child = NULL; }else { struct node * insert_point = search(root,key); if(insert_point) { insert(insert_point,key); //printf("insert %s /n",new_str); } } } } void print_node(struct node* node) { printf("%03d-%04d %d/n",node->key / 10000,node->key %10000,node->num); } //中序遍历 void inorder_tree(struct node* root) { if(!root) return ; inorder_tree(root->left_child); if(root->num>1) { print_node(root); flag=0; } inorder_tree(root->right_child); } int main() { struct node* root=NULL; build_tree(&root); inorder_tree(root); if(flag) printf("No duplicates./n"); return 0; } 

Version 2：     这个版本用全局数组表示二叉树，避免动态分配内存，节省时间。插入算法还是采用递归。
     复杂度：内存：1540K  耗时：594MS

    #include <stdio.h> #include <stdlib.h> #include <string.h> #define MAXSIZE 100001 //用全局数组代替malloc，加快速度,插入算法用递归 int flag=1;//全局结点,标记是否有重复值 int root=0;//根索引，0表示NULL int curr_size=0;//当前树中结点的个数 //树节点的定义，下标从1开始，0表示NULL int key[MAXSIZE]; int value[MAXSIZE]; int left_child[MAXSIZE]; int right_child[MAXSIZE]; //查找二叉排序树，如果树为空,返回0，号码已经存在（计数器加1），返回-1,否则返回最后查找的位置 int search(int node, int k) { if(!node) return 0; int tmp = key[node] - k; if(tmp==0) //key[node] == k { //找到 value[node]++; return -1; } if(tmp>0) { //查找左子树，如果左子树为空，则node即为要插入的位置 if(!left_child[node]) //==NULL return node; return search(left_child[node],k); } //tmp<0 //查找右子树，如果右子树为空，则node即为要插入的位置 if(!right_child[node]) return node; return search(right_child[node], k); } //插入新节点 int insert(int node,int k) { //判断一下插在左子树，还是右子树 int tmp = key[node] - k; if(tmp==0) return 0; int new_node = ++curr_size;//分配新位置 key[new_node]=k; value[new_node]=1; left_child[new_node]=0; right_child[new_node]=0; if(new_node==1)//树为空 root = 1; if(tmp>0) left_child[node] = new_node; else right_child[node] = new_node; return 1; } //装换成整数 int convert(char* str) { char new_str[8]={'/0'}; char* tmp =new_str; while(*str) { switch(*(str++)) { case 'A' : case 'B' : case 'C' : *(tmp++)='2'; break; case 'D' : case 'E' : case 'F' : *(tmp++)='3'; break; case 'G' : case 'H' : case 'I' : *(tmp++)='4'; break; case 'J' : case 'K' : case 'L' : *(tmp++)='5'; break; case 'M' : case 'N' : case 'O' : *(tmp++)='6'; break; case 'P' : case 'R' : case 'S' : *(tmp++)='7'; break; case 'T' : case 'U' : case 'V' : *(tmp++)='8'; break; case 'W' : case 'X' : case 'Y' : *(tmp++)='9'; break; case '-' : break; default: *(tmp++)=*(str-1); } } return atoi(new_str); } void build_tree() { //freopen("in.txt","r",stdin); int n; scanf("%d",&n); while(n--) { char line[50]; scanf("%s",line); //printf("%s",line); int key = convert(line); //printf("%s/n",new_str); int index = search(root,key); if(index>=0) { insert(index,key); } } } void print_node(int index) { printf("%03d-%04d %d/n",key[index] / 10000,key[index] %10000,value[index]); } //中序遍历 void inorder_tree(int root) { if(!root) return ; inorder_tree(left_child[root]); if(value[root]>1) { print_node(root); flag=0; } inorder_tree(right_child[root]); } int main() { build_tree(); inorder_tree(root); if(flag) printf("No duplicates./n"); return 0; } 

 

version 3：
     这个版本是将插入算法有递归改为循环实现，进一步加快下速度。

    //查找二叉排序树，如果树为空,返回0，号码已经存在（计数器加1），返回-1,否则返回最后查找的位置 int search(int node, int k) { int parent=0; int current=node; while(current!=0) { y=current; if(key[current] == k) { value[current]++; return -1; }else if(key[current]<k) { current = right_child[current]; }else { currrent = left_child[current]; } } return parent; }
小结：
     还可以将程序中的中序遍历改成非递归形式，再加快一下速度；另外一种用空间换时间，直接开个10000000的数组，然后在对应下标存储重复数，这个过程 比较节省时间，但是输出的时候，需要遍历比较多数组元素，比较耗时（内存：39608K  耗时：657MS）

    #include <stdio.h> #include <stdlib.h> //定义一个大数组，数组中存的是号码出现次数，但是遍历输出非常耗时 #define MAXSIZE 10000000 //装换成整数 int convert(char* str) { char new_str[8]={'/0'}; char* tmp =new_str; while(*str) { switch(*(str++)) { case 'A' : case 'B' : case 'C' : *(tmp++)='2'; break; case 'D' : case 'E' : case 'F' : *(tmp++)='3'; break; case 'G' : case 'H' : case 'I' : *(tmp++)='4'; break; case 'J' : case 'K' : case 'L' : *(tmp++)='5'; break; case 'M' : case 'N' : case 'O' : *(tmp++)='6'; break; case 'P' : case 'R' : case 'S' : *(tmp++)='7'; break; case 'T' : case 'U' : case 'V' : *(tmp++)='8'; break; case 'W' : case 'X' : case 'Y' : *(tmp++)='9'; break; case '-' : break; default: *(tmp++)=*(str-1); } } return atoi(new_str); } int num[MAXSIZE]={0}; int main() { freopen("in.txt","r",stdin); int n; int flag=1; scanf("%d",&n); int size=n; while(n--) { char line[50]; scanf("%s",line); int key = convert(line); num[key]++; //printf("key: %d/n",key); } int i=0; ///* while(size) { size-=num[i]; if(num[i]>1) { printf("%03d-%04d %d/n",i / 10000,i %10000,num[i]); flag=0; } i++; } //*/ if(flag) printf("No duplicates./n"); return 0; } 
     做这道题的本意其实是想练习一下好久都没写了的c语言，并顺便再熟悉一下二叉排序树。不写不知道，写了才发现问题很多，连基本的全局变量和局部变量的作用域都能搞错，最后还得靠一步一步的debug才知道指针所指的那些树节点已经被回收了。
     做v1的时候，起初把根节点指针直接传给build_tree()，以为就能构建出二叉树；实际上build_tree()里只是把root的形参指向了根节点，而root没有得到赋值，还是NULL。当时很怀疑，难道除了把root设为全局变量就没有办法了，最后用指针的指针搞定。
     遇到的问题还挺多，字符串常量的指针不能改变等等，都是些基础知识。还是古人说的好，纸上得来终觉浅，以后得多多实践才知道问题所在。废话了一堆，给自己提个醒，高手当且一笑呀，O(∩_∩)O~。