PAT知识储备——单链表

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

向链表头插入一个数；
删除第 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

#include <iostream>

using namespace std;

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

void init(){
    head = -1; //用-1表示单链表的末尾
    idx = 0;
}

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

//可以对照着上面那个头插来看
void add(int k, int x){
    e[idx] = x;//存数据
    ne[idx] = ne[k];//当前处理的结点指向原来第k个结点指向的结点
    ne[k] = idx ++;//当前处理的结点被原来的第k个结点所指向,相当于插在k之后了
                    //++并指向下一个处女结点
}

void remove (int k){
    //将k个结点的next指向下下个结点
    //用让k+1个结点脱离链表的方式来删除
    //会浪费一个数组元素的空间，不过算法题不在乎这些细节
    ne[k] = ne[ne[k]];
}

int main(){
    int m;

    init();

    scanf("%d", &m);
    while (m -- ){
        int k, x;
        char op;
        cin >> op;
        if(op == 'H'){
            cin >> x;
            add_to_head(x);
        }
        if(op == 'D'){
            cin >> k;
            if(!k)
                head = ne[head];
            else
                remove(k - 1);
        }
        if(op == 'I'){
            cin >> k >> x;
            add(k - 1, x);
        }
    }


    for(int i = head; i != -1;i = ne[i]){
        cout << e[i] << ' ';
    }

    return 0;
}