洛谷 P1160 队列安排

题目描述
一个学校里老师要将班上NN个同学排成一列，同学被编号为1\sim N1∼N，他采取如下的方法：

先将11号同学安排进队列，这时队列中只有他一个人；

2-N2−N号同学依次入列，编号为i的同学入列方式为：老师指定编号为i的同学站在编号为1\sim (i -1)1∼(i−1)中某位同学（即之前已经入列的同学）的左边或右边；

从队列中去掉M(M<N)M(M<N)个同学，其他同学位置顺序不变。

在所有同学按照上述方法队列排列完毕后，老师想知道从左到右所有同学的编号。

输入格式
第11行为一个正整数NN，表示了有NN个同学。

第2-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号同学已经不在队列中则忽略这一条指令。

输出格式
11行，包含最多NN个空格隔开的正整数，表示了队列从左到右所有同学的编号，行末换行且无空格。

输入输出样例
输入 #1 复制
4
1 0
2 1
1 0
2
3
3
输出 #1 复制
2 4 1

说明/提示
样例解释：

将同学22插入至同学11左边，此时队列为：

2 121

将同学33插入至同学22右边，此时队列为：

2 3 1231

将同学44插入至同学11左边，此时队列为：

2 3 4 12341

将同学33从队列中移出，此时队列为：

2 4 1241

同学33已经不在队列中，忽略最后一条指令

最终队列：

2 4 1241

数据范围

对于20%20%的数据，有N≤10N≤10；

对于40%40%的数据，有N≤1000N≤1000；

对于100%100%的数据，有N, M≤100000N,M≤100000。

---

因为这一题需要在一个位置的左侧和右侧都会有插入操作，同时还有删除结点的操作。所以我们选择使用双链表

这里是数组模拟双链表的代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 100010;

int n, m;
int e[N], l[N], r[N], idx;
bool st[N];

void init(){
    r[0] = 1;
    l[1] = 0;
    idx = 2;
}
void add(int k, int x){
    e[idx] = x;
    r[idx] = r[k];
    l[idx] = k;
    l[r[k]] = idx;
    r[k] = idx ++;
}

void remove(int k){
    r[l[k]] = r[k];
    l[r[k]] = l[k];
}

int main(){
    cin >> n;
    
    init();
    add(0, 1);
    for (int i = 2; i <= n; i ++ ){
        int k, p; cin >> k >> p;
        if (p == 0) add(l[k + 1], i);
        else add(k + 1, i);
    }
    
    cin >> m;
    while (m -- ){
        int k; cin >> k;
        if (!st[k]) remove(k + 1), st[k] = true;
        else continue;
    }
    
    for (int i = r[0]; i != 1; i = r[i]) cout << e[i] << ' ';
    
    return 0;
}