题目传送门:826. 单链表
实现一个单链表,链表初始为空,支持三种操作:
- 向链表头插入一个数;
- 删除第 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
试题解析:
解题方法:用数组构造链表。
为什么不用我们常用的结构体类型构造链表呢?请注意本题的操作次数m为10w级别的,如果用结构体进行操作,会频繁的new对象,会造成超时。
明确一下题意:
先解释一下什么叫做 删除第 k 个插入的数后面的数:
- 将插入操作先后排序,所取的是第k次插入操作后的数
- 并不是链表从左到右的第k个数字
由于我们数组下标从零开始,代码中起存储当前点位置作用idx是从零开始,所以删除第 k 个插入的数后面的数实际上是删除下标为k-1的点后面的点。
同理,在第 k 个插入的数后插入一个数实际上也是在索引为k-1的点后插入一个数。
题解:
- head表示头结点的下标
- e[i]表示节点i的值
- ne[i]表示节点i的next指针是多少
- idx存储当前已经用到了哪个点
对于在头节点后面插入元素:
- 在e的idx处存储元素的值 e[idx] = x;
- 使当前元素的下一个是原头结点的后面 ne[idx] = head;
- 将头结点更新为插入节点 head = idx;
- idx自增指向下一个可存储元素的位置 idx++;
对于在索引k后插入一个数
- 在e的idx处存储元素的值 e[idx] = x;
- 使当前元素的下一个是第k个插入的数的后面 ne[idx] = ne[k];
- 使第k个插入的数的下一个指向该元素 ne[k] = idx;
- idx自增指向下一个可存储元素的位置 idx++;
对于删除索引为k的元素的后一个元素
- 将当前索引为k的元素指向下一个的下一个元素 ne[k] = ne[ne[k]];
代码如下:
#include<iostream>
using namespace std;
constexpr 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 t;
cin>>t;
init();
while(t--)
{
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;
}