洛谷 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
分析
stl list 练习
代码
#include<bits/stdc++.h>
using namespace std;
const int max_N = 1e5+10;
list <int> List;
int N;
list<int>::iterator pos[max_N];
bool er[max_N];
int main()
{
cin>>N;
List.push_back(1);
pos[1]=List.begin();
for(int i=2;i<=N;i++)
{
int k,p;
cin>>k>>p;
if(p==0)
{
pos[i]=List.insert(pos[k],i);//返还i位置指针 *插入同学时前面已经有此人
}
else if(p==1)
{
pos[i]=List.insert(next(pos[k],1),i);
}
}
int M;
cin>>M;
for(int i=1;i<=M;i++)
{
int k;
cin>>k;
if(er[k]) continue;//不在此队中
List.erase(pos[k]);
er[k]=true;
}
list<int>::iterator it;
for(it=List.begin();it!=List.end();++it)
{
cout<<*it<<" ";
}
return 0;
}