Building Teams

n, m = map(int, input().split())

friendships = [[] for _ in range(n)]
groups = [0 for _ in range(n)]

def findandlabel(x, team):
    for friendship in friendships[x]:
        if groups[friendship] == 0:
            groups[friendship] = 2 if groups[x] == 1 else 1
            team = not team
            findandlabel(friendship, team)


for i in range(m):
    a, b = [int(x) for x in input().split()]
    friendships[a-1].append(b-1)

print(friendships)


for i in range(len(friendships)):
    team = True # True is two, false is one
    if groups[i] == 0: groups[i] = 1
    print(groups)
    findandlabel(i, team)        

print(" ".join(map(str, groups)))

Test 1

Verdict: WRONG ANSWER

Test 2

Verdict: WRONG ANSWER

Test 3

Verdict: WRONG ANSWER

Test 4

Verdict: WRONG ANSWER

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