List的应用:洛谷P1160队列安排

题目链接:P1160队列安排

#include<iostream>
#include<list>

using namespace std;

bool vis[100002]; //用来判断x同学是否还在队列中
list<int> li;  //存储学生序号
list<int>::iterator pos[100002]; //迭代器

int main()
{
	int n = 0;
	cin >> n;

	li.push_front(1);  //将一号同学安排进队列
	pos[1] = li.begin();  //将一号同学在链表中的迭代器赋值给pos[1]

	int k = 0, p = 0; //k为位置,p为操作(0左1右)
	for (int i = 2; i <= n; i++)
	{
		cin >> k >> p;
		if (p==0)
		{
			pos[i] = li.insert(pos[k], i); //p为0时插入到左边
		}
		else
		{
			list<int>::iterator it = pos[k]; //p为1时插入到右边
			it++;
			pos[i] = li.insert(it, i);
		}
	}

	int m = 0;
	cin >> m;
	int x = 0;
	for (int i = 0; i < m; i++)
	{
		cin >> x;
		if (!vis[x]) //判断x同学是否还在链表中
		{
			li.erase(pos[x]);
			vis[x] = true;
		}
	}
	//遍历链表
	for (list<int>::iterator it=li.begin();it!=li.end();it++)
	{
		cout << *it << " ";
	}
	cout << endl;

	return 0;
}
原创文章 85 获赞 86 访问量 1万+

猜你喜欢

转载自blog.csdn.net/weixin_43699840/article/details/105514259