【ACWing】1293. 夏洛克和他的女朋友

题目地址：

https://www.acwing.com/problem/content/1295/

夏洛克有了一个新女友（这太不像他了！）。情人节到了，他想送给女友一些珠宝当做礼物。他买了$n$件珠宝，第$i$件的价值是$i+1$，也就是说，珠宝的价值分别为$2,3,\dots ,n+1$。华生挑战夏洛克，让他给这些珠宝染色，使得一件珠宝的价格是另一件珠宝的价格的质因子时，两件珠宝的颜色不同。并且，华生要求他使用的颜色数尽可能少。请帮助夏洛克完成这个简单的任务。

输入格式：
只有一行一个整数$n$，表示珠宝件数。

输出格式：
第一行一个整数$k$，表示所使用的颜色数；
第二行$n$个整数，表示第$1$到第$n$件珠宝被染成的颜色。
若有多种答案，输出任意一种。
请用$1$到$k$表示你用到的颜色。

数据范围：
$1\le n\le 10^5$

首先容易知道，当$2\sim n+1$中有合数的时候，颜色数必然不少于$2$（当然如果$n\le 2$，只需要一种颜色）。并且两种颜色也够用了，可以将所有质数染成一种颜色，合数染成另一种。直接筛素数即可。代码如下：

#include <iostream>
using namespace std;

const int N = 1e5 + 10;
int n;
int p[N], idx;
bool st[N];

int main() {
  scanf("%d", &n);

  for (int i = 2; i <= n + 1; i++) {
    if (!st[i]) p[idx++] = i;
    for (int j = 0; i * p[j] <= n + 1; j++) {
      st[i * p[j]] = true;
      if (i % p[j] == 0) break;
    }
  }

  n <= 2 ? puts("1") : puts("2");
  for (int i = 2; i <= n + 1; i++)
    if (!st[i]) printf("1 ");
    else printf("2 ");
}

时空复杂度$O(n)$。