洛谷——P1160 队列安排

题目描述

一个学校里老师要将班上NN个同学排成一列，同学被编号为1\sim N1∼N，他采取如下的方法：

1. 先将11号同学安排进队列，这时队列中只有他一个人；

2. 2-N2−N号同学依次入列，编号为i的同学入列方式为：老师指定编号为i的同学站在编号为1\sim (i -1)1∼(i−1)中某位同学（即之前已经入列的同学）的左边或右边；

3. 从队列中去掉M(M<N)M(M<N)个同学，其他同学位置顺序不变。

在所有同学按照上述方法队列排列完毕后，老师想知道从左到右所有同学的编号。

输入格式

第11行为一个正整数NN，表示了有NN个同学。

第2-N行，第ii行包含两个整数k,p，其中k为小于ii的正整数，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左边，此时队列为：

2 1

将同学33插入至同学22右边，此时队列为：

2 3 1

将同学44插入至同学11左边，此时队列为：

2 3 4 1

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

2 4 1

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

最终队列：

2 4 1

数据范围

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

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

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

 解题思路：使用双链表。

Code:

#include <iostream>
#include <stdio.h>

using namespace std;

const int N = 100010;

int r[N], l[N], idx, e[N];
bool f[N];

void init() //初始化
{
    r[0] = 1;
    l[1] = 0;
    idx = 2;
}

void head()
{
    e[idx] = 1;
    r[idx] = r[0];
    l[idx] = 0;
    l[r[idx]] = idx;
    r[l[idx]] = idx;
    idx ++;
}

void add(int k, int x)
{
    e[idx] = x;
    r[idx] = r[k];
    l[idx] = k;
    r[l[idx]] = idx;
    l[r[idx]] = idx;
    idx ++;
}

void rmove(int k)
{
    r[l[k]] = r[k];
    l[r[k]] = l[k];
}

signed main()
{
    int n, m;
    scanf("%d", &n);
    init();
    head();
    for(int i = 1; i <= n; i ++ )  f[i] = true;
    for(int i = 2; i <= n; i ++ )
    {
        int x, p;
        cin >> x >> p;
        if(p == 1)  add(x + 1, i);
        else  add(l[x + 1], i);
    }
    scanf("%d", &m);
    for(int i = 1; i <= m; i ++ )
    {
        int x;
        scanf("%d", &x);
        f[x] = false;
    }
    for(int i = r[0]; i != 1; i = r[i])
    {
        if(f[e[i]])  cout << e[i] << ' ';
    }
}