Codeforces Round #630 (Div. 2) B. Composite Coloring（根据合数的最小质因子涂色）

题目链接：Codeforces Round #630 (Div. 2) B. Composite Coloring
欧拉筛模板来自：欧拉筛模板及大致原理

先解决一个问题：为什么用11种颜色一定能涂完所有数？

首先，我们可以如何涂色。因为本题中的数都是合数，而合数能由多个质数的相乘得到（例如 4 = 2·2，14 = 2·7，63 = 3·3·7等等），所以我们就可以用合数的最小质因子的编号来涂色，例如4的最小质因子是2，而2是第1个质数，那么可以将4涂色为1；14的最小值因子也是2，那么可以将14也涂色为1,；63的最小质因子是3，而3是第二个质数，那么可以将14涂色为3。我们发现4和14都被涂色为1，因为他们的最小质因子都是2，由此可见如果有两个数涂色相同，那么他们一定有共同的最小质因子，满足题目中 gcd(a[i], a[j]) != 1 的条件。

再看题目中的最大涂色可能是几。因为我们是根据某个合数的最小质因子涂色的，而题目给出的合数不超过1000，那么该合数的最小质因子的最大可能值就是31 （31×31=961，37×37=1369），而31是第11个质数，所以1000以内的数涂色不会超过11。

#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<sstream>
#include<algorithm>
#include<string>
#include<cstring>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
#include<set>
#include<map>
#include<utility>
#define JOJO cout<<"JOJO"<<endl;
using namespace std;

const long long INF = ~0ull>>2;
const int Inf = 0x3f3f3f3f;
const int alphabet = 26;
const int maxn = 1e3+5;
int color[15];  //color[i]记录第i种涂色的数的数量
bool prime[maxn];
int form[maxn];
int t = 1;

struct tagnode{
    int val, color; //node[i].val保存第i个数的数值，node[i].color保存第i个数的涂色
}node[maxn];

void getprime()
{
    //欧拉筛
    fill(prime, prime+maxn, true);
    prime[0] = prime[1] = false;
    for (int i = 2; i < maxn; i++){
        if (prime[i])
            form[t++] = i;
        for (int j = 1; j < t && i*form[j] < maxn; j++){
            prime[i*form[j]] = false;
            if (i%form[j] == 0) break;
        }
    }
    return;
}

void init()
{
    fill(color, color+15, 0);
    memset(node, 0, sizeof(node));
    return;
}

int main()
{
    getprime(); //获取质数表，目的是为了获得前11个质数

    int T;
    scanf("%d", &T);
    while (T--){
        init();

        /*m是最大的颜色编号*/
        int n, m = 0;
        scanf("%d", &n);
        for (int i = 1; i <= n; i++){
            scanf("%d", &node[i].val);
            for (int j = 1; form[j] <= 31; j++){
                /*求输入数值的最小质因数，并根据最小质因数涂色*/
                if (node[i].val%form[j] == 0){
                    node[i].color = j;
                    color[j]++;
                    m = max(m, j);
                    break;
                }
            }
        }

        /*因为根据最小质因数涂色可能会出现1~m之间某个涂色没有被使用的情况，
        所以最后要进行修改。
        修改不会破坏已满足的条件，因为同一涂色的数还是原来那些，
        并没有发生改变*/
        for (int i = 1; i <= m; i++){
            /*例：a涂色为1，b涂色为2，c涂色为4，d涂色为5，m=5，
            因为涂色3没有被使用，所以遍历所有数，将涂色>3的数的涂色减小，
            修改完之后，a涂色为1，b涂色为2，c涂色为3，d涂色为4，m=4，满足条件*/
            if (color[i] == 0){
                m--;
                for (int j = 1; j <= n; j++){
                    if (node[j].color > i){
                        color[node[j].color]--;
                        node[j].color--;
                        color[node[j].color]++;
                    }
                }
                /*可能出现a涂色为1，b涂色为4，c涂色为5，m=5的情况，
                此时涂色2没有被使用，要进行修改，
                进行一次减小后，a涂色为1，b涂色为3，c涂色为4，m=4，
                此时涂色2仍然没有被使用，仍不满足条件，需要再进行修改，
                再进行一次减小后，a涂色为1，b涂色为2，c涂色为3，m=3，
                这次修改完才满足条件*/
                i--;
            }
        }

        printf("%d\n", m);
        for (int i = 1; i <= n; i++)
            printf("%d ", node[i].color);
        printf("\n");
    }
    return 0;
}