Building Teams

input/code.cpp: In function 'int main()':
input/code.cpp:19:13: warning: unused variable 'shop' [-Wunused-variable]
   19 |         int shop;
      |             ^~~~

#include <string>
#include <iostream>
#include <set>

using namespace std;

int main()
{
    int n_pups, n_friendships;
    cin >> n_pups >> n_friendships;
    set<int> team1;
    set<int> team2;
    int pup1, pup2;
    bool possible = true;

    for(int i = 0; i < n_friendships; ++i)
    {
        cin >> pup1 >> pup2;
        int shop;
        if(team1.count(pup1))
        {
            if(!team1.count(pup2)) team1.insert(pup1), team2.insert(pup2);
            else {printf("IMPOSSIBLE\n"); possible = false; break;}
        }
        else if(team2.count(pup1))
        {
            if(!team2.count(pup2)) team2.insert(pup1), team1.insert(pup2);
            else {printf("IMPOSSIBLE\n"); possible = false; break;}
        }
        else if(team1.count(pup2))
        {
            if(!team1.count(pup1)) team1.insert(pup2), team2.insert(pup1);
            else {printf("IMPOSSIBLE\n"); possible = false; break;}
        }
        else if(team2.count(pup2))
        {
            if(!team1.count(pup1)) team2.insert(pup2), team1.insert(pup1);
            else {printf("IMPOSSIBLE\n"); possible = false; break;}
        }
        else
        {
            team1.insert(pup1);
            team2.insert(pup2);
        }
    }


    set<int>::iterator it_teams1 = team1.begin();
    set<int>::iterator it_teams2 = team2.begin();

    if(possible)
    {    for(int i = 0; i < n_pups; ++i)
        {
            if(it_teams1 == team1.end()) printf("2"), it_teams2++;
            else if(it_teams2 == team2.end()) printf("1"), it_teams1++;
            else if(*it_teams1 < *it_teams2) printf("1"), it_teams1++;
            else if(*it_teams1 > *it_teams2) printf("2"), it_teams2++;
        }
        printf("\n");
    }
    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