https://www.luogu.com.cn/problem/P1160
题目描述
一个学校里老师要将班上NN个同学排成一列,同学被编号为1\sim N1∼N,他采取如下的方法:
先将11号同学安排进队列,这时队列中只有他一个人;
2-N2−N号同学依次入列,编号为i的同学入列方式为:老师指定编号为i的同学站在编号为1\sim (i -1)1∼(i−1)中某位同学(即之前已经入列的同学)的左边或右边;
从队列中去掉M(M<N)M(M<N)个同学,其他同学位置顺序不变。
在所有同学按照上述方法队列排列完毕后,老师想知道从左到右所有同学的编号。
输入格式
第11行为一个正整数NN,表示了有NN个同学。
第2-N2−N行,第ii行包含两个整数k,pk,p,其中kk为小于ii的正整数,pp为00或者11。若pp为00,则表示将ii号同学插入到kk号同学的左边,pp为11则表示插入到右边。
第N+1N+1行为一个正整数MM,表示去掉的同学数目。
接下来MM行,每行一个正整数xx,表示将xx号同学从队列中移去,如果xx号同学已经不在队列中则忽略这一条指令。
输出格式
11行,包含最多NN个空格隔开的正整数,表示了队列从左到右所有同学的编号,行末换行且无空格。
输入输出样例
输入
4
1 0
2 1
1 0
2
3
3
输出
2 4 1
说明/提示
样例解释:
将同学22插入至同学11左边,此时队列为:
2 121
将同学33插入至同学22右边,此时队列为:
2 3 1231
将同学44插入至同学11左边,此时队列为:
2 3 4 12341
将同学33从队列中移出,此时队列为:
2 4 1241
同学33已经不在队列中,忽略最后一条指令
最终队列:
2 4 1241
数据范围
对于20%20%的数据,有N≤10N≤10;
对于40%40%的数据,有N≤1000N≤1000;
对于100%100%的数据,有N, M≤100000N,M≤100000。
题记:用一个list链表来实现即可
#include<iostream>
#include<list>
#include<cstdio>
using namespace std;
bool vis[100003];
list<int> stus;
list<int>::iterator pos[100003];
int main()
{
int n;
cin >> n;
stus.push_front(1);//push_front() 在list的头部添加一个元素
pos[1] = stus.begin();//begin返回指向第一个元素的迭代器
for (int i = 2; i <= n; i++)
{
int k, p;
cin >> k >> p;
if (p == 0)//p为0时插入到左边
pos[i] = stus.insert(pos[k], i); //插入一个元素到list中
else
{
list<int>::iterator it = pos[k];
++it;
pos[i] = stus.insert(it, i);
}
}
int m;
cin >> m;
for (int i = 0; i < m; i++)
{
int x;
cin >> x;
if (!vis[x])//判断是否已经删除过了
{
stus.erase(pos[x]);//erase删除一个元素
vis[x] = true;
}
}
for (list<int>::iterator it = stus.begin(); it != stus.end(); it++)
printf("%d ", *it);
cout << endl;
return 0;
}