算法基础-数据结构

她的店里只卖樱花

发布于 2022-10-31 08:39:55

46300

代码可运行

文章被收录于专栏：常用算法模板常用算法模板

运行总次数：0

代码可运行

单链表

01.单链表

题目描述

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

向链表头插入一个数；
删除第 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\le M\le 100000所有操作保证合法。

输入样例：

输出样例：

6 4 6 5

题解

时间复杂度：O(1)

核心思想

是一个模板题直接背板子

tps:注意特判下，要删除的元素的下标是不是0，如果是0，那我们就删除头节点。由于单链表的下标是0 ~ n - 1, 我们要判断的是1 ~ n, 所以在传入下标的时候记得 -1

核心函数

void init(){
    idx = 0;
    head = -1;
}
void add_to_head(int x){
    e[idx] = x; 
    ne[idx] = head;
    head = idx ++ ;
}

void remove(int k){
    ne[k] = ne[ne[k]];
    
}

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

代码实现

#include <iostream>

using namespace std;

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

void init(){
    idx = 0;
    head = -1;
}
void add_to_head(int x){
    e[idx] = x; 
    ne[idx] = head;
    head = idx ++ ;
}

void remove(int k){
    ne[k] = ne[ne[k]];
    
}

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

int main (){
    cin >> M;
    init();
    while(M --)
    {
        int x, k;
        char op[2]; 
        cin >> op;
        if(*op == 'H')
        {
            cin >> x;
            add_to_head(x);
        }
        else if(*op == 'D')
        {
            cin >> k;
            if(k)
                remove(k - 1);
            else
                head = ne[head]; // 删除头节点
        }
        else 
        {
            cin >> k >> x; 
            insert(x, k - 1);
        }
    }
    
    for(int i = head; i != -1; i = ne[i])  
        cout << e[i] << " ";
    cout << endl; 
    return 0;
}