题目描述
一个学校里老师要将班上N个同学排成一列,同学被编号为1~N,他采取如下的方法:
1.先将1号同学安排进队列,这时队列中只有他一个人;
2.2~N号同学依次入列,编号为i的同学入列方式为:老师指定编号为i的同学站在编号为1~i -1中某位同学(即之前已经入列的同学)的左边或右边;
3.从队列中去掉M(M<N)个同学,其他同学位置顺序不变。
在所有同学按照上述方法队列排列完毕后,老师想知道从左到右所有同学的编号。
输入输出格式
输入格式:
输入文件arrange.in的第1行为一个正整数N,表示了有N个同学。
第2~第N行,第i行包含两个整数k,p,其中k为小于i的正整数,p为0或者1。若p为0,则表示将i号同学插入到k号同学的左边,p为1则表示插入到右边。
第N+1行为一个正整数M,表示去掉的同学数目。
接下来M行,每行一个正整数x,表示将x号同学从队列中移去,如果x号同学已经不在队列中则忽略这一条指令。
输出格式:
输入文件arrange.out仅包括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%的数据,有N≤10;
对于40%的数据,有N≤1000;
对于100%的数据,有N, M≤100000。
代码
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
using namespace std;
int l[],r[];
int head,N,m; void insert(int des,int i,int oper){
if(oper){//插入右边
if(!r[des]){//右边没人
r[des]=i,l[i]=des;
}
else{
l[i]=des,r[i]=r[des];
l[r[des]]=i,r[des]=i;
}
}
else{//插入左边
if(!l[des]){//左边没人
l[des]=i,r[i]=des,head=i;
}
else{
l[i]=l[des],r[i]=des;
r[l[des]]=i,l[des]=i;
}
}
} int main(){
// freopen("01.in","r",stdin);
scanf("%d",&N);
for(int i=;i<=N;i++){
int x,oper;
scanf("%d%d",&x,&oper);
insert(x,i,oper);
} scanf("%d",&m);
for(int i=;i<=m;i++){
int temp=;
scanf("%d",&temp);
if(!l[temp]&&!r[temp]) continue;
if(temp==head) head=r[temp]; l[r[temp]]=l[temp];
r[l[temp]]=r[temp]; l[temp]=r[temp]=;
} if(m==N){
puts("");return ;
}
else{
for(int p=head;;p=r[p]){
if(r[p]) printf("%d ",p);
else {
printf("%d",p);break;
}
}
puts("");
}
return ;
}用数组模拟链表,看题解说用list会超时?