CodeForces 45 G.Prime Problem（构造+数论）

Description

给$1$~$n$每个位置填一个数字，使得每个数字位置编号之和是素数，要求所用数字种数最少

Input

一个整数$n(1\le n\le 6000)$

Output

输出$n$个整数表示给这$n$个位置填的数字，数字从$1$开始用

Sample Input

8

Sample Output

1 2 2 1 1 1 1 2

Solution

令$m=\frac{n(n+1)}{2}$表示位置编号和

1.$m$是素数则所有位置填$1$即可，只需要用一种数字

2.$m$是奇数且$m-2$不是素数，前两个位置位置放$2$，$m-3$是偶数，必然可以拆成两个素数之和，故找一个素数$x$满足$m-3-x$是素数，在第$x$个位置放$3$，其他位置都放$1$即可，此时需要用三种数字

3.$m$可以拆成两个素数，找一个素数$x$满足$m-x$是素数，在第$x$个位置放$2$，其他位置放 1 <script type="math/tex" id="MathJax-Element-24">1</script>即可，此时需要用两种数字

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=6005;
bool check(int n)
{
    if(n<2)return 0;
    for(int i=2;i*i<=n;i++)
        if(n%i==0)return 0;
    return 1;
}
int n,ans[maxn];
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)ans[i]=1;
        int m=n*(n+1)/2;
        if((m&1)&&!check(m-2))ans[3]=3,m-=3;
        for(int i=2;i<=n;i++)
            if(check(i)&&check(m-i))
            {
                ans[i]=2;
                break;
            }
        for(int i=1;i<=n;i++)printf("%d%c",ans[i],i==n?'\n':' ');
    }
    return 0;
}