알고리즘/문제풀이
[백준] 11724번 연결 요소의 개수 - 파이썬(Python)
SeongOnion
2021. 4. 20. 15:07
728x90
문제 (링크)
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
나의 풀이
# BFS 이용
from collections import deque
N, M = map(int, input().split())
visited = [False] * (N+1)
graph = [[] for _ in range(N+1)]
for _ in range(M):
x, y = map(int, input().split())
graph[x].append(y)
graph[y].append(x)
def bfs(v):
q = deque()
q.append(v)
while q:
x = q.popleft()
if visited[x] == False:
visited[x] = True
for i in graph[x]:
q.append(i)
cnt = 0
for i in range(1, N+1):
if visited[i] == False:
bfs(i)
cnt += 1
print(cnt)
# DFS 이용
N, M = map(int, input().split())
visited = [False] * (N+1)
graph = [[] for _ in range(N+1)]
for _ in range(M):
x, y = map(int, input().split())
graph[x].append(y)
graph[y].append(x)
def dfs(v):
visited[v] = True
for i in graph[v]:
if visited[i] == False:
dfs(i)
cnt = 0
for i in range(1, N+1):
if visited[i] == False:
dfs(i)
cnt += 1
print(cnt)
코드 설명 및 풀이법
BFS와 DFS 모두를 사용하여 풀 수 있는 문제이다.
왜 그런지 모르겠는데 백준에서 언어를 파이썬으로 선택하고 두 코드를 돌리면 모두 시간초과가 뜨고, PyPy3로 선택하고 돌리면 통과가 된다.
몇 가지 다른 문제들에도 비슷한 경우가 있는데 이유를 모르겠다.