acwing算法基础课/第二章/单链表 数组模拟

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

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

#include<iostream>
using namespace std;

const int N = 100010;

int e[N], ne[N]; // e[i]表示第i数的值，ne[i]表示第i个数后面的数
int head, idx; // head表示链表头，idx表示当前用到了哪个节点 

void init()
{
	head = -1;
	idx = 0;	
} 
// 将x插入到头节点后面 
void add2head(int x)
{
	e[idx] = x;
	ne[idx] = head; // idx节点指向head指向的那个节点
	head = idx; // head指向idx节点 
	idx++; 
}
// 在第k个插入的数后面插入x 
// k表示的是第k个插入的数，不是当前链表中第k个数 
void add(int k, int x)
{
	e[idx] = x;
	// 下面两句顺序不能变 
	ne[idx] = ne[k];
	ne[k] = idx;
	
	idx++; // 当前节点+1 
}
// 删除第k个插入的数后面的数 
void remove(int k)
{
	ne[k] = ne[ne[k]];
}


int main()
{
	
	init();
	
	int m;
	cin >> m;
	
	
	while(m --) 
	{
		char op;
		int x, k;
		cin >> op;
		switch(op)
		{
			case'H':
				cin >> x;
				add2head(x);
				break;
			case'D':
				cin >> k;
				if(!k) head = ne[head];
				else   remove(k - 1);
				break;
			case'I':
				cin >> k >> x;
				add(k - 1, x);
				break;
		}
	}
	for(int i = head; i != -1; i = ne[i]) cout << e[i] << " ";
	cout << endl;
	
	return 0; 
}