实现一个单链表,链表初始为空,支持三种操作:
向链表头插入一个数;
删除第 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
所有操作保证合法。
输入样例:
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
这道题其实很久以前就做过,但是再看一遍的时候还是忘了,遂决定写个博客加深印象。
变量说明
head: 头节点指针
idx:已经使用过的节点个数
e[N]: 存放的数值
ne[i]: 第i点的指针,指向链表下一个节点的位置。
功能分析
初始化:
head = -1;
idx = 0;
头插:
void add_to_head(int x)
{
e[idx] = x;//存储
ne[idx] = head;//让这个即将插入的节点的指针指向原来head指向的对象
head = idx;//head指针指向这个插入的节点
idx ++;//已经使用的节点数++
}
正常插入
void add(int x, int k)//在k这个位置插入一个数
{
e[idx] = x;
ne[idx] = ne[k];//指向原来k指向的下一个节点
ne[k] = idx;让k指向的下一个节点为这个插入的节点
idx++;
}
删除
void remove(int k)
{
ne[k] = ne[ne[k]];//将k指向的下一个节点,变成k下一个节点的下一个节点,跳过了中间的节点
}
代码
#include <iostream>
using namespace std;
const int N = 100010;
int e[N], ne[N], head, idx;
void init(){
head= -1;
idx= 0;
}
void add_head(int x){
e[idx] = x, ne[idx] = head, head = idx++;
}
void add(int k, int x){
e[idx] = x, ne[idx] = ne[k], ne[k] = idx++;
}
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_head(x);
}
else if (op == 'D'){
cin >> k;
if (!k) head = ne[head];
else 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;
}