洛谷 P1160 队列安排

https://www.luogu.org/problemnew/show/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号同学插入到k号同学的左边，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左边，此时队列为：

21

将同学3插入至同学2右边，此时队列为：

231

将同学4插入至同学1左边，此时队列为：

2341

将同学3从队列中移出，此时队列为：

241

同学3已经不在队列中，忽略最后一条指令

最终队列：

241

数据范围

对于20%的数据，有N≤10；

对于40%的数据，有N≤1000；

对于100%的数据，有N,M≤100000。

思路：用数组来模拟队列，既然每次新同学入列都是在某个已入列同学的左侧或右侧，那么自然想到用两个数组L和R来保存队列信息。L[i]和R[i]分别记录第i个同学左侧同学和右侧同学的编号。(初始化为0)这样只要知道K(即同学的编号)和P的值(0左侧 1右侧)就可以直接模拟入队操作。仔细思考就知道 入队要分两种情况 这里以p=0 即把i号同学插到k号同学左侧为例来说明。

(1)k号同学左侧没有同学 即L[k]=0时:

L[k]=i
R[i]=k

(2)k号同学左侧有同学 即L[k]!=0时:

(先修改i号同学左右两侧的同学编号 还需要修改原来k号同学左侧同学的右侧同学编号 最后修改k号同学的左侧同学编号)

 L[i]=L[k]
 R[i]=k
 R[L[k]]=i
 L[k]=i

在右侧插入同学的操作类似， 不再赘述。 至此入队模拟已经完成， 至于出队操作完全没必要实现， 用flag数组标记出队同学的编号， 在输出的时候跳过就行了。遍历队列需要最左侧同学的编号， 因此我们在执行左侧的插入操作时，顺便记录最左侧同学的编号。(最后遍历L数组找到L[i]=0的i也行 但是这种方法更快)边界即R[i]=0。(i同学右侧没有人了)

代码中也有较为详细的注释。

#include<iostream>
#include<cstdio>
using namespace std;

int L[100005],R[100005],flag[100005];//L[i]表示i号同学左侧同学的编号

int main()
{
    int n,m,left=1;
    scanf("%d",&n);
    L[1]=0,R[1]=0;
    for(int i=2;i<=n;i++)
    {
        int temp1,temp2;
        scanf("%d %d",&temp1,&temp2);
        if(temp2==0)    //i插到temp1左侧
        {
            if(L[temp1]==0) //temp1 左侧没有人
            {
                L[temp1]=i;//temp1 左侧的人是 i
                R[i]=temp1;//i 右侧的人是 temp1
            }
            else//temp1 左侧有人
            {
                L[i]=L[temp1];//i 左侧的人是temp1 左侧的人
                R[i]=temp1;//i 右侧的人是 temp1
                R[L[temp1]]=i;//temp1 原来左侧的人的现在右侧的人是 i
                L[temp1]=i;//temp1 的左侧的人是 i
            }
            if(L[i]==0) //i左侧没有人
                left=i; //记录最左侧同学的编号
        }
        else    //右侧
        {
            if(R[temp1]==0)
            {
                R[temp1]=i;
                L[i]=temp1;
            }
            else
            {
                L[i]=temp1;
                R[i]=R[temp1];
                L[R[temp1]]=i;
                R[temp1]=i;
            }
        }
    }
    scanf("%d",&m);
    for(int i=0;i<m;i++)
    {
        int temp1;
        scanf("%d",&temp1);
        flag[temp1]=1;//给出队的同学打上标记
    }
    while(left!=0)  //打印编号
    {
        if(!flag[left])
            printf("%d ",left);
        left=R[left];
    }
    return 0;
}