CF776B Sherlock and his girlfriend

CF776B Sherlock and his girlfriend

一个数和它的质因数不能同色。

素数的约数只有本身和1。

显然所有素数都可以染同色，合数同理。

欧拉筛一筛。

注意特判n<3的情况。

#include<iostream>
#include<cstdio>
#include<cstring>
#define re register
using namespace std;
#define N 100002
int n,v[N],pri[N],cct;bool is[N];
int main(){
    for(int i=2;i<N;++i){
        if(!v[i]) is[v[i]=pri[++cct]=i]=1;
        for(int j=1;j<=cct;++j){
            if(pri[j]>i||pri[j]*i>=N) break;
            v[pri[j]*i]=pri[j];
        }
    }scanf("%d",&n);
    if(n<3){
        puts("1");
        for(int i=1;i<=n;++i) printf("1 ");
    }else{
        puts("2");
        for(re int i=2;i<=n+1;++i) printf("%d ",(!is[i])+1);
    }
    return 0;
}