【ACWing】826. 单链表

题目地址：

https://www.acwing.com/problem/content/828/

实现一个单链表，初始为空，并支持三种操作：
(1) 向链表头插入一个数；
(2) 删除第$k$个插入的数后面的数；
(3) 在第$k$个插入的数后插入一个数。
对其进行$M$次操作，从头到尾输出整个链表。注意，这里的第$k$个插入的数并不是当前链表的第$k$个数，而是所有的插入操作中，第$k$个插入的那个数。

输入格式：
第一行包含整数$M$，表示操作次数。接下来$M$行，每行包含一个操作命令，操作命令可能为以下几种：
(1) “H x”，表示向链表头插入一个数$x$。
(2) “D k”，表示删除第$k$个插入的数后面的数（当$k$为$0$时，表示删除头结点）。
(3) “I k x”，表示在第$k$个插入的数后面插入一个数$x$（此操作中$k$均大于$0$）。

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

数据范围：
$1\le M\le 100000$

用数组模拟单链表。开两个数组，$e[i]$表示第$i$个用到的位置的值，$ne[i]$表示第$i$个位置的节点的下一个节点的下标。另外用两个变量，一个表示头结点下标，另一个存如果再插入数的话，它所占的位置的下标。上面的$i$都是从$0$开始计数的。接下来无论哪个操作，都按照动态链表的方式来类似做即可。代码如下：

#include <iostream>
using namespace std;

const int N = 100010;

// head第一条边的下标，
// e[i]表示边i指向的节点编号，
// ne[i]表示边i的下一个边的下标，
// idx表示下一次分配边可以用的下标
int head, e[N], ne[N], idx;

// 初始化一个空链表，头结点是-1表示是空链表，idx是0，
// 表示第一个插入的节点要用到0这个位置
void init() {
    head = -1;
    idx = 0;
}

// 将x插到开头
void add_to_head(int x) {
    e[idx] = x; 
    ne[idx] = head; 
    head = idx;
    idx++;
}

// 将x插到下标是k的点后面
void add(int k, int x) {
    e[idx] = x;
    ne[idx] = ne[k];
    ne[k] = idx; 
    idx++;
}

// 删掉下标是k的点后面的那个点
void remove(int k) {
	// 这里要特判，如果k = -1，表示的是要删头结点
    if (k == -1) head = ne[head];
    else ne[k] = ne[ne[k]];
}

int main() {
    int n;
    cin >> n;

    init();
    while (n--) {
        char op;
        int k, x;

        cin >> op;
        if (op == 'H') {
            cin >> x;
            add_to_head(x);
        } else if (op == 'I') {
            cin >> k >> x;
            add(k - 1, x);
        } else if (op == 'D') {
            cin >> k;
            remove(k - 1);
        }
    }

    while (head != -1) {
        cout << e[head] << ' ';
        head = ne[head];
    }

    return 0;
}

所有操作时间复杂度$O(1)$，空间是插入的执行次数。