题目:826. 单链表

题目描述

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

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

所有操作保证合法。

时空限制

1s / 64MB

输入样例

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 MaxM = 100000+10;

int M;
int head, idx, e[MaxM], ne[MaxM];
// head表示头节点的下标
// idx存储当前节点的下标
// e[i]表示节点i的值
// ne[i]表示节点i的指针(节点i的下一个节点的下标)

void init(){
    head = -1;
    idx = 0;
}

void insert_to_head(int x){
    e[idx] = x;//将值存储到新节点中
    ne[idx] = head;//将当前节点的指针指向head的原来后面节点的位置
    head = idx;//将head的指针指向当前节点的位置
    idx ++;//将当前节点的指针向后移
}

void insert_to_k(int k, int x){
    e[idx] = x;//将值存储到新节点中
    ne[idx] = ne[k];//将当前节点的指针指向k原来后面节点的位置
    ne[k] = idx;//将k的指针指向当前节点的位置
    idx ++;//将当前节点的指针向后移
}

void delete_k(int k){
    ne[k] = ne[ne[k]];//将k节点的指针指向下下一个节点的位置
}

int main(){
    cin >> M;
    init();
    while(M--){
        char opcode;
        cin >> opcode;
        if(opcode == 'H'){
            int x;
            cin >> x;
            insert_to_head(x);
        }
        else if(opcode == 'I'){
            int k,x;
            cin >> k >> x;
            insert_to_k(k-1, x);
        }
        else{
            int k;
            cin >> k;
            if(k == 0){
                head = ne[head];
            }
            else{
                delete_k(k-1);
            }
        }
        
        
    }
    for(int i=head;i!=-1;i=ne[i]){
        printf("%d ", e[i]);
    }
    printf("\n");
    return 0;
}

结果

在这里插入图片描述

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐