Building Teams

#include <bits/stdc++.h>

using namespace std;

//Definitions for quicker writing
#define REP(i,a,b) for (int i = a; i < b; i++)

//Typedefs for quicker writing
typedef long long ll;
typedef vector<int> vi;
typedef pair<int,int> pi;

bool dfs(int s, int teams[], vi friends[], int prev){
	if((teams[s]==1 && prev==2) || (teams[s]==2 && prev==1)) return false;
	if((teams[s]==2 && prev==2) || (teams[s]==1 && prev==1)){
		return true;
	}
	if(prev == 1){
	       	teams[s] = 2; 
		prev = 2;
	}
	else if (prev == 2){
	       	teams[s] = 1; 
		prev = 1;
	}
	for (auto u: friends[s]){
		return dfs(u, teams, friends, prev);
	}
	return 0;
}

int main() {
	//IO optimization
	ios::sync_with_stdio(0);
	cin.tie(0);
	
	//Input definition
	int n, m;
	
	//Read In
	cin >> n >> m;

	vi friends[n];

	for(int i = 0; i < m; i++) {
		int a,b;
		cin >> a >> b;
		a--;
		b--;
		friends[a].push_back(b);
		friends[b].push_back(a);
	}


	//Main part
	int teams[n] = {0};
	bool imp;

	REP(i,0,n){
		sort(friends[i].begin(), friends[i].end());
	}

	imp = dfs(0, teams, friends, 2);

	if(!imp){
		REP(i,0,n){
			if(teams[i]==0){
				imp = dfs(i, teams, friends, 2);
			}
		}
	}

	//Write out
	if(!imp){
		REP(i,0,n){
			cout << teams[i] << "\n";
		}
	}
	else{
		cout << "IMPOSSIBLE" << "\n";
	}
		
	//Return
	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