洛谷 P1160 队列安排(双向链表

题目描述

一个学校里老师要将班上N个同学排成一列，同学被编号为1∼N，他采取如下的方法：

先将1号同学安排进队列，这时队列中只有他一个人；

2−N号同学依次入列，编号为i的同学入列方式为：老师指定编号为i的同学站在编号为1∼(i−1)中某位同学（即之前已经入列的同学）的左边或右边；

从队列中去掉M(M<N)个同学，其他同学位置顺序不变。

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

输入格式

第1行为一个正整数N，表示了有N个同学。

第2−N行，第iii行包含两个整数k,p，其中k为小于iii的正整数，p为0或者1。若p为0，则表示将i号同学插入到k号同学的左边，p为1则表示插入到右边。

第N+1行为一个正整数M，表示去掉的同学数目。

接下来M行，每行一个正整数x，表示将x号同学从队列中移去，如果x号同学已经不在队列中则忽略这一条指令。
输出格式

1行，包含最多N个空格隔开的正整数，表示了队列从左到右所有同学的编号，行末换行且无空格。

输入输出样例
输入 #1
4
1 0
2 1
1 0
2
3
3
输出 #1
2 4 1

说明/提示

样例解释：

将同学2插入至同学1左边，此时队列为：

21

将同学3插入至同学2右边，此时队列为：

231

将同学4插入至同学1左边，此时队列为：

2341

将同学3从队列中移出，此时队列为：

241

同学3已经不在队列中，忽略最后一条指令

最终队列：

241

数据范围

对于20%的数据，有N≤10；

对于40%的数据，有N≤1000；

对于100%的数据，有N,M≤100000。

解题思路
题目中涉及到插入，删除操作很明显是要用到链表。
但是数据范围是N,M≤100000，那么如果查询操作要遍历一遍链表的话肯定TLE。
这个时候也是要用到双向链表，只不过利用数组就可以在O(1)的时间内查询到节点。
具体代码为：

typedef struct {
    int r,l;
    int flag;
}ss;
ss k[100030];

这里的r值为i节点右面的节点，l值是i节点左面的节点。
flag作为标记，当节点i被删除之后标记为0。

以下为完整代码：

#include <iostream>
#include <algorithm>
#include <math.h>
#include <string>
#include <vector>
#include <map>
#include <stack>
#include <cstdio>
#include <list>
#include <queue>
using namespace std;

typedef struct {
    int r,l;
    int flag;
}ss;
ss k[100030];
int main(){
    int n,m,i,x,y,s=0;
    cin>>n;
    k[1].r=-1,k[1].l=0,k[1].flag=1;
    k[0].r=1,k[0].flag=1;
    for(i=2;i<=n;i++){
        k[i].flag=1;
        cin>>x>>y;
        if(y==0){
            k[k[x].l].r=i;
            k[i].l=k[x].l;
            k[i].r=x;
            k[x].l=i;
        }
        else {
            if(k[x].r!=-1) {
                k[k[x].r].l = i;
                k[i].r = k[x].r;
                k[i].l = x;
                k[x].r = i;
            }
            else {
                k[x].r=i;
                k[i].l=x;
                k[i].r=-1;
            }
        }
    }
    cin>>m;
    for(i=0;i<m;i++){
        cin>>x;
        if(k[x].flag==1){
            k[x].flag=0;
            k[k[x].l].r=k[x].r;
            k[k[x].r].l=k[x].l;
            s++;
        }
    }
    int pr=0,p=k[pr].r;
    for(i=0;i<n-s;i++){
        cout<<p<<" ";
        pr=p;
        p=k[p].r;
    }
}

也可以申请一个二维数组来存。

int num[100030][3];

这样也可。