Codeforces 1332 B. Composite Coloring

题意:

给你 $n$ 个合数且这个合数小于等于 $1000$。问是否能由 $m$ 种颜色给数字染色，使得同种颜色的数字的最大公约数不为 $1$。

最大公约数不为 $1$，那就是不互质，就是有除 $1$ 以外的公共因子，$11$种颜色转换一下不就是 $11$ 个不同的公共因子嘛，先把前 $11$ 个素数给找出来，就用这 $11$ 个素数作为因子来找，如果两个数都是这个素数的倍数那么就涂相同颜色，没有和他一样因子就涂一个新的颜色。

AC代码：

const int N = 1010;
int n, m;
int a[N], ans[N];
int cnt, res, tmp;
int dis[11] = {2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31};
map<int, int> mp;

int main()
{
	int t;
	sd(t);
	while (t--)
	{
		sd(n);
		rep(i, 1, n)
			sd(a[i]);
		mp.clear();
		cnt = 0;
		rep(i, 1, n)
		{
			rep(j, 0, 10)
			{
				if (a[i] % dis[j] == 0)
				{
					if (!mp[dis[j]])
						mp[dis[j]] = ++cnt;
					ans[i] = mp[dis[j]];
					break;
				}
			}
		}
		pd(cnt);
		rep(i, 1, n)
			printf("%d%c", ans[i], i == n ? '\n' : ' ');
	}
	return 0;
}