個人博客鏈接:https://blog.nuoyanli.com/2020/03/27/cf1328d/
題目鏈接
http://codeforces.com/contest/1328/problem/D
題意
給你一個數組,讓你給他染色,如果,則和不能染同一種顏色,若,你可以染成一種顏色,也可以染成不同顏色的。求染色用的顏色最少是多少及每個元素的顏色是什麼(,輸出任意滿足條件的解即可)。
思路
- 沒有相同類型只需要一種顏色即可
- 如果第一個和最後一個種類相同並且顏色相同,中間也有不同,就兩種顏色即可
- 如果第一個和最後一個種類相同並且顏色相同,其他的全部不相同,就三種顏色
- 如果第一個和最後一個種類不相同或者顏色吧相同,直接兩種顏色
詳細見代碼。
參考代碼
#include <bits/stdc++.h>
using namespace std;
#define IOS ios::sync_with_stdio(false), cin.tie(0)
#define endl '\n'
#define PB push_back
#define FI first
#define SE second
#define m_p(a, b) make_pair(a, b)
const double pi = acos(-1.0);
const double eps = 1e-9;
typedef long long LL;
const int N = 1e6 + 10;
const int M = 1e5 + 10;
const int mod = 1e9 + 7;
const int inf = 0x3f3f3f3f;
const double f = 2.32349;
int a[N], colr[N], n;
void solve() {
IOS;
int t;
cin >> t;
while (t--) {
cin >> n;
for (int i = 1; i <= n; i++) {
cin >> a[i];
colr[i] = 0;
}
bool f = 0;
for (int i = 1; i <= n; i++) {
if (i == 1) {
if (a[i] != a[n] || a[i] != a[i + 1]) {
f = 1;
break;
}
} else if (i == n) {
if (a[i] != a[i - 1] || a[i] != a[1]) {
f = 1;
break;
}
} else {
if (a[i] != a[i - 1] || a[i] != a[i + 1]) {
f = 1;
break;
}
}
}
if (!f) {//沒有相同類型只需要一種顏色即可
cout << 1 << endl;
for (int i = 1; i <= n; i++) {
cout << "1 ";
}
cout << endl;
} else {
colr[1] = 1;
int id = -1;
for (int i = 2; i <= n; i++) {//找到最後一個不相同類型的位置爲id,顏色初始化爲1,不相同就爲2
if (a[i] == a[i - 1]) {
colr[i] = colr[i - 1];
id = i;
} else {
colr[i] = 3 - colr[i - 1];
}
}
if (a[1] != a[n] && colr[1] == colr[n]) {//如果第一個和最後一個種類相同並且顏色相同
if (id != -1) {//中間有不同,就兩種顏色即可
colr[id] = 3 - colr[id];
for (int i = id + 1; i <= n; i++) {
colr[i] = 3 - colr[i - 1];
}
cout << 2 << endl;
for (int i = 1; i <= n; i++) {
cout << colr[i] << " ";
}
cout << endl;
} else {//其他全部相同,就三種顏色
colr[n] = 3;
cout << 3 << endl;
for (int i = 1; i <= n; i++) {
cout << colr[i] << " ";
}
cout << endl;
}
} else {//如果第一個和最後一個種類不相同或者顏色吧相同,直接兩種顏色
cout << 2 << endl;
for (int i = 1; i <= n; i++) {
cout << colr[i] << " ";
}
cout << endl;
}
}
}
}
signed main() {
solve();
return 0;
}
/*
5
1 2 1 2 2
6
1 2 2 1 2 2
5
1 2 1 2 3
3
10 10 10
2
1 2 1 1 2
2
1 1 2 1 1 2
3
1 2 1 2 3
1
1 1 1
*/