PTA链表 删除单链表的第i个结点

本题要求实现一个函数，删除单链表的第i个结点。

函数接口定义：

Node *deletelink(Node *head, int i)

在这里，head是单链表的头指针，i是待删除的结点编号。函数不需要处理单链表为空的情况。如果删除位置错误，输出"error"。

裁判测试程序样例：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

typedef struct ListNode {
    int num;
    struct ListNode *next;
}Node;

Node *createlist(); /*根据用户输入的整数n，创建具有n个结点的单链表。裁判实现，细节不表*/
Node *deletelink(Node *head, int i);
void display(Node *head);/*输出链表结点的数据域。裁判实现，细节不表*/

int main(void)
{
    Node  *head;
     int i;
    head = createlist();
     scanf("%d",&i);
    head = deletelink(head, i);
    display(head);
    return 0;
}

/* 请在这里填写答案 */

输入样例1：

5
10 5 4 8 7
2

输出样例1：

10 4 8 7

输入样例2：

5
10 5 4 8 7
6

输出样例2：

error
10 5 4 8 7

Node *deletelink(Node *head ,int i){
    if(i<1){
        printf("error\n");
        return head;
    }
    Node *a=NULL,*b=head;
    for(int c=1;b&&c<i;c++){
        a=b,b=b->next;
    }
    if(b==NULL){
        printf("error\n");
        return head;
    }
    if(a==NULL) head=b->next;
    else a->next=b->next;
    return head;
}