2012年佛山市GDOI选拔赛题 红蓝牌

2012年佛山市GDOI选拔赛题 红蓝牌

Description

明明有N张牌，正面按编号写有数字1到N（每个数字只出现一次）。每张牌的反面根据正面的数字涂上两种颜色。如果正面的数字是质数，则为红色，反之为蓝色。 
明明的老师从这N张牌里选出了一些，按数字从小到大排序后全部反面向上放成一行。他要明明判断这些牌上的数字。

Input

第一行：一个正整数N, 1 <= N <= 1000. 
第二行：一个字符串，长度 K 在 1 到 50 之间。字符串中只有'R'和'B'，'R'表示红色，'B'表示蓝色。

Output

一行，有K个整数，每个整数代表相应位置上的牌的正面数字。如果某张牌不能确定上面的数字，否则输出-1.

分析：向前枚举一次向后枚举一次，得到第i位两个答案，如果两个答案不同说明不能确定。

代码

  const

  maxn=1000;

var

  a:array[0..maxn] of longint;

  f:array[0..maxn] of boolean;

  i,j,p,q,n:longint;

  st:string;

function max(x,y:longint):longint;

begin

  if x>y then exit(x) else exit(y);

end;

function min(x,y:longint):longint;

begin

  if x<y then exit(x) else exit(y);

end;

begin

  readln(n);

  readln(st);

  f[2]:=true;f[3]:=true;f[5]:=true;f[7]:=true;

  for i:=8 to 1000 do

    begin

      for j:=2 to i div 2 do

        if i mod j=0 then break;

      if i mod j<>0 then f[i]:=true;

    end;

  p:=2;

  q:=1;

  for i:=1 to length(st) do

    begin

      if st[i]='R' then

        begin

          while not f[p] do inc(p);

          a[i]:=p;

          inc(p);

          q:=max(p,q);

        end;

      if st[i]='B' then

        begin

          while f[q] do inc(q);

          a[i]:=q;

          inc(q);

          p:=max(p,q);

        end;

    end;

  for i:=n downto 2 do

    if f[i] then

      begin

        p:=i;

        break;

      end;

  for i:=n downto 1 do

    if not f[i] then

      begin

        q:=i;

        break;

      end;

  for i:=length(st) downto 1 do

    begin

      if st[i]='R' then

        begin

          while not f[p] do dec(p);

          if p<>a[i] then a[i]:=-1;

          dec(p);

          q:=min(p,q);

        end;

      if st[i]='B' then

        begin

          while f[q] do dec(q);

          if q<>a[i] then a[i]:=-1;

          dec(q);

          p:=min(p,q);

        end;

    end;

  for i:=1 to length(st) do

    write(a[i],' ');

end.