单链表实现

题目

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

向链表头插入一个数； 删除第 $k$ 个插入的数后面的数； 在第 $k$ 个插入的数后插入一个数。 现在要对该链表进行 $M$ 次操作，进行完所有操作后，从头到尾输出整个链表。

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

输入格式 第一行包含整数 $M$ ，表示操作次数。

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

H x，表示向链表头插入一个数 $x$。 D k，表示删除第 $k$ 个插入的数后面的数（当 $k$ 为 $0$ 时，表示删除头结点）。 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

思路

head 相当于一个头指针 对于某次插入

e[idx] = x;  // idx为当前可用的索引, 存储x值
ne[idx] = ne[k];  // 将存储x的位置指向k的下一个位置
ne[k] = idx;  // 将k指向的位置变更为idx位置
idx ++ ;  // idx + 1

对于插入头节点的特殊情况, ne[k]就等于head

对于某次删除, 可以直接将k指向的位置A更新为下一个位置A指向的位置B 即 ne[k] = ne[ne[k]]

代码

#include <iostream>

using namespace std;

const int N = 100010;
int head, idx, e[N], ne[N];

void init()
{
    head = -1;
    idx = 0;
}

void add_head(int x)
{
    e[idx] = x, ne[idx] = head, head = idx ++ ;
}

void add(int k, int x)
{
    e[idx] = x, ne[idx] = ne[k], ne[k] = idx ++ ;
}


void del_k(int k)
{
    ne[k] = ne[ne[k]];
}


int main()
{
    init();
    int n;
    cin >> n;
    while (n -- )
    {
        int k, x;
        char c;
        cin >> c;
        if (c == 'H')
        {
            cin >> x;
            add_head(x);
        }
        else if (c == 'D')
        {
            cin >> k;
            if (!k) head = ne[head];
            else del_k(k - 1);
        }
        else
        {
            cin >> k >> x;
            add(k - 1, x);
        }
    }
    for (int i = head; i != -1; i = ne[i]) cout << e[i] << " ";
    
    return 0;
}