AcWing 826. 单链表

题目来源：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\le M\le 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

二、单链表

单链表分为静态单链表和动态单链表，但是动态单链表实现方式在每次创建一个新结点时都需要使用 $new()$ 函数，然而 $new()$ 函数非常耗时！！因此，我们在做算法题的时候，都需要使用数组模拟的静态单链表。
静态单链表在算法题实际应用中非常广泛，最常见的就是由多个单链表构成的邻接表。邻接表用来存储树和图。
静态双链表，我们会在后续介绍，它的用途主要是可以对某些题目进行优化。

单链表的一般形式：

其中，每一个结点 $i$，都包含数据域 $e_i$ 和指针域 $n{e}_i$。

一般而言，定义一个单链表，需要定义头结点 $head$，数据域 $e[N]$，指针域 $ne[N]$，为下一次插入操作准备的空白结点的下标 $idx$。

下面通过一个图来表示上面几个数据的关系：

1. 数据定义

#include <iostream>
using namespace std;

const int N = 1e5 + 10;

// head  表示头结点的下标
// e[i]  表示结点i的数据域
// ne[i] 表示结点i的next指针
// idx   表示当前可用的空白结点指针
int head, e[N], ne[N], idx;

2. 初始化

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

3. 插入操作

3.1 头插法

// 头插法, x是数据域
void add_to_head(int x)
{
	e[idx] = x, ne[idx] = head, head = idx, idx++;
}

3.2 插入到指定下标结点的后面

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

4. 删除操作

// 将下标是k的结点的下一个结点删除
void remove(int k)
{
	ne[k] = ne[ne[k]];
}

5. 遍历操作

// 注意：这里的~i等价于i != -1
for (int i = head; ~i; i = ne[i]) printf("%d ", e[i]);

三、代码

#include <iostream>
using namespace std;

const int N = 1e5 + 10;
int head, e[N], ne[N], idx;
int n;

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

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

// 在下标为k的结点后面插入一个结点
void add(int k, int x)
{
    e[idx] = x, ne[idx] = ne[k], ne[k] = idx, idx++;
}

// 在下标为k的结点后面删除一个结点
void remove(int k)
{
    ne[k] = ne[ne[k]];
}

int main()
{
    init();
    scanf("%d", &n);
    
    while (n--)
    {
        char op[2];
        scanf("%s", op);
        
        if (*op == 'H')
        {
            int x;
            scanf("%d", &x);
            add_to_head(x);
        }
        else if (*op == 'D') 
        {
            int k;
            scanf("%d", &k);
            
            // 插入的第k个结点对应的下标为k-1
            // 注意题目中说k=0代表删除头节点
            if (!k) head = ne[head];
            else remove(k - 1);
        }
        else
        {
            int k, x;
            scanf("%d%d", &k, &x);
            add(k - 1, x);
        }
        
    }
    
    for (int i = head; ~i; i = ne[i]) printf("%d ", e[i]);
    puts("");
    
    return 0;
}