【夏洛克和他的女朋友】

题目

思路

• 将所有数分为两个集合：质数集合和合数集合
  • 质数集合各元素间不存在质因子关系
  • 合数集合各元素间不存在质因子关系
  • 任意合数集合的元素总能在质数集合找到质因子关系的配合
• 质数集合每个元素的颜色定为1
• 合数集合每个元素的颜色定为2
• 默认颜色种类为1，出现合数时更新为2

代码

#include <bits/stdc++.h>
using namespace std;

const int N = 1e5 + 10;
int f[N]; // 用于存储每个珠宝的颜色
bool st[N]; // 用于标记是否是合数
int primes[N], cnt; // 存储质数列表
int color = 1;
// 线性筛法获取质数
void get_primes(int n) {
    for (int i = 2; i <= n; i++) {
        if (!st[i]) {
            primes[++cnt] = i;  // i 是质数
            f[i] = 1; // 质数染成颜色 1
        }
        for (int j = 1; primes[j] * i <= n; j++) {
            st[primes[j] * i] = true; // 标记合数
            f[primes[j] * i] = 2; // 合数染成颜色 2
            color = 2;
            if (i % primes[j] == 0) break;
        }
    }
}

int main() {
    int n;
    cin >> n;
    
    get_primes(n + 1); // 获取所有到 n+1 的质数和合数
    
    // 输出使用的颜色数量（因为只有两种颜色）
    cout << color << '\n';
    
    // 输出每个珠宝的颜色
    for (int i = 2; i <= n + 1; i++) {
        cout << f[i] << " ";
    }
    cout << endl;
    
    return 0;
}