P1160 队列安排 - 双向链表
https://www.luogu.org/problemnew/show/1160
题目描述
一个学校里老师要将班上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。
思路:对于快速的插入和删除操作我们可以用链表实现。
注意:进行链表的删除操作时,别忘了把要删除的节点的l和r赋值为0。
AC Code:
1 #include<cstdio> 2 #include<cstring> 3 using namespace std; 4 const int sz=100000+10; 5 struct node{ 6 int l,r; 7 }p[sz]; 8 int main() 9 { 10 int n; 11 scanf("%d",&n); 12 int head=1; 13 int k,P; 14 for(int i=2;i<=n;i++) 15 { 16 scanf("%d%d",&k,&P); 17 if(P==0) 18 { 19 p[i].l=p[k].l;p[i].r=k;p[p[k].l].r=i;p[k].l=i; 20 if(k==head) head=i;//如果i被插到队头的左边,就更新队头 21 } 22 else 23 { 24 p[i].l=k;p[i].r=p[k].r;p[p[k].r].l=i;p[k].r=i; 25 } 26 } 27 int m; 28 scanf("%d",&m); 29 int x; 30 for(int i=1;i<=m;i++) 31 { 32 scanf("%d",&x); 33 p[p[x].l].r=p[x].r;p[p[x].r].l=p[x].l; 34 p[x].l=p[x].r=0;//别忘了把x的左右都删掉 35 } 36 int now=head; 37 printf("%d",head); 38 while(p[now].r) 39 { 40 now=p[now].r; 41 printf(" %d",now); 42 } 43 printf("\n");//行末换行且无空格 44 return 0; 45 }