[Python] 백준 #11724 연결 요소의 개수

www.acmicpc.net/problem/11724

11724번: 연결 요소의 개수

pypy3로 넣으면 통과, python3로 제출하면 시간초과가 난다 뭐가 문제일까?!

더 효율적으로 풀 수 있는 방법을 더 생각해봐야겠다

 

#pypy3로 넣으면 통과, python3로 넣으면 시간초과..뭐가 문제일까?!

from collections import deque
import sys

stack=deque()
n, m = map(int, sys.stdin.readline().split())
graph = {i: [] for i in range(1, n + 1)}

for i in range(m):
    a, b = map(int, sys.stdin.readline().split())
    graph[a].append(b)
    graph[b].append(a)

visited = [0 for i in range(n)]

cnt = 0
for i in range(1,n+1):
    if visited[i-1] == 0:
        stack.append(i)
        while stack:
            k = stack.pop()
            for node in graph[k]:
                if visited[node - 1] == 0:
                    stack.append(node)

            visited[k - 1] = 1
        cnt += 1

print(cnt)