从零开始学数据结构(Day 3)
参考了Acwing的题目和解题思路
实现一个双链表,双链表初始为空,支持 5种操作:
在最左侧插入一个数;
在最右侧插入一个数;
将第 k个插入的数删除;
在第 k个插入的数左侧插入一个数;
在第 k个插入的数右侧插入一个数
现在要对该链表进行 M次操作,进行完所有操作后,从左到右输出整个链表。
注意:题目中第 k个插入的数并不是指当前链表的第 k 个数。例如操作过程中一共插入了 n 个数,则按照插入的时间顺序,这 n 个数依次为:第 1 个插入的数,第 2 个插入的数,…第 n个插入的数。
输入格式
第一行包含整数 M,表示操作次数。
接下来 M行,每行包含一个操作命令,操作命令可能为以下几种:
L x,表示在链表的最左端插入数 x
R x,表示在链表的最右端插入数 x
D k,表示将第 k个插入的数删除。
IL k x,表示在第 k个插入的数左侧插入一个数。
IR k x,表示在第 k个插入的数右侧插入一个数。
输出格式
共一行,将整个链表从左到右输出。
#include "iostream"
using namespace std;
int r[100001],l[100001],e[100001] ,idx ;
void head_insert(int a)
{
e[idx] = a;
r[idx] = r[0];
l[idx] = 0;
r[0] = idx;
l[r[idx]] = idx;
idx++;
}
void right(int a)
{
e[idx] = a;
r[idx] = 1;
l[idx] = l[1];
r[l[1]] = idx;
l[1] = idx++;
}
void del(int a)
{
l[r[a+1]] = l[a+1];
r[l[a+1]] = r[a+1];
}
void left_insert(int k,int a)
{
e[idx] = a;
r[idx] = k+1;
l[idx] = l[k+1];
l[k+1] = idx;
r[l[idx]] = idx;
idx++;
}
void right_insert(int k,int a)
{
e[idx] = a;
l[idx] = k+1;
r[idx] = r[k+1];
r[k+1] = idx;
l[r[idx]] = idx;
idx++;
}
int main()
{
r[0] = 1,l[1] = 0,idx = 2;
int m;
cin >> m;
while(m--)
{
string a;
int c,d;
cin >> a;
if(a == "L")
{
cin >> c;
head_insert(c);
}
else if(a == "R")
{
cin >> c;
right(c);
}
else if(a == "D")
{
cin >> c;
del(c);
}
else if(a == "IL")
{
cin >> c >> d;
left_insert(c,d);
}
else if(a == "IR")
{
cin >> c >> d;
right_insert(c , d);
}
}
for(int i = r[0];i != 1;i = r[i])
{
cout << e[i] << " ";
}
return 0;
}
其实我们的代码还可以优化,比如说将几个插入函数可以合并成一个:
向左插入就等效与向当前位置的前一个插入
向头插入就等效与向头节点后面插入
向最右端就等效于向尾节点前面的节点的右端插入
笔记字有点丑,但是都是自己抽空整理的
莫欺少年穷,共勉
~~B站关注嘉然今天吃什么 和 星瞳 ~~