题目地址:
https://www.acwing.com/problem/content/829/
实现一个双链表,初始为空,支持
5
5
5种操作:
(1) 在最左侧插入一个数;
(2) 在最右侧插入一个数;
(3) 将第
k
k
k个插入的数删除;
(4) 在第
k
k
k个插入的数左侧插入一个数;
(5) 在第
k
k
k个插入的数右侧插入一个数。
现在要对该链表进行
M
M
M次操作,进行完所有操作后,从左到右输出整个链表。
输入格式:
第一行包含整数
M
M
M,表示操作次数。接下来
M
M
M行,每行包含一个操作命令,操作命令可能为以下几种:
(1) “L x”,表示在链表的最左端插入数
x
x
x。
(2) “R x”,表示在链表的最右端插入数
x
x
x。
(3) “D k”,表示将第
k
k
k个插入的数删除。
(4) “IL k x”,表示在第
k
k
k个插入的数左侧插入一个数。
(5) “IR k x”,表示在第
k
k
k个插入的数右侧插入一个数。
输出格式:
共一行,将整个链表从左到右输出。
数据范围:
1
≤
M
≤
100000
1\le M\le 100000
1≤M≤100000
可以用数组模拟双链表。开三个数组, e [ i ] e[i] e[i]表示下标 i i i的节点的值, l [ i ] l[i] l[i]表示下标 i i i的节点的左边的节点的下标, r [ i ] r[i] r[i]表示下标 i i i的节点的右边的节点的下标,然后 i d x idx idx表示如果当前执行插入,它所处的下标。可以先假定链表中已经有了两个节点,分别是下标 0 0 0和 1 1 1,那么初始化的时候就是 r [ 0 ] = 1 , l [ 1 ] = 0 , i d x = 2 r[0]=1,l[1]=0,idx=2 r[0]=1,l[1]=0,idx=2。关于操作,我们只需要实 5 5 5和 3 3 3就可以了。操作 1 1 1相当于在 0 0 0的右边插入一个数,操作 2 2 2相当于在 l [ 1 ] l[1] l[1]的右边插入一个数,操作 4 4 4相当于 l [ k ] l[k] l[k]的右边插入一个数(第 k k k个插入的数的下标是 k + 1 k+1 k+1,在它的左边插入相当于是在 l [ k + 1 ] l[k+1] l[k+1]的右边插入)。而操作 5 5 5和 3 3 3的实现可以参考动态双链表的写法。代码如下:
#include <iostream>
using namespace std;
const int N = 100010;
int e[N], l[N], r[N], idx;
void init() {
r[0] = 1, l[1] = 0;
idx = 2;
}
// 在下标是k的点的右边插入x
void add(int k, int x) {
e[idx] = x;
l[idx] = k;
r[idx] = r[k];
l[r[k]] = idx;
r[k] = idx;
idx++;
}
void remove(int k) {
r[l[k]] = r[k];
l[r[k]] = l[k];
}
int main() {
int n;
cin >> n;
init();
while (n--) {
string op;
cin >> op;
int k, x;
if (op == "L") {
cin >> x;
add(0, x);
} else if (op == "R") {
cin >> x;
add(l[1], x);
} else if (op == "D") {
cin >> k;
remove(k + 1);
} else if (op == "IL") {
cin >> k >> x;
add(l[k + 1], x);
} else if (op == "IR") {
cin >> k >> x;
add(k + 1, x);
}
}
int head = r[0];
while (head != 1) {
cout << e[head] << " ";
head = r[head];
}
return 0;
}
所有操作时间复杂度 O ( 1 ) O(1) O(1),空间取决于插入了多少次。