Codeforces Round #630 (Div. 2) B - Composite Coloring（数论基础）

不管是暴力找还是怎样，都可以发现，1000以内合数较小的质因数都是在这11个质数
2,3,5,7,11,13,17,19,23,29,31范围内的，所以知道为啥是11了嘛？
所以只要看每个数能整除这11个数中的哪个即可！复杂度为o(11*n)

#include <iostream>
#include <cstring>
#include <string.h>
#include <stdlib.h>
#include <algorithm>
#include <cstdio>
#include <stdio.h>
#include<queue>
#include<math.h>
#include<stack>
#include<set>
#include<map>
using namespace std;
int p[30]={0,2,3,5,7,11,13,17,19,23,29,31};
int a[1005];
int ans[1005];
int vis[20];
int main()
{
    int t;cin>>t;
    while(t--)
    {
        memset(vis,0,sizeof(vis));
        int n;cin>>n;
        for(int i=1;i<=n;i++)cin>>a[i];
        int num=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=11;j++)
            {
                if(a[i]%p[j]==0)
                {
                    if(!vis[j])
                    {
                        vis[j]=++num;
                    }
                    ans[i]=vis[j];
                    break;
                }
            }
        }
        cout<<num<<'\n';
        for(int i=1;i<=n;i++)cout<<ans[i]<<' ';
        cout<<'\n';
    }
    
}