struct xinxi *duqu2() { struct xinxi *head2= NULL; struct xinxi *temp = NULL; char dianhua[20], mima[20],chepaihao[20]; FILE *fp = fopen("用户信息.txt", "ab+"); if (fp == NULL) { printf("\t\t\t\t打开文件失败!文件名: 用户信息.txt\n"); return NULL; } else { while (fscanf(fp, "%s %s %s", dianhua, mima,chepaihao) != EOF) { struct xinxi *p = (struct xinxi *)malloc(sizeof(struct xinxi)); if (p == NULL) { printf("分配内存失败!\n"); } strcpy(p->dianhua, dianhua); strcpy(p->mima, mima); strcpy(p->chepaihao,chepaihao); if (head2 == NULL) { head2 = temp = p; } else { temp->next = p; temp = p; } } fclose(fp); if (temp != NULL) { temp->next = NULL; } return head2; } }优化
时间: 2024-01-19 17:04:47 浏览: 60
可以对这段代码进行如下优化:
1. 增加错误处理逻辑,当文件打开失败或读取数据失败时,应该及时关闭文件并释放内存。
2. 使用fgets函数读取文件内容,避免使用%s格式符造成的数据读取不正确的问题。
3. 在读取文件内容时,对读取到的数据进行有效性检查,避免出现异常情况。
4. 在动态分配内存时,对内存分配结果进行有效性检查,避免分配失败导致的程序异常。
5. 在链表的构建过程中,增加错误处理逻辑,避免链表出现异常。
优化后的代码如下:
```
struct xinxi *duqu2() {
struct xinxi *head2 = NULL;
struct xinxi *temp = NULL;
char line[64];
char dianhua[20], mima[20], chepaihao[20];
FILE *fp = fopen("用户信息.txt", "r");
if (fp == NULL) {
printf("打开文件失败!文件名: 用户信息.txt\n");
return NULL;
}
while (fgets(line, sizeof(line), fp) != NULL) {
if (sscanf(line, "%s %s %s", dianhua, mima, chepaihao) != 3) {
printf("读取文件内容失败!\n");
fclose(fp);
return NULL;
}
struct xinxi *p = (struct xinxi *)malloc(sizeof(struct xinxi));
if (p == NULL) {
printf("分配内存失败!\n");
fclose(fp);
return NULL;
}
strcpy(p->dianhua, dianhua);
strcpy(p->mima, mima);
strcpy(p->chepaihao, chepaihao);
if (head2 == NULL) {
head2 = temp = p;
} else {
temp->next = p;
temp = p;
}
}
fclose(fp);
if (temp != NULL) {
temp->next = NULL;
}
return head2;
}
```
阅读全文