3-Coloring

#include <bits/stdc++.h>

using namespace std;

// 3-coloring problem

int main()
{
    // ios_base::sync_with_stdio(0);
    // cin.tie(0);

    int n;
    cin >> n;

    vector<int> edges(n);
    for (int i = 0; i < n; ++i) {
        cin >> edges[i];
        edges[i]--;
    }

    vector<int> color(n, 0);
    for (int i = 0; i < n; ++i) {
        // free to color
        if (color[i] == 0) {
            int actual = i;
            color[actual] = 1;
            while (true) {
                int next = edges[actual];
                if (color[next] == 0) {
                    color[next] = (color[actual] % 3) + 1;
                    actual = next;
                } else {
                    break;
                }
            }
        }
    }

    for (int i = 0; i < n; ++i) {
        cout << color[i] << " ";
    }
    cout << endl;
}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: WRONG ANSWER

Test 3

Verdict: WRONG ANSWER

Test 4

Verdict: WRONG ANSWER

Test 5

Verdict: WRONG ANSWER

Test 6

Verdict: WRONG ANSWER

Test 7

Verdict: WRONG ANSWER

Test 8

Verdict: WRONG ANSWER

Test 9

Verdict: WRONG ANSWER

Test 10

Verdict: WRONG ANSWER

Test 11

Verdict: WRONG ANSWER

Test 12

Verdict: WRONG ANSWER

Test 13

Verdict: WRONG ANSWER