AcWing 826. 单链表

题目

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

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

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

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

输入格式

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

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

1. H x，表示向链表头插入一个数 。
2. D k，表示删除第 k个插入的数后面的数（当 k为 00 时，表示删除头结点）。
3. I k x，表示在第 k个插入的数后面插入一个数 x（此操作中 k均大于 00）。

输出格式

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

数据范围

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

题解

使用两个数组e[]和ne[]来表示链表，因为效率高于链接的数据结构

#include
using namespace std;
const int N = 100001;

//head表示头指针的下标
//e[i]表示节点i的值
//ne[i]表示节点i的next指针指向的节点位置
//idx表示当前用到哪个节点,即记录插入到第几个数字
int head,e[N],ne[N],idx;

//初始化
int init(){
    //头节点
    head = -1;
    idx = 0;
}

//头插
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){
    ne[k] = ne[ne[k]];
}

int main(){
    int M,k,x;
    char choice;
    cin >> M;
    init();
    for (int i = 0; i < M; ++i) {
        cin >> choice;
        if (choice == 'H'){
            cin >> x;
            add_to_head(x);
        }
        else if (choice == 'D'){
            cin >> k;
            if (!k) {
                head = ne[head];
            }
            else {
                remove(k - 1);
            }
        }
        else{
            cin >> k;
            cin >> x;
            add(k-1,x);
        }
    }
    for (int i = head; i != -1; i = ne[i]){
        cout << e[i] << ' ';
    }
    cout << endl;
}