STUDY/Algorithm

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

sinawi95 2021. 5. 4. 19:32
728x90

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
input = sys.stdin.readline
N, M = map(int, input().split())
# 방향 없는 그래프
G = [list() for _ in range(N + 1)]
visit = [1] + [0] * N
for _ in range(M):
    u, v = map(int, input().split())
    G[u].append(v)
    G[v].append(u)

result = 0
for i in range(1, N + 1):
    if visit[i]: continue
    def dfs(start):
        s = [start]
        while s:
            cur = s[-1]
            # if visit[cur]: continue
            visit[cur] = 1
            for adj in G[cur]:
                if visit[adj]: continue
                s.append(adj)
                break
            else:
                s.pop()
    result += 1
    dfs(i)
print(result)

기본을 묻는 문제이다.