AcWing 826. 单链表

101 阅读3分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第14天,点击查看活动详情

题目

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

  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

思路

void init(){ //初始化
    head = -1; 
    //开始的时候,链表头节点指向 -1,指向空结点,在链表中进入元素后,指向第一个元素,当它再次指向空结点的时候,表示链表中没有内容
    idx = 0; 
    //链表的下标,方便查、插、删的操作
}

void add_to_head(int x){ //将x插到头结点
    e[idx] = x,ne[idx] = head,head = idx++;
    //先存值
    //令元素 x 的指针指向了head原本指向的
    //head 指向当前的第一个元素
    //指针后移,为下一次插入元素做准备
}

4329_65d76184b7-捕获1.png

Snipaste_2022-07-09_09-50-11.png

void add(int k,int x){ //将x插到下标是k的点后面
    e[idx] = x,ne[idx] = ne[k],ne[k] = idx++;
    //存值
    //让 x 的指针 指向它所占位置的后一个元素
    //让原来的指针,指向自己
    //idx后移
}

Snipaste_2022-07-09_09-55-29.png

Snipaste_2022-07-09_09-57-52.png

4329_1c111542b7-捕获2.png

void remove(int k){ //将下标是k的点后面的点删掉
    ne[k] = ne[ne[k]]; //让 k 的指针 指向 它所指向的 值的指针 指向的 值
}

Snipaste_2022-07-09_09-58-50.png

4329_d6c7a5c2b7-捕获4.png

单链表模板

// head存储链表头,e[]存储节点的值,ne[]存储节点的next指针,idx表示当前用到了哪个节点
int head, e[N], ne[N], idx;

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

// 在链表头插入一个数a
void insert(int a)
{
    e[idx] = a, ne[idx] = head, head = idx ++ ;
}

// 将头结点删除,需要保证头结点存在
void remove()
{
    head = ne[head];
}

ac代码

#include<iostream>
using namespace std;
const int N =  100010;
int head,e[N],ne[N],idx;
// head 表示头结点的下标
// e[i] 表示节点i的值
// ne[i] 表示节点i的next指针是多少
// idx 存储当前已经用到了哪个点
void init(){ //初始化
    head = -1;
    idx = 0;
}
void add_to_head(int x){ //将x插到头结点
    e[idx] = x,ne[idx] = head,head = idx++;
}
void add(int k,int x){ //将x插到下标是k的点后面
    e[idx] = x,ne[idx] = ne[k],ne[k] = idx++;
}
void remove(int k){ //将下标是k的点后面的点删掉
    ne[k] = ne[ne[k]];
}
int main(){
    int n;
    cin >> n;
    init();
    while (n -- ){
        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;
}