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：

#include<cstdio>
#include<cstring>
using namespace std;
const int sz=100000+10;
struct node{
    int l,r;
}p[sz];
int main()
{
    int n;
    scanf("%d",&n);
    int head=1;
    int k,P;
    for(int i=2;i<=n;i++)
    {
        scanf("%d%d",&k,&P);
        if(P==0)
        {
            p[i].l=p[k].l;p[i].r=k;p[p[k].l].r=i;p[k].l=i;
            if(k==head) head=i;//如果i被插到队头的左边，就更新队头 
        }
        else 
        {
            p[i].l=k;p[i].r=p[k].r;p[p[k].r].l=i;p[k].r=i;
        }
    }
    int m;
    scanf("%d",&m);
    int x;
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&x);
        p[p[x].l].r=p[x].r;p[p[x].r].l=p[x].l;
        p[x].l=p[x].r=0;//别忘了把x的左右都删掉 
    }
    int now=head;
    printf("%d",head);
    while(p[now].r)
    {
        now=p[now].r;
        printf(" %d",now);
    }
    printf("\n");//行末换行且无空格
    return 0;
}