Codeforces Round #630 (Div. 2) B.Composite Coloring

题目传送


题目大意:
给你n个"合数"**(这里的合数在题目中规定为由大于1的俩个数之积)**且这个合数小于等于1000。问是否能由m(1 <= m <= 11)种颜色给数字染色，使得同种颜色的数字的最大公约数不为1，(同一个数字可能有多个颜色，且这m个颜色必须至少用一次)。

思路
博主不知道我的办法是不是太笨了。。。。。。
由于这个合数一定小于等于1000且这个数至少为俩个数之积，那么我们可以暴力，判断其是否能被2整除，3整除，4整除依次，那么到31的时候，31*31已经大于1000了。但是我们得知m最大有11中颜色，显然超了。
但是我们可以优化一下，能4整除的一定能被2整除，能被6整除的一定能被3整除，依次类推。那么我们最后优化后，得到我们只用判断 2 3 5 7 11 13 17 19 23 29 31
为什么呢，最开始我们可以优化到先判断1到9的整除关系，如果不能被1到9整除的话，就剩下质数了(俩个质数相乘也可以得到一个合适)，11 13 17 19 23 29 31。这个时候再把1到9优化一下就好了

下面还得注意一个条件:就是m种颜色至少用一次，本来写出来没有多少行代码的，但是硬生生给我加了这么长。。。。。

AC代码

#include <bits/stdc++.h>
using namespace std;
#define NewNode (ListNode *)malloc(sizeof(ListNode))
#define Mem(a,b) memset(a,b,sizeof(a))
const int N = 1e6 + 5000;
const int INF = 0x3f3f3f3f;
const double EPS = 1e-10;
const unsigned long long mod = 998244353;
const int II = 3.1415926535;
typedef long long ll;
typedef unsigned long long ull;
int main()
{
    std::ios::sync_with_stdio(false);
    cin.tie(0),cout.tie(0);
    int t;
    cin >> t;
    while(t--)
    {
        int n;
        int vis[35] = {0};
        int ans = 0;
        cin >> n;
        int arr[n+5];
        for(int i = 0;i < n;i++)
            cin >> arr[i];
        for(int i = 0;i < n;i++)
        {
            if(arr[i] % 2 == 0)
            {
                if(!vis[2])//只想到了这个笨办法。。。
                {
                    ans++;
                    vis[2] = ans;
                }
                arr[i] = vis[2];
            }
            else if(arr[i] % 3 == 0)
            {
                if(!vis[3])
                {
                    ans++;
                    vis[3] = ans;
                }
                arr[i] = vis[3];
            }
            else if(arr[i] % 5 == 0)
            {
                if(!vis[5])
                {
                    ans++;
                    vis[5] = ans;
                }
                arr[i] = vis[5];
            }
            else if(arr[i] % 7 == 0)
            {
                if(!vis[7])
                {
                    ans++;
                    vis[7] = ans;
                }
                arr[i] = vis[7];
            }
            else if(arr[i] % 11 == 0)
            {
                if(!vis[11])
                {
                    ans++;
                    vis[11] = ans;
                }
                arr[i] = vis[11];
            }
            else if(arr[i] % 13 == 0)
            {
                if(!vis[13])
                {
                    ans++;
                    vis[13] = ans;
                }
                arr[i] = vis[13];
            }
            else if(arr[i] % 17 == 0)
            {
                if(!vis[17])
                {
                    ans++;
                    vis[17] = ans;
                }
                arr[i] = vis[17];
            }
            else if(arr[i] % 19 == 0)
            {
                if(!vis[19])
                {
                    ans++;
                    vis[19] = ans;
                }
                arr[i] = vis[19];
            }
            else if(arr[i] % 23 == 0)
            {
                if(!vis[23])
                {
                    ans++;
                    vis[23] = ans;
                }
                arr[i] = vis[23];
            }
            else if(arr[i] % 29 == 0)
            {
                if(!vis[29])
                {
                    ans++;
                    vis[29] = ans;
                }
                arr[i] = vis[29];
            }
            else if(arr[i] % 31 == 0)
            {
                if(!vis[31])
                {
                    ans++;
                    vis[31] = ans;
                }
                arr[i] = vis[31];
            }
        }
        cout << ans << endl;
        for(int i = 0;i < n;i++)
            cout << arr[i] << " ";
        cout << endl;
    }
}