PTA-7-3 单链表按位插入 (10 分)

7-3 单链表按位插入 (10 分)

有一字符型单链表List，假设表中无重复值，现要在指定位置i执行插入x操作。插入成功时输出新表，否者输出“location error”。
输入格式:

有多组数据，每组数据占两行，代表一次插入操作。 每组第一行有三个数，第一个为表长n（0<n<50），第二个为位置i，第三个为值x； 第二行为单链表的各元素值。
输出格式:

每组数据的输出占一行，输出如题。
输入样例:

在这里给出一组输入。例如：

5 4 f
abcde
结尾无空行

输出样例:

在这里给出相应的输出。例如：

abcfde
结尾无空行

#include<stdio.h>
#include<malloc.h>
#include<stdlib.h>
typedef struct st {
	char data;
	struct st* next;
}NODE,*PNODE;
//创建一个单链表 
PNODE creat_list(int length)
{
	PNODE pHead=(PNODE)malloc(sizeof(NODE)*1);
	pHead->next=NULL;
    if(NULL==pHead)
    {
        printf("动态内存分配失败！程序终止\n");
        exit(-1);
    }
    char val; 
    PNODE pTail=pHead;
    pTail->next=NULL;
   for(int j=0;j<length+1;j++)
   {
       PNODE pNew=(PNODE)malloc(sizeof(NODE)*1);
       if(NULL==pNew)
    {
        printf("动态内存分配失败！程序终止\n");
        exit(-1);
    }
    scanf("%c",&val);
       pNew->data=val;
       pTail->next=pNew;
	   pNew->next=NULL;
	   pTail=pNew; 
   }
   return pHead;
}
//按位插入pos表示在第几个位置前插入，val表示要插入的内容
void insert_list(PNODE pHead,int pos,char val )
{
	int i=0;
	PNODE p=pHead;
	while(NULL!=p&&i<pos-1)
	{
		p=p->next;
		i++;
	}
	if(i>pos-1||NULL==p)
	printf("location error");
	PNODE pNew=(PNODE)malloc(sizeof(NODE));
	if(NULL==pNew)
	{
		printf("动态内存分配失败！\n");
	    exit(-1);
	}
	pNew->data=val;
	PNODE q=p->next;
	p->next=pNew;
	pNew->next=q;

}
//遍历输出 
void traverse_list(PNODE pHead)
{
   PNODE p=NULL;
   p=pHead->next;
   while(p)
    {
	    printf("%c ",p->data);
    	p=p->next;
    }	
    printf("\n");
    return;
}
int main(void)
{
	PNODE pHead=NULL;//初始化头指针 
	int len,i;
	char x;//存储找到那个字符的值
	while(~scanf("%d %d %c",&len,&i,&x)){
        getchar();
		pHead=creat_list(len);
		insert_list(pHead,i,x);
		traverse_list(pHead);
		free(pHead);
		} 
	return 0;
 }