单链表的插入

插入运算是将值为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;
}