P1160 队列安排(list)

题目链接:https://www.luogu.org/problemnew/show/P1160

思路:直接用list容器按要求模拟,另外要用一个pos数组记录每一位同学在迭代器的位置。list容器是一个双向链表,可以高效地进行插入删除元素。使用list容器之前必须加上头文件:#include

AC代码:

#include 
using namespace std;
const int maxn = 1e5 + 10;
list ::iterator pos[maxn];
list  l;
bool vis[maxn];
int n, m, x, k, p;
int main()
{
    scanf("%d", &n);
    l.push_front(1);
    pos[1] = l.begin();
    for (int i = 2; i <= n; i++)
    {
        scanf("%d%d", &k, &p);
        if (p == 0)
            pos[i] = l.insert(pos[k], i);
        else
        {
            auto nextit = next(pos[k]);
            pos[i] = l.insert(nextit, i);
        }
            
    }
    scanf("%d", &m);
    while(m--)
    {
        scanf("%d", &x);
        if (!vis[x])
            l.erase(pos[x]);
        vis[x] = true;
    }
    for(int x: l)
        printf("%d ", x);
    return 0;
}

 

你可能感兴趣的:(【STL运用】,【模拟】)