Building Teams

import sys
sys.setrecursionlimit(100000)
line = input()

parts = line.split(" ")
pupils, friendships = int(parts[0]), int(parts[1])

adj = {}
for i in range(1, pupils + 1):
    adj[i] = []

for _ in range(friendships):
    parts = input().split(" ")
    a, b = int(parts[0]), int(parts[1])

    adj[a].append(b)
    adj[b].append(a)


res= []
for i in range(pupils+1):
    res.append("0")

#team = 2, !team = 1
def rec(p, team):
    
    if res[p] == "1" and team == "2":
        print("IMPOSSIBLE")
        exit(0)

    if res[p] == "2" and team == "1":
        print("IMPOSSIBLE")
        exit(0)

    if res[p] != "0":
        return
    
    res[p] = team
    
    for friend in adj[p]:
        rec(friend, "1" if team == "2" else "2")

for p in range(1, pupils + 1):
    if res[p] == "0":
        rec(p, "1")

print(" ".join(res[1:]))

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: ACCEPTED

Test 4

Verdict: ACCEPTED

Test 5

Verdict: ACCEPTED

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: ACCEPTED

Test 11

Verdict: ACCEPTED

Test 12

Verdict: ACCEPTED