📚 출처 - 11724 - 연결 요소의 개수
문제 설명
방향 없는 그래프가 주어졌을 때, 연결 요소 (Connected Component)의 개수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 정점의 개수 N과 간선의 개수 M이 주어진다. (1 ≤ N ≤ 1,000, 0 ≤ M ≤ N×(N-1)/2) 둘째 줄부터 M개의 줄에 간선의 양 끝점 u와 v가 주어진다. (1 ≤ u, v ≤ N, u ≠ v) 같은 간선은 한 번만 주어진다.
출력
첫째 줄에 연결 요소의 개수를 출력한다.
입출력 예
예제 입력 | 예제 출력 |
---|---|
6 5 1 2 2 5 5 1 3 4 4 6 | 2 |
6 8 1 2 2 5 5 1 3 4 4 6 5 4 2 4 2 3 | 1 |
Logic
저는 DFS 알고리즘을 이용해 풀었는데 푼 방법은 아래와 같습니다.
위의 첫 번째 예제를 예로 든다면,1-2-5, 3-4-6
이렇게 2개로 나뉘게 되어 정답은 2가 됩니다.
따라서 처음에만 counting해주고, 이어지는걸 체킹할때는 counting을 하지 않게끔 해주었습니다.
from collections import deque
import sys
input = sys.stdin.readline
def bfs(v):
queue = deque()
queue.append(v)
visitied[v] = 1
while queue:
n = queue.popleft()
for i, v in enumerate(graph[n]):
if v == 1 and visitied[i] == 0:
queue.append(i)
visitied[i] = 1
N, M = map(int, input().split())
graph = [[0] * (N+1) for _ in range(N+1)] # 0 제외 index-value 매칭
visitied = [0] * (N+1) # 0 제외 index-value 매칭
cnt = 0
# 간선 연결
for _ in range(M):
a, b = map(int, input().split())
graph[a][b] = 1
graph[b][a] = 1
# bfs checking
for i in range(1, N+1):
if visitied[i] == 0:
bfs(i)
cnt += 1
print(cnt)