sdutoj-3781-数组操作

Description
给你一个长度为 n 的数组，并给出如下几种操作：

在下标为 a 的位置插入一个整数 b，如果其后有元素，则全部后移。例如，数组为 1, 2, 3，在下标为 1 的位置插入 4，则数组变为：1, 4, 2, 3。
删除下标为 a 的元素，如果其后有元素，则全部前移。例如，数组为 1, 2, 3，删除下标为 0 的元素，则数组变为：2, 3。
修改下标为 a 的元素的值为 b。
输出整个数组。
Input
输入数据有多组（数据组数不超过 10），到 EOF 结束。

对于每组数据：

首先输入一行 n, m (1 <= n <= 10^6, 1 <= m <= 10^4)，分别表示初始数组的长度和操作次数。
接下来一行输入 n 个用空格隔开的整数 Ai (0 <= Ai <= 1000)，表示初始的数组。
最后有 m 行，每行对应一次操作，根据操作类型分为不同的输入格式（具体含义参见描述部分，0 <= a, b <= 1000）：
1 a b
2 a
3 a b
4
Output
对于每组数据中的每次类型为 4 的操作，输出一行，表示此时的数组，每个整数之间用空格隔开。

Sample
Input
3 6
1 2 3
1 1 4
4
2 2
4
3 2 8
4
Output
1 4 2 3
1 4 3
1 4 8
Hint
建议使用链表完成。

#include<bits/stdc++.h>

using namespace std;

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

List *creat(int n)
{
    List *head, *p, *tail;
    head = new List;
    head->next = NULL;
    tail = head;
    for(int i = 0; i < n; i++)
    {
        p = new List;
        scanf("%d", &p->data);
        p->next = NULL;
        tail->next = p;
        tail = p;
    }
    return head;
}
void List_in(List *head, int a, int b)
{
    List *q, *p;
    q = new List;
    q->data = b;
    p = head;
    while(a--)//如果在下标为0的地方插入，那么就不执行这条
        p = p->next;
    q->next = p->next;
    p->next = q;
}

void List_del(List *head, int a)//删除操作，这个是链表的基本操作不难
{
    List *p, *q;
    p = head;//这里要注意，因为要删除一个节点，都是要找到他前个一个节点3
    while(a--)
        p = p->next;
    q = p->next;
    p->next = q->next;
}
void List_re(List *head, int a, int b)
{
    List *p;
    p = head->next;
    while(a--)//这个地方很巧妙
        p = p->next;
    p->data = b;
}
void display(List *head)
{
    List *p;
    p = head->next;
    while(p)
    {
        printf("%d%c", p->data, p->next? ' ':'\n');
        p = p->next;
    }
}
int main()
{
    int a, b, num, n, m;
    List *head;
    while(scanf("%d %d", &n, &m) != EOF)
    {
        head = creat(n);
        while(m--)
        {
            scanf("%d", &num);
            if(num == 1)
            {
                scanf("%d%d", &a, &b);
                List_in(head, a, b);
            }
            else if(num == 2)
            {
                scanf("%d", &a);
                List_del(head, a);
            }
            else if(num == 3)
            {
                scanf("%d %d", &a, &b);
                List_re(head, a, b);
            }
            else if(num == 4)
            {
                display(head);
            }
        }
    }
    return 0;
}