我们之前完成了简单通讯录,没有看过的朋友们可以先看一下简单通讯录
这次我们来对通讯录进行升级,装载动态内存管理以及文件操作,我们一步一步来
目录
1.装载动态内存管理
1.1结构体的改造及初始化的改造
在使用通讯录时,对于不同的人可能有不同的需求,比如有人可能只需要几个联系人的信息,而有人可能需要几百个,对于需求少的人,会有空间浪费,对于需求多的人,我们要进行扩容,所以我们先要对通讯录的结构进行改造,我们要把存储人信息的空间改为动态,比如初始只能存放3个人的信息,当空间满后会进行扩容,每次扩容两个空间,所以我们需要对结构体先进行改造,将之前的数组改造为指针,并且,我们还需要添加一个变量来记录当前空间的最大容量
typedef struct Contact {
PeoInfo* data;//指向存放人的信息的空间
int sz;//当前存放信息数量
int capacity;//当前通讯录的最大容量
}Contact;
既然我们对结构体进行了改造,那么初始化就得跟着进行改造,并且为了方便后续代码,我们要使用#define来定义一些变量
#define DEFAULT_SZ 3//初始容量
#define INC_SZ 2 //扩容容量
void InitContact(Contact* pc) {
assert(pc);
pc->sz = 0;
PeoInfo* ptr = (PeoInfo*)calloc(DEFAULT_SZ ,sizeof(PeoInfo));
if (ptr==NULL) {
perror("InitContact->calloc");
return;
}
pc->data = ptr;
pc->capacity = DEFAULT_SZ;
}
我们使用calloc进行开辟空间,并且对接收的指针进行判断,如果有错误,用perror打印出错误信息,然后直接返回,开辟空间成功,我们将指针传给data,最后将容量置为初始值
1.2增容及新建联系人改造
void check_capacity(Contact* pc) {//扩容
if (pc->sz == pc->capacity) {
//扩容
PeoInfo* ptr = (PeoInfo*)realloc(pc->data, sizeof(PeoInfo) * (pc->capacity + INC_SZ));
if (ptr == NULL) {
perror("check_capacity->realloc");
return;
}
pc->data = ptr;
pc->capacity += INC_SZ;
}
}
//新建联系人,动态版本
void AddContact(Contact* pc) {
assert(pc);
check_capacity(pc);
//添加一个联系人
printf("请输入姓名:>");
scanf("%s", pc->data[pc->sz].name);
printf("请输入年龄:>");
scanf("%d", &(pc->data[pc->sz].age));
printf("请输入性别:>");
scanf("%s", pc->data[pc->sz].sex);
printf("请输入住址:>");
scanf("%s", pc->data[pc->sz].addr);
printf("请输入电话:>");
scanf("%s", pc->data[pc->sz].tele);
pc->sz++;
}
我们将扩容函数单独封装为一个函数,也方便以后使用,扩容需要先判断当前存放信息数是否等于最大信息数,扩容时我们使用realloc进行扩容,接着进行判断扩容是否成功,失败时用perror打印错误信息,AddContact变化不大,改掉判断是否已满的部分即可
1.3空间释放及通讯录销毁
当我们关闭通讯录时,我们之前申请的空间也需要释放,所以我们要在退出时添加一个函数
//销毁通讯录
void DestroyContact(Contact* pc) {
free(pc->data);
pc->data = NULL;
pc->sz = 0;
pc->capacity = 0;
pc = NULL;
}
2.装载文件管理系统
我们的通讯录每次关闭都会导致之前输入的数据消失,每次运行代码和上一次毫无关联,要想解决这个问题,就需要我们的文件操作
2.1保存信息到文件
首先我们要在每次退出程序前保存我们的信息
//保存通讯录信息到文件里
void SaveContact(Contact* pc) {
FILE* pf = fopen("contact.txt", "wb");
if (pf == NULL) {
perror("SaveContact");
return;
}
else {
int i = 0;
for (i = 0; i < pc->sz; i++) {
fwrite(pc->data + i,sizeof(PeoInfo),1,pf);
}
fclose(pf);
pf = NULL;
printf("保存数据成功!\n");
}
}
我们使用二进制写入数据,每次写一个,使用pc->data+i这样的偏移量计算来确定写入数据的位置,然后关闭文件,输出提示信息即可,最后别忘了在switch语句里加上函数调用
2.2加载文件信息到通讯录
当我们保存好信息时,就可以使用这些信息了,我们要改造一下通讯录的初始化,让初始化时将文件的信息加载到我们的通讯录里,当然这个功能我们封装成函数在初始化里调用更好
//加载文件信息到通讯录
void LoadContact(Contact* pc) {
FILE* pf = fopen("contact.txt", "rb");
if (pf == NULL) {
perror("LoadContact");
return;
}
else {
PeoInfo tmp = { 0 };
int i = 0;
while (fread(&tmp, sizeof(PeoInfo), 1, pf)) {
check_capacity(pc);
pc->data[i] = tmp;
pc->sz++;
i++;
}
fclose(pf);
pf = NULL;
}
}
我们在初始化函数的最后调用这个函数,我们来看加载信息函数,我们读取数据时不知道文件里有多少个数据,但是fread的返回值是读取到数据的个数,比如我们每次读取5个,但如果只有三个数据的话,返回值就是3,所以我们每次读取一个,用while循环来控制,当文件里数据读取完时,返回值就是0,循环也会停止,我们在加载信息时还要先调用扩容函数,防止空间不足
以上即为本次全部内容,希望大家可以有所收获,下面我会附上完整代码
3.完整代码
#pragma once
//Contact.h
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<assert.h>
#define MAX 100//最大联系人数量
#define NAME_MAX 20//最大名字字符数
#define SEX_MAX 5//最大性别字符数
#define ADDR_MAX 30//最大住址字符数
#define TELE_MAX 12//最大电话字符数
#define DEFAULT_SZ 3//初始容量
#define INC_SZ 2 //扩容容量
typedef struct PeoInfo {
char name[NAME_MAX];//名字
int age;//年龄
char sex[SEX_MAX];//性别
char addr[ADDR_MAX];//住址
char tele[TELE_MAX];//电话
}PeoInfo;
//静态版本
//typedef struct Contact {
// PeoInfo data[MAX];//存放人的信息
// int sz;//当前存放信息数量
//}Contact;
//动态版本
typedef struct Contact {
PeoInfo* data;//指向存放人的信息的空间
int sz;//当前存放信息数量
int capacity;//当前通讯录的最大容量
}Contact;
//初始化通讯录
void InitContact(Contact* pc);
//销毁通讯录
void DestroyContact(Contact* pc);
//新建联系人
void AddContact(Contact* pc);
//显示通讯录中所有联系人信息
void ShowContact(const Contact* pc);
//删除联系人
void DelContact(Contact* pc);
//查找联系人
void SearchContact(const Contact* pc);
//修改联系人信息
void ModifyContact(Contact* pc);
//通讯录排序
void SortContact(Contact* pc);
//保存通讯录信息到文件里
void SaveContact(Contact* pc);
//加载文件信息到通讯录
void LoadContact(Contact* pc);
#include"Contact.h"
//Contact.c
初始化通讯录,静态版本
//void InitContact(Contact* pc){
// assert(pc);
// pc->sz = 0;
// memset(pc->data, 0,sizeof(pc->data));
//}
//初始化通讯录,动态版本
void InitContact(Contact* pc) {
assert(pc);
pc->sz = 0;
PeoInfo* ptr = (PeoInfo*)calloc(DEFAULT_SZ ,sizeof(PeoInfo));
if (ptr==NULL) {
perror("InitContact->calloc");
return;
}
pc->data = ptr;
pc->capacity = DEFAULT_SZ;
//加载文件信息到通讯录
LoadContact(pc);
}
//销毁通讯录
void DestroyContact(Contact* pc) {
free(pc->data);
pc->data = NULL;
pc->sz = 0;
pc->capacity = 0;
pc = NULL;
}
//新建联系人,静态版本
//void AddContact(Contact* pc) {
// assert(pc);
// if (pc->sz == MAX) {
// printf("通讯录已满,无法添加!\n");
// return;
// }
// //添加一个联系人
// printf("请输入姓名:>");
// scanf("%s", pc->data[pc->sz].name);
// printf("请输入年龄:>");
// scanf("%d", &(pc->data[pc->sz].age));
// printf("请输入性别:>");
// scanf("%s", pc->data[pc->sz].sex);
// printf("请输入住址:>");
// scanf("%s", pc->data[pc->sz].addr);
// printf("请输入电话:>");
// scanf("%s", pc->data[pc->sz].tele);
// pc->sz++;
//}
void check_capacity(Contact* pc) {//扩容
if (pc->sz == pc->capacity) {
//扩容
PeoInfo* ptr = (PeoInfo*)realloc(pc->data, sizeof(PeoInfo) * (pc->capacity + INC_SZ));
if (ptr == NULL) {
perror("check_capacity->realloc");
return;
}
pc->data = ptr;
pc->capacity += INC_SZ;
}
}
//新建联系人,动态版本
void AddContact(Contact* pc) {
assert(pc);
check_capacity(pc);
//添加一个联系人
printf("请输入姓名:>");
scanf("%s", pc->data[pc->sz].name);
printf("请输入年龄:>");
scanf("%d", &(pc->data[pc->sz].age));
printf("请输入性别:>");
scanf("%s", pc->data[pc->sz].sex);
printf("请输入住址:>");
scanf("%s", pc->data[pc->sz].addr);
printf("请输入电话:>");
scanf("%s", pc->data[pc->sz].tele);
pc->sz++;
}
//显示通讯录中所有联系人信息
void ShowContact(const Contact* pc) {
assert(pc);
int i = 0;
printf("%-20s\t%-4s\t%-5s\t%-20s\t%-12s\n","姓名","年龄","性别","住址","电话号码");
for (i = 0; i < pc->sz; i++) {
printf("%-20s\t%-4d\t%-5s\t%-20s\t%-12s\n",pc->data[i].name,
pc->data[i].age,
pc->data[i].sex,
pc->data[i].addr,
pc->data[i].tele);
}
}
//根据名字查找
FindByName(const Contact* pc,char name[]) {
int i = 0;
for (i = 0; i < pc->sz; i++) {
if (strcmp(pc->data[i].name, name) == 0) {
return i;
}
}
return -1;
}
//删除联系人
void DelContact(Contact* pc) {
assert(pc);
char name[NAME_MAX] = { 0 };
if (pc->sz == 0) {
printf("通讯录为空!无法删除!\n");
return;
}
//查找需要删除的人
printf("请输入需要删除联系人的姓名:>");
scanf("%s", name);
int ret = FindByName(pc, name);
if (ret == -1) {
printf("该联系人不存在!\n");
return;
}
//删除
int i = 0;
for (i = ret; i < pc->sz - 1; i++) {
pc->data[i] = pc->data[i + 1];
}
pc->sz--;
printf("删除联系人成功!\n");
}
//查找联系人
void SearchContact(const Contact* pc) {
assert(pc);
char name[NAME_MAX] = { 0 };
printf("请输入要查找的联系人的姓名:>");
scanf("%s", name);
int pos = FindByName(pc, name);
if (pos == -1) {
printf("该联系人不存在!\n");
return;
}
printf("%-20s\t%-4s\t%-5s\t%-20s\t%-12s\n", "姓名", "年龄", "性别", "住址", "电话号码");
printf("%-20s\t%-4d\t%-5s\t%-20s\t%-12s\n", pc->data[pos].name,
pc->data[pos].age,
pc->data[pos].sex,
pc->data[pos].addr,
pc->data[pos].tele);
}
//修改联系人信息
void ModifyContact(Contact* pc) {
assert(pc);
char name[NAME_MAX] = { 0 };
printf("请输入要修改的联系人的姓名:>");
scanf("%s", name);
int pos = FindByName(pc, name);
if (pos == -1) {
printf("该联系人不存在!\n");
return;
}
printf("请输入姓名:>");
scanf("%s", pc->data[pos].name);
printf("请输入年龄:>");
scanf("%d", &(pc->data[pos].age));
printf("请输入性别:>");
scanf("%s", pc->data[pos].sex);
printf("请输入住址:>");
scanf("%s", pc->data[pos].addr);
printf("请输入电话:>");
scanf("%s", pc->data[pos].tele);
printf("修改成功!\n");
}
int cmp_name(const void* e1, const void* e2) {
return strcmp(((PeoInfo*)e1)->name, ((PeoInfo*)e2)->name);
}
//通讯录排序
void SortContact(Contact* pc) {
assert(pc);
qsort(pc, pc->sz, sizeof(pc->data[0]),cmp_name);
printf("已将通讯录按名字进行排序!\n");
}
//保存通讯录信息到文件里
void SaveContact(Contact* pc) {
FILE* pf = fopen("contact.txt", "wb");
if (pf == NULL) {
perror("SaveContact");
return;
}
else {
int i = 0;
for (i = 0; i < pc->sz; i++) {
fwrite(pc->data + i,sizeof(PeoInfo),1,pf);
}
fclose(pf);
pf = NULL;
printf("保存数据成功!\n");
}
}
//加载文件信息到通讯录
void LoadContact(Contact* pc) {
FILE* pf = fopen("contact.txt", "rb");
if (pf == NULL) {
perror("LoadContact");
return;
}
else {
PeoInfo tmp = { 0 };
int i = 0;
while (fread(&tmp, sizeof(PeoInfo), 1, pf)) {
check_capacity(pc);
pc->data[i] = tmp;
pc->sz++;
i++;
}
fclose(pf);
pf = NULL;
}
}
#include"Contact.h"
//test.c
void menu() {
printf("*****************************\n");
printf("*****1. 新建联系人 ******\n");
printf("*****2. 删除联系人 ******\n");
printf("*****3. 查找联系人 ******\n");
printf("*****4. 修改联系人信息 ******\n");
printf("*****5. 联系人列表 ******\n");
printf("*****6. 联系人列表排序 ******\n");
printf("*****0. 退出通讯录 ******\n");
printf("*****************************\n");
}
enum Option {
EXIT,
ADD,
DEL,
SEARCH,
MODIFY,
SHOW,
SORT
};
int main() {
int input=0;
//创建通讯录
Contact con;
//初始化通讯录
InitContact(&con);
do {
menu();
printf("请选择:>");
scanf("%d", &input);
switch (input) {
case ADD:
AddContact(&con);
break;
case DEL:
DelContact(&con);
break;
case SEARCH:
SearchContact(&con);
break;
case MODIFY:
ModifyContact(&con);
break;
case SHOW:
ShowContact(&con);
break;
case SORT:
SortContact(&con);
break;
case EXIT:
SaveContact(&con);
DestroyContact(&con);
printf("退出通讯录!\n");
break;
default:
printf("选择错误!请重新选择!\n");
break;
}
} while (input);
}
如有错误,还请指正