题目描述
一个学校里老师要将班上 N 个同学排成一列,同学被编号为 1∼N,他采取如下的方法:
-
先将 11 号同学安排进队列,这时队列中只有他一个人;
-
2∼N 号同学依次入列,编号为 i 的同学入列方式为:老师指定编号为 i 的同学站在编号为 1∼(i−1) 中某位同学(即之前已经入列的同学)的左边或右边;
-
从队列中去掉 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;
}