AcWing 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

所有操作保证合法。

输入样例：

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

此代码是在杨dalao代码的基础上修改的

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10;

// head表示第一个结点的下标
//e[i]表示数组下标为i的元素的值，注意这里的i并不代表链表的第i个结点，只表示数组的下标
//ne[i]表示数组下标为i的元素的指针
//idx表示已经插入了多少个元素，是新插入元素的下标 
int head, e[N], ne[N], idx;

//初始化
void init()
{
    head = -1;//一开始没有第一个结点时，把头指针设为-1
    idx = 0;//索引值
}

//让x成为链表的第一个结点，即向链表头插入一个数 
void add_to_head(int x)
{
    e[idx] = x;//把x赋给第idx个元素
    ne[idx] = head;//把第idx个元素的指针指向之前的第一个结点
    head = idx++;//让之前第一个结点指向这个元素
}

//删除数组中下标为k的结点，即删除插入的第k个数后面的数 
void move(int k)
{
    ne[k-1] = ne[ne[k-1]];//把本来指向下标k的指针指向k+1，即跳过k 
}  

//在下标k-1之后添加结点，即在第k个插入的数后插入一个数
void add(int k, int x)
{
    e[idx] = x;//数据
    ne[idx] = ne[k-1];//让新结点的指针改为之前第k个数的指针
    ne[k-1] = idx;//让第k个结点指向新结点
    idx++;
}

int main(void)
{
	int m,k,x;
	char c;
	cin>>m;
	init();
	while(m--)
	{
		cin>>c;
		if(c=='H')
		{
			cin>>x;
			add_to_head(x);
		}
		else if(c=='I')
		{
			cin>>k>>x;;
			add(k,x);
		}
		else if(c=='D')
		{
			cin>>k;
			if(k==0)//k等于0时比较特殊，表示删除链表中的第一个结点，而不是删除插入的第一个数 
				head=ne[head];
			else//删除第k个插入的数后面的数 
				move(k);	
		}
	}
	for(int i=head;i>=0;i=ne[i])
		cout<<e[i]<<" ";
	cout<<"\n";
	return 0;
}

模板

#include<iostream>
using namespace std;
const int N = 1e5 + 5;

int idx, head, n[N], ne[N];
int a;

//初始化
void init()
{
    head = -1;
    idx = 0;
}
//在链表头插入一个数a
void add_head(int x){
    n[idx] = x;
    ne[idx] = head;
    head = idx++;
}
//向链表中k位置插入x 
void add(int k, int x){
    n[idx] = x;
    ne[idx] = ne[k];
    ne[k] = idx++;
}
//将k删除，需要保证头结点存在
void remove(int k){
    ne[k] = ne[ne[k]];
}

int main(void)
{
	init(); 
    cin >> a;
    while (a--){
        string op;
        int k, x;
        cin >> op;
        if (op == "D"){
            cin >> k;
            if (!k) 
				head = ne[head];
            remove(k - 1);
        }
        else if (op == "H"){
            cin >> x;
            add_head(x);
        }
        else if (op == "I"){
            int k, x;
            cin >> k >> x;
            add(k - 1, x);
        }
    }
    for (int i = head; i != -1; i = ne[i])
    	cout << n[i] << " ";
    cout << endl;
	return 0;
}