Building Teams

#include <iostream>
#include <algorithm>
#include <queue>

using namespace std;

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

    vector<int> dist(n+1,0);

    vector<pair<int,int>> friendships;
    vector<vector<int>> adj(n+1);
    vector<int> team(n+1, 0); 

    for(int i = 0; i < m; i++) {
        int a, b;
        cin >> a >> b;
        
        adj[a].push_back(b);
        adj[b].push_back(a);
    }
    
    int start = 1;
    team[start] = 1;

    vector<bool> visited(n+1, false);
    
    for(int i = 1; i <= n; i++) {
        queue<int> q;


        visited[start] = true;
        q.push(start);

        while (!q.empty()) {
            int s = q.front(); q.pop();
            for (auto u : adj[s]) {
                if(visited[u]) continue;
                visited[u] = true;
                q.push(u);
                if(team[s] == 1) team[u] = 2;
                else team[u] = 1;
                if (team[u] == team[s]) {
                    cout << "IMPOSSIBLE" << endl;
                    return 0;
                }
                
            }
        }

        for(int j = 1; j <= n; j++) {
            if(!visited[j]) {
                
                start = j;
                team[start] = 1;
                break;
            }
        }
        /*for( int i = 1; i <= n; i++) {
            cout << i << ": " << visited[i] <<endl;
        }*/
    


}
        


    
    //sort(friendships.begin(), friendships.end());

       /*if(dist[friendships[i].first] == 1 && dist[friendships[i].second] == 0) {
           dist[friendships[i].second] = 2;
       }
       else if(dist[friendships[i].first] == 2 && dist[friendships[i].second] == 0) {
           dist[friendships[i].second] = 1;
       }
       else if(dist[friendships[i].first] == 0 && dist[friendships[i].second] == 1) {
           dist[friendships[i].first] = 2;
       }
       else if(dist[friendships[i].first] == 0 && dist[friendships[i].second] == 2) {
           dist[friendships[i].first] = 1;
       }
       else if(dist[friendships[i].first] == 0 && dist[friendships[i].second] == 0) {
           dist[friendships[i].first] = 1;
           dist[friendships[i].second] = 2;
           //cout << friendships[i].first << " " << friendships[i].second << endl;

       }
       else if(dist[friendships[i].first] == 2 && dist[friendships[i].second] == 1) {
           cout << "IMPOSSIBLE" << endl;
           return 0;
       }
       else if(dist[friendships[i].first] == 1 && dist[friendships[i].second] == 2) {
           cout << "IMPOSSIBLE" << endl;
           return 0;
       }*/
       
    

    for(int i = 1; i <= n; i++) {
        if(team[i] == 0 || team[i] == 1) cout << 1 << " ";
        else cout << 2 << " ";
        //cout << dist[i] << endl;
    }

}

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: WRONG ANSWER

Test 6

Verdict: TIME LIMIT EXCEEDED

Test 7

Verdict: TIME LIMIT EXCEEDED

Test 8

Verdict: TIME LIMIT EXCEEDED

Test 9

Verdict: TIME LIMIT EXCEEDED

Test 10

Verdict: TIME LIMIT EXCEEDED

Test 11

Verdict: WRONG ANSWER

Test 12

Verdict: WRONG ANSWER