由于链表写起来非常长并且效率比较低,所以我们采用数组来模拟一下链表,执行快且写起来快
首先先发一个题目背景,来自luogu的1160队列安排
题目描述
一个学校里老师要将班上 NN 个同学排成一列,同学被编号为 1\sim N1∼N,他采取如下的方法:
-
先将 11 号同学安排进队列,这时队列中只有他一个人;
-
2\sim N2∼N 号同学依次入列,编号为 ii 的同学入列方式为:老师指定编号为 ii 的同学站在编号为 1\sim(i-1)1∼(i−1) 中某位同学(即之前已经入列的同学)的左边或右边;
-
从队列中去掉 MM 个同学,其他同学位置顺序不变。
在所有同学按照上述方法队列排列完毕后,老师想知道从左到右所有同学的编号。
输入格式
第一行一个整数 NN,表示了有 NN 个同学。
第 2\sim N2∼N 行,第 ii 行包含两个整数 k,pk,p,其中 kk 为小于 ii 的正整数,pp 为 00 或者 11。若 pp 为 00,则表示将 ii 号同学插入到 kk 号同学的左边,pp 为 11 则表示插入到右边。
第 N+1N+1 行为一个整数 MM,表示去掉的同学数目。
接下来 MM 行,每行一个正整数 xx,表示将 xx 号同学从队列中移去,如果 xx 号同学已经不在队列中则忽略这一条指令。
输出格式
一行,包含最多 NN 个空格隔开的整数,表示了队列从左到右所有同学的编号。
输入输出样例
输入 #1复制
4 1 0 2 1 1 0 2 3 3
输出 #1复制
2 4 1
说明/提示
【样例解释】
将同学 22 插入至同学 11 左边,此时队列为:
2 1
将同学 33 插入至同学 22 右边,此时队列为:
2 3 1
将同学 44 插入至同学 11 左边,此时队列为:
2 3 4 1
将同学 33 从队列中移出,此时队列为:
2 4 1
同学 33 已经不在队列中,忽略最后一条指令
最终队列:
2 4 1
看到这里有个在k左边插入和右边插入我们自然就想到用双链表来解决这个问题
详解
初始化
void init()
{
r[0] = 1,l[1] = 0, idx = 2;
}
r表示向右的指针,l表示向左指针,初始化有两个元素,0和1,所以我们把idx初始化为2
插入操作
右边插入
//在k的右边插入一个元素
void add(int k, int x)
{
e[idx] = x;
l[idx] = k;
r[idx] = r[k];
l[r[k]] = idx;
r[k] = idx++;
}
注意倒数第二行和最后一行是不可以颠倒的,因为没进行最后一步的时候r[k]还是指向后一个元素,要是先最后一行,r[k]就指向了那个插入 的元素,我们就找不到了后面这个元素了。
左边插入
我们可以偷懒一下,k这个位置的左边插入,我们可以写成k左边元素,即l[k]的右边插入
调用的时候k位置的参数填k[l[k]]即可
删除
void remove(int k)
{
if (l[k] == 0) return;
else r[l[k]] = r[k];
l[r[k]] = l[k];
l[k] = 0, r[k] = 0;
}
直接看代码,如果元素空了,直接返回,因为初始化的时候l[k]=0,所以判断是否是初始化的状态就知道是不是空的了,后面这两个操作就是k位置指向后边的后边了,直接跳过要删除的元素
最后我们再把删除的这个元素的左指针和右指针置为0,表示已经删除了他。
显示
判断是否达到初始状态即遍历了链表
要想反方向输出,用r[ ]右指针即可,结束条件变成不等于0.
void show()
{
for (int i = r[0]; i != 1; i = r[i]) printf("%d ", e[i]);
}
奉上全代码
#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int h, e[N], l[N], r[N], idx;
void init()
{
r[0] = 1,l[1] = 0, idx = 2;
}
//在k的右边插入一个元素
void add(int k, int x)
{
e[idx] = x;
l[idx] = k;
r[idx] = r[k];
l[r[k]] = idx;
r[k] = idx++;
}
void remove(int k)
{
if (l[k] == 0) return;
else r[l[k]] = r[k];
l[r[k]] = l[k];
l[k] = 0, r[k] = 0;
}
void show()
{
for (int i = r[0]; i != 1; i = r[i]) printf("%d ", e[i]);
}
int main()
{
int n;
init();
scanf("%d", &n);
add(l[1], 1);
for (int i = 2; i <= n; i++)
{
int k, p;
scanf("%d%d", &k, &p);
if (p == 1)
{
add(k+1, i);
}
else if (p == 0)
{
add(l[k+1], i);
}
}
int m; scanf("%d", &m);
for(int i=1;i<=m;i++)
{ int x;
scanf("%d", &x);
remove(x+1);
}
show();
return 0;
}