AcWing 826. 单链表（C++算法）

AcWing 826. 单链表

1、题目（来源于AcWing）：
实现一个单链表，链表初始为空，支持三种操作：

(1) 向链表头插入一个数；

(2) 删除第k个插入的数后面的数；

(3) 在第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≤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、基本思想：
①将一个数x插到现有头结点之前
②将一个数x插到下标为k的数的后面一位
③删除一个节点

3、步骤：
①写出基本思想中需要的三个函数
②写主函数，idx记录节点下标（从0开始）

4、C++代码如下（该代码引用AcWing网站的代码）：

#include <iostream>

using namespace std;

void init();
void add_to_head(int x);
void remove_(int k);
void add(int k, int x);

const int N = 100010;

int head, idx, e[N], ne[N];//head表示头结点的下标，idx记录用到第几个点（从0开始），e[idx]表示这个点的值，ne[idx]表示这个点后一个点的下标

int main()
{
    int m;
    cin >> m;
    
    init();//一定要初始化
    
    while (m -- )
    {
        char op;
        int x, k;
        
        cin >> op;
        
        if (op == 'H')
        {
            cin >> x;
            add_to_head(x);
        }
        else if (op == 'D')
        {
            cin >> k;
            if (k == 0) head = ne[head];//题中“当k为0时，表示删除头结点”
            else remove_(k - 1);
        }
        else
        {
            cin >> k >> x;
            add(k - 1, x);
        }
    }
    
    for (int i = head; i != -1; i = ne[i]) cout << e[i] << ' ';
    
    return 0;
}

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

void add_to_head(int x)//将一个数x插入头结点，原头结点的数后移
{
    e[idx] = x;
    ne[idx] = head;
    head = idx ++ ;//其实是先让head表示新的头结点的下标idx,idx再计数
}

void remove_(int q)//删掉下标为q的数的后面那个数
{
    ne[q] = ne[ne[q]];
}

void add(int q, int x)//将一个数x插入下标为q的数的后面一位
{
    e[idx] = x;
    ne[idx] = ne[q];
    ne[q] = idx ++ ;
}//该代码引用AcWing网站的代码

注意事项：
①一定要事先初始化
②注意题中k的含义，在调用函数时用k - 1
③注意k = 0时是删除头结点，此时不管下标
④head表示头结点下标，并不存储具体的值，不能当做节点
⑤scanf读入字符时不会忽略空格和回车，但cin默认会忽略空格和回车，所以本题用cin更好。
⑥在单链表中找点只能找某一个点的右侧（后面）的点，不能找左侧（前面）的点。