CF-Round #630-div2-B题

CF-Round #630-div2-B题

B. Composite Coloring

传送门

这道题。算一个贪心把~
就是模拟。

题目大意:涂色问题。给定一个序列。涂的颜色不能超过11种。涂相同颜色的数字的gcd(a, b) > 1。（他们的最大公约数要大于1）
输出每个数的颜色。

贪心策略：我们知道一个数分解质因数的质数个数不会超过11个。所以我们最多需要11种颜色可以达到题目要求。所以我们找找一个数的约数。取它的最小约数。涂上一个颜色。把这个约数标记一下。把下标加进去。最后输出只需要按照下标输出。所以我们记录一下。就行啦~

哈哈哈别问。问就是想试一下gif的效果。

代码部分：

#include <bits/stdc++.h>
using namespace std;
const int N = 1e3 + 10;

vector<int> ans[N];
int a[N];

int solve(int n)
{
	for (int i = 2; i <= n; i++)
	{
		if (!(n % i))
		{
			return i;
		} 
	}
}

int main()
{
	int t, n;
	cin >> t;
	while (t--)
	{
		for (int i = 2; i <= 1000; i++)
		{
			ans[i].clear();
		}
		cin >> n;
		for (int i = 1; i <= n; i++)
		{
			int t;
			cin >> t;
			ans[solve(t)].push_back(i);
		}
		int res = 0;
		for (int i = 2; i <= 1000; i++)
		{
			int t = ans[i].size();
			if (t > 0)
			{
				res++;
				for (int j = 0; j < t; j++)
				{
					a[ans[i][j]] = res;
				}
			}
		}
		cout << res << endl;
		for (int i = 1; i <= n; i++)
		{
			cout << a[i] << " ";
		}
		cout << endl;
	}
	return 0;
}