【AcWing】826. 单链表

原题链接：活动 - AcWing

目录

1. 题目描述

输入格式

输出格式

数据范围

输入样例：

输出样例：

2. 思路分析

3. 代码实现

---

1. 题目描述

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

1. 向链表头插入一个数；
2. 删除第 k 个插入的数后面的一个数；
3. 在第 k 个插入的数后插入一个数。

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

注意:题目中第 k 个插入的数并不是指当前链表的第 k 个数。例如操作过程中一共插入了 n 个数，则按照插入的时间顺序，这 n 个数依次为：第 11 个插入的数，第 22 个插入的数，…第 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

2. 思路分析

用数组模拟实现单链表。

详见如下注释，需要结合画图分析。

主要注意第一个插的结点下标是0，第二插的结点下标是1......所以第k个插的结点下标是k-1。

3. 代码实现

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define endl '\n'
const int N = 100010;
int head, e[N], ne[N], id;
//head 表示头结点的下标
//e[i] 表示结点i的值
//ne[i] 表示结点i的指针是多少
//id    存储当前已经用到了哪个点

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

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

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

//将下标是k的点的后面的点删掉
void remove(int k) {
    ne[k] = ne[ne[k]];
}

signed main() {
    ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int m; cin >> m;
    init();
    while (m--) {
        char op; cin >> op;
        if (op == 'H') {
            int x; cin >> x;
            add_to_head(x);
        }else if (op == 'D') {
            int k; cin >> k;
            if (!k) head = ne[head]; //k=0,删除头结点后面一个元素
            remove(k - 1);
        }else if (op == 'I') {
            int k, x; cin >> k >> x;
            add(k - 1, x);
        }
    }
    for (int i = head; i != -1; i = ne[i]) cout << e[i] << " ";
    return 0;
}