P1160 队列安排
题目描述:
一个学校里老师要将班上N个同学排成一列,同学被编号为1∼N,他采取如下的方法:
1、先将1号同学安排进队列,这时队列中只有他一个人;
2、2-N号同学依次入列,编号为i的同学入列方式为:老师指定编号为i的同学站在编号为1∼(i−1)中某位同学(即之前已经入列的同学)的左边或右边;
3、从队列中去掉M(M<N)个同学,其他同学位置顺序不变。
在所有同学按照上述方法队列排列完毕后,老师想知道从左到右所有同学的编号。
输入格式
第1行为一个正整数N,表示了有N个同学。
第2−N行,第i行包含两个整数k,p,其中k为小于i的正整数,p为0或者1。若p为0,则表示将i号同学插入到kk号同学的左边,p为1则表示插入到右边。
第N+1行为一个正整数M,表示去掉的同学数目。
接下来M行,每行一个正整数x,表示将x号同学从队列中移去,如x号同学已经不在队列中则忽略这一条指令。
输出格式
1行,包含最多N个空格隔开的正整数,表示了队列从左到右所有同学的编号,行末换行且无空格。
输入样例#1
4
1 0
2 1
1 0
2
3
3
输出样例#1
2 4 1
说明/提示
将同学2插入至同学1左边,此时队列为:
2 1
将同学3插入至同学2右边,此时队列为:
2 3 1
将同学4插入至同学1左边,此时队列为:
2 3 4 1
将同学3从队列中移出,此时队列为:
2 4 1
同学3已经不在队列中,忽略最后一条指令
最终队列:
2 4 1
数据范围
对于20%20%的数据,有N≤10N≤10;
对于40%40%的数据,有N≤1000N≤1000;
对于100%100%的数据,有N, M≤100000N,M≤100000。
思路: 题目确实很长,但仔细读即可以读懂;虽然名字是队列,但是队列的方法并不好做,所以采用链表的形式比较好,我的想法是采用C++中的STL中的List方法,通过具体迭代器返回相应的位置元素
上代码:
#include<iostream>
#include<cstdio>
#include<list>
using namespace std;
bool vis[100003];//标记x元素
list<int> stu;
list<int>::iterator pos[100003];//迭代器,方便查找
int main()
{
int n,m;
scanf("%d",&n);
stu.push_front(1);//设置元素从1开始
pos[1]=stu.begin();
for(int i=2;i<=n;++i){//读入第2~n行
int k,p;
scanf("%d%d",&k,&p);
if(p==0){
pos[i]=stu.insert(pos[k],i);//插入左边,并返回一个插入位置的迭代器
}else{
list<int>:: iterator it=pos[k];
++it;
pos[i]=stu.insert(it,i);//否则插入在右边
}
}
scanf("%d",&m);
for(int i=0;i<m;i++){
int x;
scanf("%d",&x);
if(!vis[x]){//如果没有被标记
stu.erase(pos[x]);//删除该元素
vis[x]=true;
}
}
for(list<int>::iterator it=stu.begin();it!=stu.end();it++){
printf("%d ",*it);
}
printf("\n");
return 0;
}