计蒜客-1870-队列安排(STL的list)

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

先将 11 号同学安排进队列,这时队列中只有他一个人;

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

从队列中去掉 MM 个同学,其他同学位置顺序不变。

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

输入格式
第 11 行为一个正整数 N\ (1\le N \le 10^5)N (1≤N≤10
5
),表示了有 NN 个同学。

第 2\sim N2∼N 行,第 ii 行包含两个整数 k,pk,p,其中 kk 为小于 ii 的正整数,pp 为 00 或者 11。若 pp 为 00,则表示将 ii 号同学插入到 kk 号同学的左边,pp 为 11 则表示插入到右边。

第 N+1N+1 行为一个正整数 M (1\le M \le 10^5)M(1≤M≤10
5
),表示去掉的同学数目。

接下来 MM 行,每行一个正整数 xx,表示将 xx 号同学从队列中移去,如果 xx 号同学已经不在队列中则忽略这一条指令。

输出格式
一行,包含最多 NN 个空格隔开的正整数,表示了队列从左到右所有同学的编号,行末换行且无空格。

输出时每行末尾的多余空格,不影响答案正确性

样例输入复制
4
1 0
2 1
1 0
2
3
3
样例输出复制
2 4 1

本题涉及到大量的元素的查找和插入,所以使用list,可以提前保存其他元素的内存位置,从而更加方便的查找和删除,bingo

//@author:hairu,WU
//@from:ahut
#include
#include
#include
#include
using namespace std;

int n;
list<int> v;
list<int>::iterator ites[100100];
int visited[100100];

int main(){
	cin >> n;
	v.push_back(1);
	ites[1]=v.begin();
	for(int i=2;i<=n;i++){
		int k,p;
		cin >> k>>p;
		if(p==0){
			ites[i] = v.insert(ites[k],i);
		}else if(p==1){
			list<int>::iterator ite=ites[k];
			ite++;
			ites[i]=v.insert(ite,i);
		}
	}
	int m;
	cin >> m;
	memset(visited,false,sizeof(visited));
	for(int i=0;i<m;i++){
		int x;
		cin >> x;
		if(!visited[x]){
			v.erase(ites[x]);
			visited[x]=true;
		}
	}
	
	for(list<int>::iterator ite=v.begin();ite!=v.end();ite++){
		cout<<*ite<<" ";
	}
	return 0;
}

你可能感兴趣的:(难题)