AcWing 1293. 夏洛克和他的女友（二分图染色 思维 线性筛）

夏洛克有了一个新女友（这太不像他了！）。

情人节到了，他想送给女友一些珠宝当做礼物。

他买了 n 件珠宝，第 i 件的价值是 i+1，也就是说，珠宝的价值分别为 2,3,…,n+1。

华生挑战夏洛克，让他给这些珠宝染色，使得一件珠宝的价格是另一件珠宝的价格的质因子时，两件珠宝的颜色不同。

并且，华生要求他使用的颜色数尽可能少。

请帮助夏洛克完成这个简单的任务。

输入格式

只有一行一个整数 n，表示珠宝件数。

输出格式

第一行一个整数 k，表示所使用的颜色数；

第二行 n 个整数，表示第 1 到第 n 件珠宝被染成的颜色。

若有多种答案，输出任意一种。

请用 1 到 k 表示你用到的颜色。

数据范围

1 ≤ n ≤ 10^5

输入样例1：
3
1

输出样例1：
2
1 1 2
1
2

输入样例2：
4
1

输出样例2：
2
2 1 1 2

题意：

给这些珠宝染色，使得一件珠宝的价格是另一件珠宝的价格的质因子时，两件珠宝的颜色不同。

即：从一个质数向合数连一条边，最后会构成二分图。

注：

①质数之间彼此互质没有连边，合数之间不满足题意也没有连边。

②第i件物品代表的数字是i+1。

思路：

当n<=2时只有质数：2、3，输出1，由二分图（有且被1,2染色）可知其他情况均为2（n>2）。

代码：

#include<bits/stdc++.h>

using namespace std;
int n;
const int N = 1e6+10;
int cnt,primes[N];
bool st[N];

void init(int n)
{
    for(int i=2;i<=n;++i)
    {
        if(!st[i])
        {
            primes[cnt++] = i;
        }
        for(int j=0;primes[j]*i<=n;++j)
        {
            st[primes[j]*i] = true;
            if(i%primes[j]==0) break;
        }
    }
}

int main()
{
    cin>>n;
    init(n+1);//由题意第i件物品代表的数字是i+1，因此n件物品意味着要筛2~n+1之间的数，而不是1~n
    
    if(n<=2) cout<<1<<endl;//只有2，3两个质数，因此只有一种颜色
    else cout<<2<<endl;//既有质数又有合数，因此有两种颜色，且最多就两种颜色，因为构成的是二分图（有且被1,2染色）
    
    for(int i=2;i<=n+1;++i)
    {
        if(!st[i]) cout<<1<<' ';//质数输出1
        else cout<<2<<' ';//合数输出2
    }
    
    return 0;
}