单链表 AcWing 826. 单链表
实现一个单链表,链表初始为空,支持三种操作:
向链表头插入一个数;删除第 k 个插入的数后面的数;在第 k 个插入的数后插入一个数。现在要对该链表进行 M MM 次操作,进行完所有操作后,从头到尾输出整个链表。
注意:题目中第 k 个插入的数并不是指当前链表的第 k kk 个数。例如操作过程中一共插入了 n 个数,则按照插入的时间顺序,这 n 个数依次为:第 1 个插入的数,第 2 个插入的数,…第 n 个插入的数。
#include<iostream>
using namespace std;
const int N=100010;
// head 表示头节点的下标
// e[i] 表示节点i的值
// ne[i] 表示节点i的next指针是多少
// idx 存储已经用到了那个点
int head,e[N],ne[N],idx;
// 初始化
void init ()
{
head=-1;
idx=0;
}
// 将x插到头节点
void add_to_head(int x)
{
e[idx]=x;
ne[idx]=head;
head=idx;
idx++;
}
// 将x插到下标是k的点后面
void add (int k,int x)
{
e[idx]=x;
ne[idx]=ne[k];
ne[k]=idx;
idx++;
}
//将下标是k的后面的值删掉
void remove (int k)
{
ne[k]=ne[ne[k]];
}
int main ()
{
int m;
cin >> m;
init ();
while (m--)
{
int k,x;
char op;
cin >> op;
if (op=='D')
{
cin >> k;
if (!k) head=ne[head];
remove (k-1);
}
else if (op=='H')
{
cin >> x;
add_to_head (x);
}
else
{
cin >> k >> x;
add (k-1,x);
}
}
for (int i=head;i!=-1;i=ne[i]) cout << e[i] << ' ';
return 0;
}