单链表的插入

插入运算是将值为x的新节点，插入到单链表的第i个节点的位置上，即插入到节点ai-1 与 ai之间，具体算法如下：
（1）找到节点ai－1存储位置p；
（2）生成一个数据域为x的新节点*s；
（3）新节点*s的指针域指向节点ai；
（4）令节点*p的指针域指向新节点*s；

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

typedef struct node
{
    int data;
    struct node *next;
} NODE;


// 尾插法创建单链表(带头节点)
NODE *createEnd(int arr[], int len)
{
    // 有了头节点，头指针指向头节点，无论链表是否为空，头指针总是非空；
    NODE *head = (NODE *)malloc(sizeof(NODE)); // 生成头节点
    head->next = NULL;
    NODE *end = head;  // 尾指针初始化

    for (int i = 0; i < len; i++) {

        NODE *p = (NODE *)malloc(sizeof(NODE)); // 为每个数组元素建立一个节点
        p->data = arr[i];
        end->next = p;  // 将节点p插入到终端节点之后
        end = p;
    }

    end->next = NULL;  // 单链表建立完毕，将终端节点的指针域置空

    return head;
}

// 在单链表的第pos位置处插入一个节点，数据为x
NODE *insert(NODE *head, int pos, int x)
{
    NODE *p = head;

    int j = 1; // 从链表的第一个节点(即头节点后的节点)位置开始，考虑插入，防止插入到头节点之前；
    while (p && j < pos) { // 当pos特别大，远超链表长度时，有p为空的情况，跳出循环；当p不为空，但循环到j>＝pos时，跳出循环；
        p = p->next;
        ++j;
    }

    if (!p || j > pos) { // 经过循环，若最终p为空，或最初pos为0或负数时，都说明插入位置不合理；
        printf("Position Error\n");
        return 0;
    }

    NODE *s = (NODE *)malloc(sizeof(NODE)); // 新节点
    s->data = x;


    s->next = p->next;
    p->next = s;

    return head;
}


// 单链表打印
void print(NODE *head)
{
    if (head == NULL) return;

    NODE *p = head->next;
    while (p != NULL) {
        printf("%d\n", p->data);
        p = p->next;
    }
}

int main(void)
{
    int arr[] = {1,2,3,4,5,6,7};
    int len = sizeof(arr)/sizeof(int);

    NODE *head = createEnd(arr, len);

    // 插入前
    print(head);


    insert(head, 5, 999);
    // 插入后
    print(head); // 若插入失败，则打印出的仍是原来的表；

    return 0;
}