洛谷 队列安排

队列安排 p1160

题目描述

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

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

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

3. 从队列中去掉 M 个同学，其他同学位置顺序不变。

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

输入格式

第一行一个整数 𝑁N，表示了有 𝑁N 个同学。

第 2∼𝑁2∼N 行，第 𝑖i 行包含两个整数 k,p，其中 k 为小于 i 的正整数，p 为 0 或者 1。若 p 为 0，则表示将 i 号同学插入到 𝑘 号同学的左边，𝑝为 1 则表示插入到右边。

第 𝑁+1N+1 行为一个整数 𝑀M，表示去掉的同学数目。

接下来 𝑀M 行，每行一个正整数 𝑥x，表示将 𝑥x 号同学从队列中移去，如果 𝑥x 号同学已经不在队列中则忽略这一条指令。

输出格式

一行，包含最多 𝑁N 个空格隔开的整数，表示了队列从左到右所有同学的编号

。

输入输出样例

输入 #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

【数据范围】

对于 20%20% 的数据，1≤𝑁≤101≤N≤10。

对于 40%40% 的数据，1≤𝑁≤10001≤N≤1000。

对于 100%100% 的数据，1<𝑀≤𝑁≤1051<M≤N≤105

解题思路

用两个数组分别记录前后节点元素，下标即代表当前元素。插入元素的两种情况可概括为插在k后或者k前元素的后面。

难点
插入元素，删除元素。

题解

#include <iostream>
using namespace std;
const int N=100005;
int l[N],r[N]; 
int n,m,k,p;
void dfs(int i){
	cout<<i<<" ";
	if(r[i]!=-1) dfs(r[i]);
}
void del(int i){
	
	if (l[i] != -1) r[l[i]] = r[i];  
    if (r[i] != -1) l[r[i]] = l[i];  
    r[i] = l[i] = -1; // 将当前节点设置为无效 可以想一下没有这一步会有什么问题 
}
void add(int i,int k){
	l[i]=k;
	r[i]=r[k];
	if(r[k]!=-1) l[r[k]]=i;
	r[k]=i;
}
int main(){
	cin>>n;
//初始化
	r[0]=1;
	l[1]=0;
	r[1]=-1;
	for(int i=2;i<=n;i++){
		cin>>k>>p;
		add(i,p?k:l[k]);
	}
	cin>>m;
	while(m--){
		cin>>k;
		del(k);
	}
	if(r[0]!=-1) dfs(r[0]);
} 

时间复杂度为O(N)。