AcWing 826. 单链表

题目

实现一个单链表，链表初始为空，支持三种操作：

(1) 向链表头插入一个数；

(2) 删除第k个插入的数后面的数；

(3) 在第k个插入的数后插入一个数

现在要对该链表进行M次操作，进行完所有操作后，从头到尾输出整个链表。

注意:题目中第k个插入的数并不是指当前链表的第k个数。例如操作过程中一共插入了n个数，则按照插入的时间顺序，这n个数依次为：第1个插入的数，第2个插入的数，…第n个插入的数。

输入格式：

第一行包含整数M，表示操作次数。

接下来M行，每行包含一个操作命令，操作命令可能为以下几种：

(1) “H x”，表示向链表头插入一个数x。

(2) “D k”，表示删除第k个插入的数后面的数（当k为0时，表示删除头结点）。

(3) “I k x”，表示在第k个插入的数后面插入一个数x（此操作中k均大于0）。

输出格式：

共一行，将整个链表从头到尾输出。

数据范围

1≤M≤100000
所有操作保证合法。

输入样例：

10
H 9
I 1 1
D 1
D 0
H 6
I 3 6
I 4 5
I 4 5
I 3 4
D 6

输出样例：

6 4 6 5

思路分析：

插入可达到 O ( 1 ) O(1) O(1)的复杂度，注意插入和删除的过程是a - 1，（插入第一个点的下标为0则插入第k个点的下标为k-1，即插入k-1后面的一个结点，要注意在删除的过程要特判删除头结点if(a == 0) head = ne[head]）。

代码：

#include <iostream>

using namespace std;

const int maxn = 100010;

//头结点下标，当前节点下标，当前结点的值，next的结点的下标
int head, idx, e[maxn], ne[maxn];

//初始化
void init(){
    head = -1;
    idx = 0;
}

//在链表头插入一个数
void insert2head(int x){
    e[idx] = x, ne[idx] = head, head = idx++;
}
//在第k个数后面插入一个数
void insert2k(int k, int x){
    e[idx] = x, ne[idx] = ne[k], ne[k] = idx++;
}
//将头结点删除需要保证头结点的存在
void removek(int k){
    ne[k] = ne[ne[k]];
}

int main(){
    int m, a, b;
    cin >> m;
    init();
    while(m--){
        char op;
        cin >> op;
        if(op == 'H'){
            cin >> a;
            insert2head(a);
        }else if(op == 'D'){
            cin >> a;
            if(a == 0) head = ne[head];
            removek(a - 1);
        }else if(op == 'I'){
            cin >> a >> b;
            insert2k(a - 1, b);
        }
    }
    for(int i = head; i != -1; i = ne[i]) cout << e[i] << " ";
    cout << endl;
    return 0;
}

AcWing题解
题目链接