e[]存储当前元素
l[]存储当前元素的上一个位置
r[]存储当前元素的下一个位置
idx表示当前可用坐标
初始化
//初始化左端点和右端点
void init()
{
r[0] = 1;
l[1] = 0;
idx = 2;//因为0,1坐标已经被用掉了,所以可用坐标从2开始
}
注意!!!:
下标从2开始,0为左端点, 1为右端点, 第k个数的下标为k + 1,设a = k + 1
在第k个数后面插入元素x
void insert(int a, int x)
{
e[idx] = x;//存储当前元素
r[idx] = r[a];//操作1
l[idx] = a;//操作2
l[r[a]] = idx;//操作3
r[a] = idx;//操作4
idx ++;//表示当前坐标已经被用掉了
}
删除第k个元素
void remove(int a)
{
r[l[a]] = r[a];//操作1
l[r[a]] = l[a];//操作2
}
例子(来自acwing827. 双链表):
输入样例:
10
R 7
D 1
L 3
IL 2 10
D 3
IL 2 7
L 8
R 9
IL 4 7
IR 2 2
代码展示:
#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;
}
void insert(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)
{
l[r[k]] = l[k];
r[l[k]] = r[k];
}
int main()
{
int m;
cin >> m;
init();
while (m --)
{
string op;
int k, x;
cin >> op;
if (op == "L")
{
cin >> x;
insert(0, x);
}
if (op == "R")
{
cin >> x;
insert(l[1], x);
}
if (op == "D")
{
cin >> k;
remove(k + 1);
}
if (op == "IL")
{
cin >> k >> x;
insert(l[k + 1], x);
}
if (op == "IR")
{
cin >> k >> x;
insert(k + 1, x);
}
}
for (int i = r[0]; i != 1; i = r[i]) cout << e[i] << ' ';
}