菜鸟修炼C语言小设计之——通讯录(二)

通讯录在通讯录（一）的基础上作了一些完善，添加了保存联系人的功能。

主要涉及的C语言重要知识点有：

文件流的操作

代码：

main.c

#include <stdio.h> 

#include "record.h" 

int menu_select(void); 

void hand_menu(int cmd, int *flag); 

int main(int argc, char *argv[]) 

{ 

    int cmd = 0; 

    int flag = 1; 

    while(1){ 

        cmd = menu_select(); 

        if(cmd == '0') 

            return 0; 

        hand_menu(cmd, &flag); 

    } 

} 

int menu_select(void) 

{ 

    int select; 

    printf("    <------通信薄-------->/n"); 

    printf("1:添加联系人     2:删除联系人/n"); 

    printf("3:显示所有联系人   4:保存/n"); 

    printf("0:退出/n"); 

    printf("请输入:/n"); 

    select = getch(); 

    while(select  < '0'|| select>'4') { 

        printf("输入错误,请重新输入:/n"); 

        select = getch(); 

    } 

    return select; 

} 

void hand_menu(int cmd, int *flag) 

{ 

    static ADDR *list_head = NULL; 

    if(1 == *flag){ 

        list_head = init_person(list_head); 

        *flag  = 0; 

    }  

    switch(cmd){ 

        case '1': 

            list_head  = add_person(list_head); 

            break; 

        case '2': 

            list_head = del_person(list_head); 

            break; 

        case '3': 

            dis_person(list_head); 

            break; 

        case '4': 

            save_person(list_head); 

            break; 

        default: 

            break; 

    } 

}<strong> 

</strong> 

record.h

#ifndef _RECORD_H_ 

#define _RECORD_H_ 

typedef struct{ 

    char name[8]; 

    char tel[20]; 

}DATA; 

typedef struct node{ 

    DATA data; 

    struct node *next; 

}ADDR; 

#define ASK(p) do{/ 

    p = (ADDR *)malloc(sizeof(ADDR));/ 

    if(p==NULL){printf("malloc memory failed!");exit(-1);}/ 

}while(0) 

#endif 

opre.c

#include <stdio.h> 

#include "record.h" 

#define FILE_NAME "phonebook.dat" 

ADDR* add_person(ADDR *list_head) 

{ 

    ADDR *head = list_head; 

    ADDR *node = list_head; 

    ADDR *new_p; 

    ASK(new_p); 

    new_p->next = NULL; 

    printf("请输入姓名:"); 

    scanf("%s", new_p->data.name); 

    printf("请输入电话号码:"); 

    scanf("%s", new_p->data.tel); 

    if(!node){ 

        head = new_p; 

        return head; 

    } 

    while(node->next) 

        node=node->next; 

    node->next = new_p; 

    return head; 

} 

ADDR *del_person(ADDR *list_head) 

{ 

    ADDR *node = list_head; 

    ADDR *head = list_head; 

    char name[8]; 

    ADDR *pre = node; 

    printf("请输入要删除的名字:"); 

    scanf("%s", name); 

    if(!strcmp(head->data.name, name)){ 

        pre = head; 

        head = head->next; 

        free(pre); 

        return head; 

    } 

    while(node){ 

        if(!strcmp(node->data.name, name)){ 

            pre->next = node->next; 

            free(node); 

            printf("成功删除!/n"); 

            return; 

        } 

        pre = node; 

        node = node->next; 

    } 

    printf("没有找到该名字!/n"); 

    return head; 

} 

void dis_person(ADDR *list_head) 

{ 

    ADDR *node = list_head; 

    if(!node) 

        return; 

    printf("姓名          号码/n"); 

    while(node){ 

        printf("%s          %s/n", node->data.name, node->data.tel); 

        node = node->next; 

    }        

} 

void save_person(ADDR *list_head) 

{ 

    FILE *pf; 

    ADDR *node = list_head; 

    pf = fopen(FILE_NAME, "w+"); 

    while(node){ 

        fprintf(pf, "%s         %s/n", node->data.name, node->data.tel); 

        node = node->next; 

    }        

    fclose(pf); 

    printf("保存成功！/n"); 

} 

ADDR *init_person(ADDR *list_head) 

{ 

    ADDR *node = list_head; 

    ADDR *head = list_head; 

    ADDR *new_node; 

    FILE *pf; 

    char name[8]; 

    char tel[20]; 

    ASK(new_node); 

    pf = fopen(FILE_NAME, "r"); 

    if(!fscanf(pf, "%s%s", new_node->data.name, new_node->data.tel)){ 

        free(new_node); 

        return head; 

     } 

    rewind(pf); 

    while(fscanf(pf, "%s%s", name, tel) == 2){ 

            ASK(new_node); 

            new_node->next = NULL; 

            strcpy(new_node->data.name, name); 

            strcpy(new_node->data.tel, tel); 

            if(!head) 

                node = head = new_node; 

            else{ 

                while(node->next) 

                    node = node->next; 

                node->next = new_node; 

            } 

     } 

    return head; 

} 

<pre>