codeforces B. Composite Coloring

题目

题意：

给你一个$a$序列，让你给这个$a$序列涂色，但是两种颜色相同的$a_i$必须有$gcd(a_i,a_j)>1$且颜色最大不能超过$11$。

思路：

因为颜色$m$最大不超过$11$，所以我们可以从$11$入手我们可以发现质数有$2,3,5,7,11,13,17,19,23,29,31$刚好$11$个，那么第$12$个是$37，37\ast 37>1000$，所以不考虑，我们可以参考埃式筛法的思路，把所有的相同的从$2$开始的$gcd(a_i,a_i)=2$的进行标记，比如$4$可以写成$book[4]=2$，但是还有一个问题那就是有多少种颜色就不能有超过这个颜色的数字，所以最后要设置一个数组映射一下就行了。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <vector>
#include <string>
#include <cmath>
#include <set>
#include <map>
#include <deque>
#include <stack>
using namespace std;
typedef long long ll;
typedef vector<int> veci;
typedef vector<ll> vecl;
typedef pair<int, int> pii;
template <class T>
inline void read(T &ret) {
    char c;
    int sgn;
    if (c = getchar(), c == EOF) return ;
    while (c != '-' && (c < '0' || c > '9')) c = getchar();
    sgn = (c == '-') ? -1:1;
    ret = (c == '-') ? 0:(c - '0');
    while (c = getchar(), c >= '0' && c <= '9') ret = ret * 10 + (c - '0');
    ret *= sgn;
    return ;
}
inline void out(int x) {
    if (x > 9) out(x / 10);
    putchar(x % 10 + '0');
}
const int maxn = 1010;
int a[maxn], book[maxn];
int prime[11] = {2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31};
void init() {
    for (int i = 0; i < 11; i++) {
        for (int j = prime[i]; j <= maxn; j += prime[i]) {
            book[j] = i + 1;
        }
    }
}
int gcd(int a, int b) {
    if (b == 0) return a;
    else return gcd(b, a % b);
}
int main() {
    int t;
    read(t);
    init();
    while (t--) {
        int n, cnt = 0;
        int xx[20] = {0};
        read(n);
        for (int i = 0; i < n; i++) {
            read(a[i]);
            if (xx[book[a[i]]] == 0) xx[book[a[i]]] = ++cnt;
        }
        printf("%d\n", cnt);
        for (int i = 0; i < n; i++) {
            out(xx[book[a[i]]]);
            putchar(' ');
        }
        putchar('\n');
    }
    return 0;
}