Building Teams

import sys 

def solve():
    input = sys.stdin.read().split()

    n = int(input[0])
    m = int(input[1])

    adj = [[]for _ in range(n+1)]

    i = 2 
    for _ in range(m):
        u = int(input[i])
        v = int(input[i+1])
        adj[u].append(v)
        adj[v].append(u)
        i = i +2

    team = [0] *(n+1)

    visited = [False] *(n+1)

    for i in range(1,n+1): 
        if not visited[i] : 
            team[i] = 1
            stack = [i]
            while stack : 
                node = stack.pop()
                if not visited[node] : 
                    visited[node] = True
                    for neighbor in adj[node]:
                        if not visited[neighbor]: 
                            team[neighbor] = 2
                            stack.append(neighbor)
                            visited[neighbor] = True

    
    for i in range(1,len(team)):
        print(team[i], end =' ')

    
if __name__ == '__main__' :
    solve()

Test 1

Verdict: ACCEPTED

Test 2

Verdict: ACCEPTED

Test 3

Verdict: WRONG ANSWER

Test 4

Verdict: ACCEPTED

Test 5

Verdict: WRONG ANSWER

Test 6

Verdict: WRONG ANSWER

Test 7

Verdict: WRONG ANSWER

Test 8

Verdict: WRONG ANSWER

Test 9

Verdict: WRONG ANSWER

Test 10

Verdict: WRONG ANSWER

Test 11

Verdict: WRONG ANSWER

Test 12

Verdict: WRONG ANSWER