实现一个单链表,链表初始为空,支持三种操作:
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
#include <iostream>
using namespace std;
const int N = 1e5 + 10;
//head 表示头结点的下标
//e[i] 表示节点i的值
//ne[i] 表示节点i的next指针是多少
//idx 存储当前已经用到了哪个节点
int head, e[N], ne[N], idx;
//初始化
void init()
{
head = -1;
idx = 0;
}
//x插到头节点
void insert_head(int x)
{
e[idx] = x; // 新结点的数值
ne[idx] = head; // 新结点指针指向的是原来 head 指向的结点 (1)
head = idx; // head 指向新结点 (2)
idx ++;
}
//一般的插入操作(将x点插到下标是k的点的后面)
void insert(int x, int k)
{
e[idx] = x;
ne[idx] = ne[k];
ne[k] = idx;
idx ++;
}
// 删除下标为 k 的结点之后的那个结点
void remove(int k)
{
ne[k] = ne[ne[k]]; // 下标为 k 的结点的指针指向 原来下一个结点的下一个结点
}
void remove_head()
{
head = ne[head]; // 头指针 head 指向下下个结点(删除了下一个结点)
}
int main()
{
int m;
cin >> m;
init(); //初始化
while (m --)
{
int k, x;
char op;
cin >> op;
if (op == 'H')
{
cin >> x;
insert_head(x);
}
else if (op == 'D')
{
cin >> k;
if (k == 0)
{
remove_head();
}
else
{
remove(k - 1); //数组下标是从0开始;因为是第k个插入的元素,所以下标是k - 1;
}
}
else //第三种情况
{
cin >> k >> x;
add(x, k - 1);
}
}
//i从头节点开始,i ≠ -1(即i没有遍历到空节点),每次i往后走一步
for (int i = head; i != -1; i = ne[i]) cout << e[i] << " ";
cout << endl;
return 0;
}