洛谷1160 队列安排

题目

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

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

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

3.从队列中去掉M个同学，其他同学位置顺序不变。

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

题解

因为n还是比较大的(n<=100000),又因为要不停的插入和删除，所以我们可以用链表。

a[i,2]表示学号为i的同学右边同学的学号
a[i,3]表示学号为i的同学左边同学的学号
读入每一个同学时，都把他左边和右边的同学更新；删除同学时，先把这个同学赋为0，再把他左边的同学连上右边的同学；最后找到排在最左边的同学，挨个输出。
时间复杂度O(n)

代码

var
  a:array[1..100000,1..3]of longint;
  n,m,i,j,k,t,p:longint;
begin
  readln(n);
  a[1,1]:=1;j:=1;
  for i:=2 to n do
    begin
      readln(k,t);
      a[i,1]:=i;
      if t=0 then begin
                    a[a[k,3],2]:=i;
                    a[i,2]:=k;
                    a[i,3]:=a[k,3];
                    a[k,3]:=i;
                    if k=j then
                      j:=i;
                  end
             else begin
                    a[i,2]:=a[k,2];
                    a[a[k,2],3]:=i;
                    a[k,2]:=i;
                    a[i,3]:=k;
                  end;
    end;
  readln(m);
  for i:=1 to m do
    begin
      readln(k);
      if a[k,1]<>0 then
        begin
          a[k,1]:=0;
          t:=a[k,2];
          p:=a[k,3];
          a[p,2]:=t;
          a[t,3]:=p;
          dec(n);
          if k=j then
             j:=p;
        end;
    end;
  i:=1;k:=j;
  while i<=n do
    begin
      write(a[k,1],' ');
      k:=a[k,2];
      inc(i);
    end;
end.