Codeforces 1198F

题目链接：Codeforces 1198F

Description

给出 (n) 个数，把它们分成两组使得两组的 (gcd) 均为 (1)，或判断无解。

Solution

随机的都是好东西

太菜了不会正解只能用随机来骗分。

先把序列打乱。

如果当前数加到第一组使其 (gcd) 变小了就加到第一组，否则加到第二组。

然后卡一下时间。

证明不存在的

---

Code

#include <cstdio>
#include <cmath>
#include <cstring>
#include <cstdlib>
#include <ctime>
#include <algorithm>
 
const int MAXN = 1e5 + 10;
int n, a[MAXN], id[MAXN], ans[MAXN];
 
int gcd(int x, int y) {
	return y ? gcd(y, x % y) : x;
}
 
int main() {
	scanf("%d", &n);
	for (int i = 1; i <= n; ++i) scanf("%d", &a[i]), id[i] = i;
	clock_t st = clock();
	while (clock() - st < 0.4 * CLOCKS_PER_SEC) {
		std::random_shuffle(id + 1, id + n + 1);
		int X = 0, Y = 0;
		for (int i = 1; i <= n; ++i) {
			int t = gcd(X, a[id[i]]);
			if (t == X) Y = gcd(Y, a[id[i]]), ans[id[i]] = 2;
			else X = t, ans[id[i]] = 1;
		}
		if (X == 1 && Y == 1) {
			puts("YES");
			for (int i = 1; i <= n; ++i) printf("%d%c", ans[i], " 
"[i == n]);
			return 0;
		}
	}
	puts("NO");
	return 0;
}