티스토리 뷰

PS/Python

[Python] 백준 알고리즘 #11724 연결 요소의 개수

yoo.o 2020. 8. 12. 00:38
반응형

문제

 

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

풀이

인풋을 인접 리스트에 저장하고, dfs로 탐색하는 방법으로 풀었다. 

 

import sys

# V: vertex, E: edge
V, E = map(int, sys.stdin.readline().split())
adj = [[] for _ in range(V)]
visited = [0 for _ in range(V)]
stack = []

def dfs():
    while stack:
        x = stack.pop()

        for each in adj[x]:
            if visited[each] == 0:
                visited[each] = 1
                stack.append(each)

# 모든 edge를 리스트에 저장
for _ in range(E):
    x, y = map(int,sys.stdin.readline().split(" "))
    adj[x-1].append(y-1)
    adj[y-1].append(x-1)


# 모든 vertex를 확인하기
count = 0
for i in range(V):
    if visited[i] == 0:
        count += 1
        stack.append(i)
        dfs()

print(count)

반응형
저작자표시