Building Teams

input/code.cpp: In function 'int main(int, char**)':
input/code.cpp:8:10: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
    8 |     scanf("%d %d", &people, &friendships);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
input/code.cpp:13:14: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   13 |         scanf("%d %d", &person1, &person2);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~

#include <cstdio>
#include <cstdlib>

using namespace std;

int main (int argc, char *argv[]) {
    int people, friendships;
    scanf("%d %d", &people, &friendships);
    int *team = (int *)calloc(people, sizeof(int));

    for (int i = 0; i < friendships; i++) {
        int person1, person2;
        scanf("%d %d", &person1, &person2);
        person1--;
        person2--;
        if (team[person2] && team[person1]){
            printf("IMPOSSIBLE\n");
            free(team);
            return 0;
        }
        if (team[person1]){
            team[person2] = (team[person1] == 1 ? 2:1);
        }else if(team[person2]){
            team[person1] = (team[person2] == 1 ? 2:1);
        }else if((!team[person1]) && (!team[person2])){
            team[person1] = 1;
            team[person2] = 2;
        }
    }

    for (int i = 0; i < people; i++) {
        if (!team[i]){
            team[i]++;
        }
    }

    for (int i = 0; i < people; i++) {
        printf("%d ", team[i]);
    }

    printf("\n");

    free(team);

    return 0;
}

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