题目链接:https://www.acwing.com/problem/content/828/
题意:实现一个单链表,链表初始为空,支持三种操作:
(1) 向链表头插入一个数;
(2) 删除第k个插入的数后面的数;
(3) 在第k个插入的数后插入一个数
现在要对该链表进行M次操作,进行完所有操作后,从头到尾输出整个链表。
注意:题目中第k个插入的数并不是指当前链表的第k个数。例如操作过程中一共插入了n个数,则按照插入的时间顺序,这n个数依次为:第1个插入的数,第2个插入的数,…第n个插入的数。
(1) “H x”,表示向链表头插入一个数x。
(2) “D k”,表示删除第k个输入的数后面的数(当k为0时,表示删除头结点)。
(3) “I k x”,表示在第k个输入的数后面插入一个数x(此操作中k均大于0)。
数据范围
1≤M≤1e5
所有操作保证合法。
输入样例:
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
思路:其实数组模拟的单链表至少比寻常数组多一个ne[]数组用来存储指针所指向的下一节点,这样感觉所有数都被串在了一条链上一样;而且数据的插入和删除会比寻常数组方便很多,因为少去了很多下标的改动。
代码实现:
#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,n) for(int i=a;i<=n;i++)
typedef long long ll;
const int INF=0x3f3f3f3f;
const int N = 1e5+5;
//head表示头节点下标
//e[i]表示第i点的值,
//ne[i]表示第i点所指向的下一节点的下标
//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 ++ ;
}
//将x插到第k个节点后面
void add(int k, int x)
{
e[idx] = x;
ne[idx] = ne[k];
ne[k] = 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 == 'H'){
cin >> x;
add_to_head(x);
}
else if(op == 'D'){
cin >> k;
if(!k) head = ne[head];
remove(k - 1);
}
else{
cin >> k >> x;
add(k - 1, x);
}
}
for(int i = head; i != -1; i = ne[i])
cout << e[i] << " ";
cout << endl;
return 0;
}