본문 바로가기

코딩테스트/Python

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

https://www.acmicpc.net/problem/11724

 

11724번: 연결 요소의 개수

첫째 줄에 정점의 개수 N과 간선의 개수 M이 주어진다. (1 ≤ N ≤ 1,000, 0 ≤ M ≤ N×(N-1)/2) 둘째 줄부터 M개의 줄에 간선의 양 끝점 u와 v가 주어진다. (1 ≤ u, v ≤ N, u ≠ v) 같은 간선은 한 번만 주

www.acmicpc.net

 

내 코드

import sys
sys.setrecursionlimit(10000)

def dfs(graph,v,visited):
    visited[v] = True
    for i in graph[v]:
        if visited[i] == False:
            dfs(graph,i,visited)


N,M = map(int,input().split())
graph = [[] for _ in range(N+1)]
visited = [False]*(N+1)
visited[0] = True
for _ in range(M):
    l,r = map(int,sys.stdin.readline().split())
    graph[l].append(r)
    graph[r].append(l)

cnt=0
for i in range(1,N+1):
    if visited[i] == False:
        dfs(graph,i,visited)
        cnt+=1

print(cnt)
 
sys.stdin.readline()을 사용해야지 시간초과가 안난다. 
반복문을 사용할때는 input대신 이걸 사용해야지 시간절약이 많이 된다고한다.