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

Description

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

Input

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

Output

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

Sample Input

样例1
5
RRR

样例2
7
BBB

样例3
6
RBR
Sample Output

样例1
2 3 5

样例2
1 4 6

样例3
-1 4 5
Hint

数据范围：
对于10%数据 1 ≤ n ≤ 20
对于50%数据 1 ≤ n ≤ 200
对于所有数据 1 ≤ n ≤ 1000

---

先将n以内的合数用数组表示出来。
然后我们先正的搜一遍字符串，求出第一种可能性。
再到过来搜一遍，判断这个位置还有没有其他的可能，有则替换为-1。

---

代码如下:

var n,i,j,k,t:longint;
    s:ansistring;
    c:array[0..1001]of longint;
    a,b:array[1..1000]of boolean;
begin
  readln(n);
  readln(s);
  fillchar(a,sizeof(a),true);
  a[1]:=false;
  for i:=2 to n do if a[i] then for j:=2 to n div i do a[i*j]:=false;
  fillchar(b,sizeof(b),true);
  j:=1;
  for i:=1 to n do
    if ((s[j]='R')and(a[i]))or((s[j]='B')and(not a[i])) then
      begin
        c[j]:=i;
        inc(j);
        if j>length(s) then break;
      end;
  c[j]:=n+1;
  for i:=length(s) downto 1 do
    for j:=c[i]+1 to c[i+1]-1 do
      if ((s[i]='R')and(a[j]))or((s[i]='B')and(not a[j])) then
        begin
          c[i]:=j;
          b[i]:=false;
        end;
  for i:=1 to length(s) do if b[i] then write(c[i],' ') else write('-1 ');
end.