Building Teams

input/code.cpp: In function 'int main()':
input/code.cpp:28:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   28 |             for (int j = 0; j < friendships[i].size(); ++j) {
      |                             ~~^~~~~~~~~~~~~~~~~~~~~~~
input/code.cpp:38:31: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   38 |             for (int j = 0; j < friendships[i].size(); ++j) {
      |                             ~~^~~~~~~~~~~~~~~~~~~~~~~

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;


int main() {
    int n, m;
    cin >> n >> m;

    vector<vector<int>> friendships(n+1);

    // Read friendships
    for (int i = 0; i < m; ++i) {
        int f1, f2;
        cin >> f1 >> f2;
        friendships[f1].push_back(f2);
        friendships[f2].push_back(f1);
    }

    vector<int> team(n+1, -1);

    for (int i = 1; i <= n; ++i) {
        if (team[i] == -1)
            team[i] = 1;
        if (team[i] == 1) {
            for (int j = 0; j < friendships[i].size(); ++j) {
                int fr = friendships[i][j];
                if (team[fr] == 1) {
                    cout << "IMPOSSIBLE" << endl;
                    return 0;
                }
                else if (team[fr] == -1) team[fr] = 2;
            }
        }
        else {
            for (int j = 0; j < friendships[i].size(); ++j) {
                int fr = friendships[i][j];
                if (team[fr] == 2) {
                    cout << "IMPOSSIBLE" << endl;
                    return 0;
                }
                else if (team[fr] == -1) team[fr] = 2;
            }
        }
    }

    for (int i = 1; i <= n; ++i)
        cout << team[i] << " ";
    cout << endl;
}

Test 1

Verdict: WRONG ANSWER

Test 2

Verdict: WRONG ANSWER

Test 3

Verdict: WRONG ANSWER

Test 4

Verdict: WRONG ANSWER

Test 5

Verdict: ACCEPTED

Test 6

Verdict: WRONG ANSWER

Test 7

Verdict: WRONG ANSWER

Test 8

Verdict: WRONG ANSWER

Test 9

Verdict: WRONG ANSWER

Test 10

Verdict: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED