AcWing 826. 单链表

题目描述

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

向链表头插入一个数；
删除第 $k$ 个插入的数后面的数；
在第 $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

算法

用数组模拟单链表

C++ 代码

/*
    用数组模拟单链表
*/
#include <iostream>

using namespace std;

const int N = 1e5 + 10;

//head 表示头结点的下标
//e[i] 表示节点i的值
//ne[i] 表示节点i的next指针是多少
//idx 表示当前已经用到数组的哪个下标了

int e[N], ne[N];

int n, head, idx;

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

//向链表头插入数x
void insert_to_head(int x) {
    
    e[idx] = x;
    
    ne[idx] = head;
    
    head = idx;
    
    idx ++;
    
}

//删除第 k 个插入的数后面的数；
void remove(int k) {
    
    if(k == -1) head = ne[head];
    
    else ne[k] = ne[ne[k]];
}

//在第 k 个插入的数后插入一个数x
void add(int k, int x) {
    
    e[idx] = x;
    
    ne[idx] = ne[k];
    
    ne[k] = idx;
    
    idx ++;
    
}

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

原博文链接