洛谷 P1160 队列安排

题目描述

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

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

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

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

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

输入格式

第一行一个整数 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 号同学已经不在队列中则忽略这一条指令。

输出格式

一行，包含最多 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

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

最终队列：

2 4 1

【数据范围】

对于 20%20% 的数据，1≤N≤10。

对于 40%40% 的数据，1≤N≤1000。

对于 100%100% 的数据，1<M≤N≤10^5。

 解题思路

看完题目我们知道对于这题我们要进行大量的插入和删除操作,而一般线性表顺序储存结构对于插入和删除是十分复杂的，而链表的插入和删除却十分方便（顺序储存结构对于插入和删除需要大量移动元素很容易TLE），因此这题可以用链表，而这题的插入有两种情况，一种是插入一个数的前面和一个数的后面，因此使用链表中的双向链表；

看到这里是不是以为就可以了？不！还不够。

我第一次写这题的时候，当时就想到这然后就开始写了，然而等着你的还是TLE，因为每次要查找k需要大量时间；

我们知道链表查找一个元素是比较麻烦的，需要从头依此开始找，对于这题也一样，我们如何找到要插入的元素 K 呢？如果从头开始，首先循环N-1次输入k,p，然后又要在链表中找到k,那么整体时间复杂度为O(N^2)，这里就可以用一个数组来标记使时间复杂度降为O(N)，这题的数据最大为10^5,所以数组的大小为10^5。数组的下标为每个元素的编号，这样就可以快速找到要插入的地方在哪里了，具体操作请看代码。

AC代码

#include<stdio.h>
#include<stdlib.h>
struct list {
	struct list* prior;//前一个结点的地址
	int data;//编号
	struct list* next;//后一个结点的地址
}a[100010];
int b[100010];
int main()
{
	int n, i, j, m, p, k;
	struct list* hard, * t;
	hard = (struct list*)malloc(sizeof(struct list));//建立头结点方便操作
	a[1].prior = hard;a[1].data = 1;a[1].next = NULL;//将第一个人放进链表
	hard->next = &a[1];
	scanf("%d", &n);
	for (i = 2; i <= n; i++)
	{
		scanf("%d %d", &k, &p);
		a[i].data = i;
		if (p == 1)//插在k的后面
		{
			if (a[k].next != NULL)//如果不是链表最后一个元素
			{
				a[i].prior = &a[k];
				a[i].next = a[k].next;
				a[k].next->prior = &a[i];
				a[k].next = &a[i];
			}
			else//如果是链表最后一个元素
			{
				a[i].prior = &a[k];
				a[i].next = NULL;
				a[k].next = &a[i];
			}
		}
		else//插在k的前面
		{
			a[i].next = &a[k];
			a[i].prior = a[k].prior;
			a[k].prior->next = &a[i];
			a[k].prior = &a[i];
		}
	}
	scanf("%d", &m);
	while (m--)
	{
		scanf("%d", &k);
		b[k] = 1;//标记要删除的元素
	}
	t = hard->next;
	while (t != NULL)
	{
		if (b[t->data] == 0)//输出不删除的元素
			printf("%d ", t->data);
		t = t->next;
	}
	free(hard);
	return 0;
}