第二讲 数据结构

单链表

826. 单链表 - Acwing题库

数据结构：

• e[N]：用于存储节点的值的数组。
• ne[N]：作为"下一个"指针的数组，用于连接节点。
• head：指向链表头部的索引。
• idx：当前可用的下一个索引。

初始化：

• init() 函数将 head 设置为 -1（表示链表为空），idx 设置为 0。

添加元素：

• add_to_head(int x)：在链表头部插入一个值为 x 的新节点。
• add(int k, int x)：在索引为 k 的节点后面插入一个值为 x 的新节点。

删除元素：

remove(int k)：删除索引为 k 的节点后面的节点。

主功能：

程序读取操作的数量 m，然后处理每个操作：

• H x：将 x 添加到链表头部。
• D k：删除第 k 个节点后面的节点（如果 k 为 0，则删除头节点）。
• I k x：在第 k 个节点后面插入值为 x 的节点。
• 最后，打印链表中剩余的元素。

#include <iostream>

using namespace std;

const int N = 100010;


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

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

// 将x插到头结点
void add_to_head(int x)
{
    e[idx] = x, ne[idx] = head, head = idx ++ ;
}

// 将x插到下标是k的点后面
void add(int k, int x)
{
    e[idx] = x, ne[idx] = ne[k], ne[k] = idx ++ ;
}

// 将下标是k的点后面的点删掉
void remove(int k)
{
    ne[k] = ne[ne[k]];
}

int main()
{
    int m;
    cin >> m;

    init();

    while (m -- )
    {
        int k, x;
        char op;

        cin >> op;
        if (op == 'H')
        {
            cin >> x;
            add_to_head(x);
        }
        else if (op == 'D')
        {
            cin >> k;
            if (!k) head = ne[head];
            else remove(k - 1);
        }
        else
        {
            cin >> k >> x;
            add(k - 1, x);
        }
    }

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

    return 0;
}