最近在学习《C Primer Plus》第六版第十七章时,发现书中的film2.c程序在执行过程中出现了一些问题,先附上书中的代码:
#include<stdio.h>
#include<stdlib.h>//包含malloc函数和free函数的头文件
#include<string.h>//包含strcpy函数和strchr函数的头文件
#define SIZE 40
struct film { //储存一部电影信息的结构
char title[SIZE];//电影名
int rating; //电影评级
struct film * next;//指向下一结构的指针
};
char * s_gets(char * st, int n);
int main(void)
{
struct film * current, * prev;
struct film * head = NULL;//头指针
char input[SIZE];
puts("请输入电影名称");
while(s_gets(input, SIZE) != NULL && input[0] != '\0')
{
current = (struct film *) malloc(sizeof(struct film));
if(head == NULL)
head = current;
else
prev -> next = current;
current -> next = NULL;
strcpy(current -> title, input);
puts("请输入该电影的评级(0 - 10)");
scanf("%d", ¤t -> rating);
while(getchar() != '\n')
continue;
puts("请输入下一电影名称(空行结束输入)");
prev = current;
}
if(head == NULL)
printf("没有电影数据\n");
else
printf("电影目录如下:\n");
current = head;
while(current != NULL)
{
printf("电影名称:%s\t 电影评分:%d\n", current -> title, current -> rating);
current = current -> next;
}
current = head;
while(current != NULL)
{
current = head;
head = current -> next;
free(current);
}
printf("Bye\n");
return 0;
}
char * s_gets(char * st, int n)
{
char * ret_val;
char * find;
ret_val = fgets(st, n, stdin);
if(ret_val)
{
find = strchr(st, '\n');
if(find)
*find = '\0';
else
while(getchar() != '\n')
continue;
}
return ret_val;
}
然后程序运行结果如下:
程序并没有打印“Bye”,说明没有正常终止,检查程序后发现原来是释放链表节点时出现了错误:
current = head;//current指向第一个结构
while(current != NULL)
{
current = head;//当current指向最后一个结构时
head = current -> next;//head此时是空指针NULL
free(current);//释放最后一个结构分配的内存
}
/*但是free()释放完最后一个结构之后,循环仍然继续,(current = head) == NULL,此时语句free(current)等效于free(NULL),造成程序无法正常终止,出现错误*/
原因找到了,那么就是修改程序了,可以将上面的这段代码改成以下两种形式:
//第一种方法:
current = head;
while(head != NULL)//head == NULL,程序退出,不会出现free(NULL)这种情况
{
current = head;//当current指向最后一个结构时
head = current -> next;//head此时是空指针NULL
free(current);//释放最后一个结构分配的内存
}
//第二种方法
current = head;
while(current != NULL)
{
head = current -> next;//当current指向倒数第二个结构时,head此时是指向最后一个结构
free(current);//释放倒数第二个结构分配的内存
current = head;//current指向最后一个结构
}
/*程序继续循环一次,head变成空指针NULL,free(current)释放最后一个结构分配的内存,(current = head) == NULL,回到循环头,不满足条件,退出循环*/