题目描述
给定 n n n,你现在需要给整数 1 1 1 到 n n n 进行染色,使得对于所有的 1 ≤ i < j ≤ n 1\leq i<j\leq n 1≤i<j≤n,若 j − i j - i j−i 为质数,则 i i i 和 j j j 不同色。
求出颜色尽可能少的染色方案。如果有多种方案,输出任意一种即可。
输入格式
第一行一个整数 n n n。
输出格式
第一行一个整数 k k k,表示颜色数。
第二行 n n n 个整数 c o l i col_i coli ( 1 ≤ c o l i ≤ k 1 \leq col_i \leq k 1≤coli≤k),表示 i i i 的颜色。
样例
样例输入
7
样例输出
4
1 2 2 3 3 4 1
数据范围与提示
对于 30 % 30\% 30% 的数据, n ≤ 10 n \leq 10 n≤10;
对于 60 % 60\% 60% 的数据, n ≤ 20 n \leq 20 n≤20;
对于 100 % 100\% 100% 的数据, n ≤ 1 0 4 n \leq 10^4 n≤104 。
SOLUTION
方案一: 由于质数除了 2 2 2 以外都是奇数,所以有一种必然可行的方案:
1 1 2 2 3 3 ⋯ 1\ 1\ 2\ 2\ 3\ 3\cdots 1 1 2 2 3 3⋯
该方案所需的颜色种类为 ⌈ n 2 ⌉ \lceil \frac{n}{2} \rceil ⌈2n⌉ 。
方案二: 而又由于这个性质,所以我们让每两个相同的颜色的对应数值之差都是偶数的时候也可行。除了 2 2 2 外,最小可选 4 4 4 ,于是构造如下方案:
1 2 3 4 1 2 3 4 ⋯ 1\ 2\ 3\ 4\ 1\ 2\ 3\ 4\cdots 1 2 3 4 1 2 3 4⋯
因此每种方案最多只需要 4 4 4 种颜色。
又当 n ≤ 6 n \le 6 n≤6 时,采取方案一 ⌈ n 2 ⌉ < 4 \lceil \frac{n}{2} \rceil < 4 ⌈2n⌉<4,
因此分类讨论即可。
CODE
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int n;
int main()
{
scanf("%d", &n);
if(n <= 6)
{
printf("%d\n", (n + 1) >> 1);
for(register int i = 1; i <= n; i ++ ) printf("%d ", (i + 1) >> 1);
putchar('\n');
}
else
{
printf("4\n");
for(register int i = 1; i <= n; i ++ ) printf("%d ", i % 4 + 1);
putchar('\n');
}
return 0;
}
END.