제로베이스 20220513 코테 연습

1. 연결 요소의 개수

• 방향 없는 그래프가 주어졌을 때, 연결 요소 (Connected Component)의 개수를 구하는 프로그램을 작성하시오.

입력

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

출력

• 첫째 줄에 연결 요소의 개수를 출력한다.

풀이

import sys

N, M = [int(x) for x in sys.stdin.readline().split()]

graph = [[N + 1] * N for _ in range(N)]

for _ in range(M):
    a, b = [int(x) for x in sys.stdin.readline().split()]
    graph[a - 1][b - 1] = min(graph[a - 1][b - 1], b - 1)
    graph[b - 1][a - 1] = min(graph[b - 1][a - 1], a - 1)

left = [x for x in range(N)]


def dfs(x, visited):
    if x not in visited:
        visited.append(x)
        for k in graph[x]:
            if k != N + 1:
                visited = dfs(k, visited)
    return visited


cnt = 1
while left:
    v = dfs(left[0], [])
    for point in v:
        left.remove(point)

    if left:
        cnt += 1

print(cnt)

채점 결과

• 정답

리뷰

• 이번 코드의 경우, Python3로 제출시 recursionError가 발생하지만 PyPy3으로 제출할경우 정답처리가 나온다.

  • 이는 PyPy와 Python의 구조 차이로 인해 발생하는 문제라고 한다. 연관 글

• dfs로 연결된 요소를 모두 찾고, 리스트에서 해당 요소들을 제거하는것을 반복하며 연결된 요소의 갯수를 리턴한다.