红蓝牌 ssl 2627

题目大意

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

分析

正着来一次
倒着来一次
比较一下，不同的就是不确定的。

code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>
#include<stack>

using namespace std;

int f[2000];
int a[2000],b[2000];
int n;
int main()
{
    scanf("%d",&n);
    f[1]=1;
    for (int i=1;i<=(int)sqrt((double)n);i++)
        if (!f[i])
            for (int j=2;j<=n/i;j++)
                f[j*i]=1;
    char s[2000];
    scanf("%s",s);
    int j=0;
    for (int i=0;i<=strlen(s)-1;i++)
    {
        if (s[i]=='R')
        {
            j++;
            while (f[j]) j++;
            a[++a[0]]=j;
        }
        if (s[i]=='B')
        {
            j++;
            while (!f[j]) j++;
            a[++a[0]]=j;
        }
    }
    j=n+1;
    b[0]=strlen(s)+1;
    for (int i=strlen(s)-1;i>=0;i--)
    {
        if (s[i]=='R')
        {
            j--;
            while (f[j]) j--;
            b[--b[0]]=j;
        }
        if (s[i]=='B')
        {
            j--;
            while (!f[j]) j--;
            b[--b[0]]=j;
        }
    }
    for (int i=1;i<=a[0];i++)
        if (a[i]!=b[i])
            printf("-1 ");
        else
            printf("%d ",a[i]);
}