글 읽기 - [파이썬] 시간초과인데 제 코드가 비효율적인가요?

hyuninlee 5년 전 0

테스트 케이스 검색해서 다 넣어봤는데 잘 됐습니다.

진행하다가 2%에서 멈추면서 시간초과로 끝나버리네요

비효율적인 부분이 있나요??

import sys
sys.setrecursionlimit(100000)

N, M = map(int, input().split())

line_list = []
for i in range(M):
    a, b = map(int, input().split())
    line_list.append([a, b])

visit_list = []
for i in range(N):
    visit_list.append(False)

def DFS(z):
    visit_list[z - 1] = True
    for i in range(len(line_list)):
        if line_list[i][0] == z and visit_list[line_list[i][1] - 1] == False:
            new_z = line_list[i][1]
            DFS(new_z)

        elif line_list[i][1] == z and visit_list[line_list[i][0] - 1] == False:
            new_z = line_list[i][0]
            DFS(new_z)

value = 0
for i in range(N):
    if visit_list[i] == False:
        value += 1
        DFS(i + 1)

print(value)